caching_test.rb 24.6 KB
Newer Older
1
require 'logger'
2
require 'abstract_unit'
J
Jeremy Kemper 已提交
3
require 'active_support/cache'
4

5
class CacheKeyTest < ActiveSupport::TestCase
6
  def test_expand_cache_key
7 8
    assert_equal '1/2/true', ActiveSupport::Cache.expand_cache_key([1, '2', true])
    assert_equal 'name/1/2/true', ActiveSupport::Cache.expand_cache_key([1, '2', true], :name)
9
  end
10 11

  def test_expand_cache_key_with_rails_cache_id
12 13 14
    begin
      ENV['RAILS_CACHE_ID'] = 'c99'
      assert_equal 'c99/foo', ActiveSupport::Cache.expand_cache_key(:foo)
15 16
      assert_equal 'c99/foo', ActiveSupport::Cache.expand_cache_key([:foo])
      assert_equal 'c99/foo/bar', ActiveSupport::Cache.expand_cache_key([:foo, :bar])
17
      assert_equal 'nm/c99/foo', ActiveSupport::Cache.expand_cache_key(:foo, :nm)
18 19
      assert_equal 'nm/c99/foo', ActiveSupport::Cache.expand_cache_key([:foo], :nm)
      assert_equal 'nm/c99/foo/bar', ActiveSupport::Cache.expand_cache_key([:foo, :bar], :nm)
20 21 22
    ensure
      ENV['RAILS_CACHE_ID'] = nil
    end
23
  end
24 25

  def test_expand_cache_key_with_rails_app_version
26 27 28 29 30 31
    begin
      ENV['RAILS_APP_VERSION'] = 'rails3'
      assert_equal 'rails3/foo', ActiveSupport::Cache.expand_cache_key(:foo)
    ensure
      ENV['RAILS_APP_VERSION'] = nil
    end
32 33 34
  end

  def test_expand_cache_key_rails_cache_id_should_win_over_rails_app_version
35 36 37 38 39 40 41 42 43 44
    begin
      ENV['RAILS_CACHE_ID'] = 'c99'
      ENV['RAILS_APP_VERSION'] = 'rails3'
      assert_equal 'c99/foo', ActiveSupport::Cache.expand_cache_key(:foo)
    ensure
      ENV['RAILS_CACHE_ID'] = nil
      ENV['RAILS_APP_VERSION'] = nil
    end
  end

45
  def test_expand_cache_key_respond_to_cache_key
46 47 48 49 50
    key = 'foo'
    def key.cache_key
      :foo_key
    end
    assert_equal 'foo_key', ActiveSupport::Cache.expand_cache_key(key)
51 52
  end

53
  def test_expand_cache_key_array_with_something_that_responds_to_cache_key
54 55 56 57
    key = 'foo'
    def key.cache_key
      :foo_key
    end
58
    assert_equal 'foo_key', ActiveSupport::Cache.expand_cache_key([key])
59 60
  end

61 62 63 64 65 66 67 68 69 70 71
  def test_expand_cache_key_of_nil
    assert_equal '', ActiveSupport::Cache.expand_cache_key(nil)
  end

  def test_expand_cache_key_of_false
    assert_equal 'false', ActiveSupport::Cache.expand_cache_key(false)
  end

  def test_expand_cache_key_of_true
    assert_equal 'true', ActiveSupport::Cache.expand_cache_key(true)
  end
72 73
end

74
class CacheStoreSettingTest < ActiveSupport::TestCase
75 76 77 78 79
  def test_file_fragment_cache_store
    store = ActiveSupport::Cache.lookup_store :file_store, "/path/to/cache/directory"
    assert_kind_of(ActiveSupport::Cache::FileStore, store)
    assert_equal "/path/to/cache/directory", store.cache_path
  end
80

81
  def test_mem_cache_fragment_cache_store
82
    MemCache.expects(:new).with(%w[localhost], {})
83 84
    store = ActiveSupport::Cache.lookup_store :mem_cache_store, "localhost"
    assert_kind_of(ActiveSupport::Cache::MemCacheStore, store)
85 86 87 88 89 90 91
  end

  def test_mem_cache_fragment_cache_store_with_given_mem_cache
    mem_cache = MemCache.new
    MemCache.expects(:new).never
    store = ActiveSupport::Cache.lookup_store :mem_cache_store, mem_cache
    assert_kind_of(ActiveSupport::Cache::MemCacheStore, store)
92
  end
93

94 95
  def test_mem_cache_fragment_cache_store_with_given_mem_cache_like_object
    MemCache.expects(:new).never
96 97 98
    memcache = Object.new
    def memcache.get() true end
    store = ActiveSupport::Cache.lookup_store :mem_cache_store, memcache
99 100 101
    assert_kind_of(ActiveSupport::Cache::MemCacheStore, store)
  end

102
  def test_mem_cache_fragment_cache_store_with_multiple_servers
103
    MemCache.expects(:new).with(%w[localhost 192.168.1.1], {})
104 105 106
    store = ActiveSupport::Cache.lookup_store :mem_cache_store, "localhost", '192.168.1.1'
    assert_kind_of(ActiveSupport::Cache::MemCacheStore, store)
  end
107

108
  def test_mem_cache_fragment_cache_store_with_options
B
Brian Durand 已提交
109 110
    MemCache.expects(:new).with(%w[localhost 192.168.1.1], { :timeout => 10 })
    store = ActiveSupport::Cache.lookup_store :mem_cache_store, "localhost", '192.168.1.1', :namespace => 'foo', :timeout => 10
111
    assert_kind_of(ActiveSupport::Cache::MemCacheStore, store)
B
Brian Durand 已提交
112
    assert_equal 'foo', store.options[:namespace]
113
  end
114 115 116 117 118 119 120

  def test_object_assigned_fragment_cache_store
    store = ActiveSupport::Cache.lookup_store ActiveSupport::Cache::FileStore.new("/path/to/cache/directory")
    assert_kind_of(ActiveSupport::Cache::FileStore, store)
    assert_equal "/path/to/cache/directory", store.cache_path
  end
end
121

B
Brian Durand 已提交
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
class CacheStoreNamespaceTest < ActiveSupport::TestCase
  def test_static_namespace
    cache = ActiveSupport::Cache.lookup_store(:memory_store, :namespace => "tester")
    cache.write("foo", "bar")
    assert_equal "bar", cache.read("foo")
    assert_equal "bar", cache.instance_variable_get(:@data)["tester:foo"].value
  end

  def test_proc_namespace
    test_val = "tester"
    proc = lambda{test_val}
    cache = ActiveSupport::Cache.lookup_store(:memory_store, :namespace => proc)
    cache.write("foo", "bar")
    assert_equal "bar", cache.read("foo")
    assert_equal "bar", cache.instance_variable_get(:@data)["tester:foo"].value
  end

  def test_delete_matched_key_start
    cache = ActiveSupport::Cache.lookup_store(:memory_store, :namespace => "tester")
    cache.write("foo", "bar")
    cache.write("fu", "baz")
    cache.delete_matched(/^fo/)
144 145
    assert !cache.exist?("foo")
    assert cache.exist?("fu")
B
Brian Durand 已提交
146 147 148 149 150 151 152
  end

  def test_delete_matched_key
    cache = ActiveSupport::Cache.lookup_store(:memory_store, :namespace => "foo")
    cache.write("foo", "bar")
    cache.write("fu", "baz")
    cache.delete_matched(/OO/i)
153 154
    assert !cache.exist?("foo")
    assert cache.exist?("fu")
B
Brian Durand 已提交
155 156 157 158 159 160
  end
end

# Tests the base functionality that should be identical across all cache stores.
module CacheStoreBehavior
  def test_should_read_and_write_strings
161
    assert @cache.write('foo', 'bar')
B
Brian Durand 已提交
162 163 164 165 166 167 168
    assert_equal 'bar', @cache.read('foo')
  end

  def test_should_overwrite
    @cache.write('foo', 'bar')
    @cache.write('foo', 'baz')
    assert_equal 'baz', @cache.read('foo')
169
  end
170

171
  def test_fetch_without_cache_miss
B
Brian Durand 已提交
172
    @cache.write('foo', 'bar')
173 174 175
    @cache.expects(:write).never
    assert_equal 'bar', @cache.fetch('foo') { 'baz' }
  end
176

177
  def test_fetch_with_cache_miss
B
Brian Durand 已提交
178
    @cache.expects(:write).with('foo', 'baz', @cache.options)
179 180
    assert_equal 'baz', @cache.fetch('foo') { 'baz' }
  end
181

182
  def test_fetch_with_forced_cache_miss
B
Brian Durand 已提交
183
    @cache.write('foo', 'bar')
184
    @cache.expects(:read).never
B
Brian Durand 已提交
185
    @cache.expects(:write).with('foo', 'bar', @cache.options.merge(:force => true))
186
    @cache.fetch('foo', :force => true) { 'bar' }
187
  end
J
Joshua Peek 已提交
188

B
Brian Durand 已提交
189 190 191 192
  def test_fetch_with_cached_nil
    @cache.write('foo', nil)
    @cache.expects(:write).never
    assert_nil @cache.fetch('foo') { 'baz' }
J
Joshua Peek 已提交
193 194 195
  end

  def test_should_read_and_write_hash
196
    assert @cache.write('foo', {:a => "b"})
J
Joshua Peek 已提交
197 198 199
    assert_equal({:a => "b"}, @cache.read('foo'))
  end

200
  def test_should_read_and_write_integer
201
    assert @cache.write('foo', 1)
202 203 204
    assert_equal 1, @cache.read('foo')
  end

J
Joshua Peek 已提交
205
  def test_should_read_and_write_nil
206
    assert @cache.write('foo', nil)
J
Joshua Peek 已提交
207
    assert_equal nil, @cache.read('foo')
208 209
  end

210
  def test_should_read_and_write_false
211
    assert @cache.write('foo', false)
212
    assert_equal false, @cache.read('foo')
213 214
  end

B
Brian Durand 已提交
215
  def test_read_multi
216
    @cache.write('foo', 'bar')
B
Brian Durand 已提交
217 218 219
    @cache.write('fu', 'baz')
    @cache.write('fud', 'biz')
    assert_equal({"foo" => "bar", "fu" => "baz"}, @cache.read_multi('foo', 'fu'))
220
  end
221

222 223 224 225 226 227 228
  def test_read_multi_with_expires
    @cache.write('foo', 'bar', :expires_in => 0.001)
    @cache.write('fu', 'baz')
    @cache.write('fud', 'biz')
    sleep(0.002)
    assert_equal({"fu" => "baz"}, @cache.read_multi('foo', 'fu'))
  end
229

B
Brian Durand 已提交
230 231 232 233
  def test_read_and_write_compressed_small_data
    @cache.write('foo', 'bar', :compress => true)
    raw_value = @cache.send(:read_entry, 'foo', {}).raw_value
    assert_equal 'bar', @cache.read('foo')
234
    assert_equal 'bar', Marshal.load(raw_value)
235 236
  end

B
Brian Durand 已提交
237 238 239 240 241
  def test_read_and_write_compressed_large_data
    @cache.write('foo', 'bar', :compress => true, :compress_threshold => 2)
    raw_value = @cache.send(:read_entry, 'foo', {}).raw_value
    assert_equal 'bar', @cache.read('foo')
    assert_equal 'bar', Marshal.load(Zlib::Inflate.inflate(raw_value))
242
  end
243

B
Brian Durand 已提交
244 245 246
  def test_read_and_write_compressed_nil
    @cache.write('foo', nil, :compress => true)
    assert_nil @cache.read('foo')
247 248
  end

B
Brian Durand 已提交
249 250 251 252 253 254 255
  def test_cache_key
    obj = Object.new
    def obj.cache_key
      :foo
    end
    @cache.write(obj, "bar")
    assert_equal "bar", @cache.read("foo")
256
  end
257

B
Brian Durand 已提交
258 259 260 261 262 263 264
  def test_param_as_cache_key
    obj = Object.new
    def obj.to_param
      "foo"
    end
    @cache.write(obj, "bar")
    assert_equal "bar", @cache.read("foo")
265
  end
266

B
Brian Durand 已提交
267 268 269
  def test_array_as_cache_key
    @cache.write([:fu, "foo"], "bar")
    assert_equal "bar", @cache.read("fu/foo")
270 271
  end

B
Brian Durand 已提交
272 273 274
  def test_hash_as_cache_key
    @cache.write({:foo => 1, :fu => 2}, "bar")
    assert_equal "bar", @cache.read("foo=1/fu=2")
275 276
  end

B
Brian Durand 已提交
277 278 279 280
  def test_keys_are_case_sensitive
    @cache.write("foo", "bar")
    assert_nil @cache.read("FOO")
  end
281

B
Brian Durand 已提交
282
  def test_exist
283
    @cache.write('foo', 'bar')
284 285
    assert @cache.exist?('foo')
    assert !@cache.exist?('bar')
286
  end
287

B
Brian Durand 已提交
288 289
  def test_nil_exist
    @cache.write('foo', nil)
290
    assert @cache.exist?('foo')
291 292
  end

B
Brian Durand 已提交
293 294 295
  def test_delete
    @cache.write('foo', 'bar')
    assert @cache.exist?('foo')
296
    assert @cache.delete('foo')
B
Brian Durand 已提交
297 298
    assert !@cache.exist?('foo')
  end
299

300
  def test_read_should_return_a_different_object_id_each_time_it_is_called
301
    @cache.write('foo', 'bar')
302 303 304 305
    assert_not_equal @cache.read('foo').object_id, @cache.read('foo').object_id
    value = @cache.read('foo')
    value << 'bingo'
    assert_not_equal value, @cache.read('foo')
306
  end
307 308 309 310 311 312

  def test_original_store_objects_should_not_be_immutable
    bar = 'bar'
    @cache.write('foo', bar)
    assert_nothing_raised { bar.gsub!(/.*/, 'baz') }
  end
313

B
Brian Durand 已提交
314 315 316 317 318 319 320 321 322 323 324 325
  def test_expires_in
    time = Time.local(2008, 4, 24)
    Time.stubs(:now).returns(time)

    @cache.write('foo', 'bar')
    assert_equal 'bar', @cache.read('foo')

    Time.stubs(:now).returns(time + 30)
    assert_equal 'bar', @cache.read('foo')

    Time.stubs(:now).returns(time + 61)
    assert_nil @cache.read('foo')
326
  end
327

B
Brian Durand 已提交
328 329 330 331 332 333 334
  def test_race_condition_protection
    time = Time.now
    @cache.write('foo', 'bar', :expires_in => 60)
    Time.stubs(:now).returns(time + 61)
    result = @cache.fetch('foo', :race_condition_ttl => 10) do
      assert_equal 'bar', @cache.read('foo')
      "baz"
335
    end
B
Brian Durand 已提交
336 337
    assert_equal "baz", result
  end
338

B
Brian Durand 已提交
339 340 341 342 343 344 345 346 347 348
  def test_race_condition_protection_is_limited
    time = Time.now
    @cache.write('foo', 'bar', :expires_in => 60)
    Time.stubs(:now).returns(time + 71)
    result = @cache.fetch('foo', :race_condition_ttl => 10) do
      assert_equal nil, @cache.read('foo')
      "baz"
    end
    assert_equal "baz", result
  end
349

B
Brian Durand 已提交
350 351 352 353 354 355
  def test_race_condition_protection_is_safe
    time = Time.now
    @cache.write('foo', 'bar', :expires_in => 60)
    Time.stubs(:now).returns(time + 61)
    begin
      @cache.fetch('foo', :race_condition_ttl => 10) do
356
        assert_equal 'bar', @cache.read('foo')
B
Brian Durand 已提交
357
        raise ArgumentError.new
358
      end
S
Santiago Pastorino 已提交
359
    rescue ArgumentError
360
    end
B
Brian Durand 已提交
361 362 363 364 365 366 367
    assert_equal "bar", @cache.read('foo')
    Time.stubs(:now).returns(time + 71)
    assert_nil @cache.read('foo')
  end

  def test_crazy_key_characters
    crazy_key = "#/:*(<+=> )&$%@?;'\"\'`~-"
368
    assert @cache.write(crazy_key, "1", :raw => true)
B
Brian Durand 已提交
369 370
    assert_equal "1", @cache.read(crazy_key)
    assert_equal "1", @cache.fetch(crazy_key)
371
    assert @cache.delete(crazy_key)
B
Brian Durand 已提交
372 373 374 375 376 377 378
    assert_equal "2", @cache.fetch(crazy_key, :raw => true) { "2" }
    assert_equal 3, @cache.increment(crazy_key)
    assert_equal 2, @cache.decrement(crazy_key)
  end

  def test_really_long_keys
    key = ""
379
    900.times{key << "x"}
380
    assert @cache.write(key, "bar")
B
Brian Durand 已提交
381 382 383 384
    assert_equal "bar", @cache.read(key)
    assert_equal "bar", @cache.fetch(key)
    assert_nil @cache.read("#{key}x")
    assert_equal({key => "bar"}, @cache.read_multi(key))
385
    assert @cache.delete(key)
B
Brian Durand 已提交
386 387
  end
end
388

389 390 391 392
# https://rails.lighthouseapp.com/projects/8994/tickets/6225-memcachestore-cant-deal-with-umlauts-and-special-characters
# The error is caused by charcter encodings that can't be compared with ASCII-8BIT regular expressions and by special
# characters like the umlaut in UTF-8.
module EncodedKeyCacheBehavior
393 394 395
  Encoding.list.each do |encoding|
    define_method "test_#{encoding.name.underscore}_encoded_values" do
      key = "foo".force_encoding(encoding)
396
      assert @cache.write(key, "1", :raw => true)
397 398
      assert_equal "1", @cache.read(key)
      assert_equal "1", @cache.fetch(key)
399
      assert @cache.delete(key)
400 401 402 403
      assert_equal "2", @cache.fetch(key, :raw => true) { "2" }
      assert_equal 3, @cache.increment(key)
      assert_equal 2, @cache.decrement(key)
    end
404
  end
405

406 407 408 409 410 411 412 413 414 415 416 417 418 419 420
  def test_common_utf8_values
    key = "\xC3\xBCmlaut".force_encoding(Encoding::UTF_8)
    assert @cache.write(key, "1", :raw => true)
    assert_equal "1", @cache.read(key)
    assert_equal "1", @cache.fetch(key)
    assert @cache.delete(key)
    assert_equal "2", @cache.fetch(key, :raw => true) { "2" }
    assert_equal 3, @cache.increment(key)
    assert_equal 2, @cache.decrement(key)
  end

  def test_retains_encoding
    key = "\xC3\xBCmlaut".force_encoding(Encoding::UTF_8)
    assert @cache.write(key, "1", :raw => true)
    assert_equal Encoding::UTF_8, key.encoding
421 422 423
  end
end

B
Brian Durand 已提交
424 425 426 427
module CacheDeleteMatchedBehavior
  def test_delete_matched
    @cache.write("foo", "bar")
    @cache.write("fu", "baz")
428 429
    @cache.write("foo/bar", "baz")
    @cache.write("fu/baz", "bar")
B
Brian Durand 已提交
430
    @cache.delete_matched(/oo/)
431 432 433 434
    assert !@cache.exist?("foo")
    assert @cache.exist?("fu")
    assert !@cache.exist?("foo/bar")
    assert @cache.exist?("fu/baz")
B
Brian Durand 已提交
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461
  end
end

module CacheIncrementDecrementBehavior
  def test_increment
    @cache.write('foo', 1, :raw => true)
    assert_equal 1, @cache.read('foo').to_i
    assert_equal 2, @cache.increment('foo')
    assert_equal 2, @cache.read('foo').to_i
    assert_equal 3, @cache.increment('foo')
    assert_equal 3, @cache.read('foo').to_i
  end

  def test_decrement
    @cache.write('foo', 3, :raw => true)
    assert_equal 3, @cache.read('foo').to_i
    assert_equal 2, @cache.decrement('foo')
    assert_equal 2, @cache.read('foo').to_i
    assert_equal 1, @cache.decrement('foo')
    assert_equal 1, @cache.read('foo').to_i
  end
end

module LocalCacheBehavior
  def test_local_writes_are_persistent_on_the_remote_cache
    retval = @cache.with_local_cache do
      @cache.write('foo', 'bar')
462
    end
463
    assert retval
B
Brian Durand 已提交
464 465
    assert_equal 'bar', @cache.read('foo')
  end
466

B
Brian Durand 已提交
467 468 469 470 471
  def test_clear_also_clears_local_cache
    @cache.with_local_cache do
      @cache.write('foo', 'bar')
      @cache.clear
      assert_nil @cache.read('foo')
472 473
    end

B
Brian Durand 已提交
474 475
    assert_nil @cache.read('foo')
  end
476

B
Brian Durand 已提交
477 478 479 480
  def test_local_cache_of_write
    @cache.with_local_cache do
      @cache.write('foo', 'bar')
      @peek.delete('foo')
481 482
      assert_equal 'bar', @cache.read('foo')
    end
B
Brian Durand 已提交
483
  end
484

B
Brian Durand 已提交
485 486 487 488
  def test_local_cache_of_read
    @cache.write('foo', 'bar')
    @cache.with_local_cache do
      assert_equal 'bar', @cache.read('foo')
489
    end
B
Brian Durand 已提交
490
  end
491

B
Brian Durand 已提交
492 493
  def test_local_cache_of_write_nil
    @cache.with_local_cache do
494
      assert @cache.write('foo', nil)
B
Brian Durand 已提交
495 496 497
      assert_nil @cache.read('foo')
      @peek.write('foo', 'bar')
      assert_nil @cache.read('foo')
498
    end
B
Brian Durand 已提交
499
  end
500

B
Brian Durand 已提交
501 502 503 504 505
  def test_local_cache_of_delete
    @cache.with_local_cache do
      @cache.write('foo', 'bar')
      @cache.delete('foo')
      assert_nil @cache.read('foo')
506
    end
B
Brian Durand 已提交
507
  end
508

B
Brian Durand 已提交
509 510 511 512
  def test_local_cache_of_exist
    @cache.with_local_cache do
      @cache.write('foo', 'bar')
      @peek.delete('foo')
513
      assert @cache.exist?('foo')
514
    end
B
Brian Durand 已提交
515
  end
516

B
Brian Durand 已提交
517 518 519 520 521 522
  def test_local_cache_of_increment
    @cache.with_local_cache do
      @cache.write('foo', 1, :raw => true)
      @peek.write('foo', 2, :raw => true)
      @cache.increment('foo')
      assert_equal 3, @cache.read('foo')
523
    end
B
Brian Durand 已提交
524
  end
525

B
Brian Durand 已提交
526 527 528 529 530 531
  def test_local_cache_of_decrement
    @cache.with_local_cache do
      @cache.write('foo', 1, :raw => true)
      @peek.write('foo', 3, :raw => true)
      @cache.decrement('foo')
      assert_equal 2, @cache.read('foo')
532
    end
B
Brian Durand 已提交
533
  end
534

B
Brian Durand 已提交
535 536 537 538 539 540 541 542 543 544
  def test_middleware
    app = lambda { |env|
      result = @cache.write('foo', 'bar')
      assert_equal 'bar', @cache.read('foo') # make sure 'foo' was written
      assert result
    }
    app = @cache.middleware.new(app)
    app.call({})
  end
end
545

B
Brian Durand 已提交
546 547 548 549 550
class FileStoreTest < ActiveSupport::TestCase
  def setup
    Dir.mkdir(cache_dir) unless File.exist?(cache_dir)
    @cache = ActiveSupport::Cache.lookup_store(:file_store, cache_dir, :expires_in => 60)
    @peek = ActiveSupport::Cache.lookup_store(:file_store, cache_dir, :expires_in => 60)
551
    @cache_with_pathname = ActiveSupport::Cache.lookup_store(:file_store, Pathname.new(cache_dir), :expires_in => 60)
B
Brian Durand 已提交
552 553 554 555 556 557 558 559 560 561 562 563 564 565 566
  end

  def teardown
    FileUtils.rm_r(cache_dir)
  end

  def cache_dir
    File.join(Dir.pwd, 'tmp_cache')
  end

  include CacheStoreBehavior
  include LocalCacheBehavior
  include CacheDeleteMatchedBehavior
  include CacheIncrementDecrementBehavior

567 568 569 570
  def test_key_transformation
    key = @cache.send(:key_file_path, "views/index?id=1")
    assert_equal "views/index?id=1", @cache.send(:file_path_key, key)
  end
571 572 573 574 575 576

  def test_key_transformation_with_pathname
    FileUtils.touch(File.join(cache_dir, "foo"))
    key = @cache_with_pathname.send(:key_file_path, "views/index?id=1")
    assert_equal "views/index?id=1", @cache_with_pathname.send(:file_path_key, key)
  end
577

578 579 580 581
  # Because file systems have a maximum filename size, filenames > max size should be split in to directories
  # If filename is 'AAAAB', where max size is 4, the returned path should be AAAA/B
  def test_key_transformation_max_filename_size
    key = "#{'A' * ActiveSupport::Cache::FileStore::FILENAME_MAX_SIZE}B"
582
    path = @cache.send(:key_file_path, key)
583 584 585
    assert path.split('/').all? { |dir_name| dir_name.size <= ActiveSupport::Cache::FileStore::FILENAME_MAX_SIZE}
    assert_equal 'B', File.basename(path)
  end
586

587 588 589 590 591
  # If nothing has been stored in the cache, there is a chance the cache directory does not yet exist
  # Ensure delete_matched gracefully handles this case
  def test_delete_matched_when_cache_directory_does_not_exist
    assert_nothing_raised(Exception) do
      ActiveSupport::Cache::FileStore.new('/test/cache/directory').delete_matched(/does_not_exist/)
592 593
    end
  end
B
Brian Durand 已提交
594
end
595

B
Brian Durand 已提交
596 597
class MemoryStoreTest < ActiveSupport::TestCase
  def setup
598 599
    @record_size = Marshal.dump("aaaaaaaaaa").bytesize
    @cache = ActiveSupport::Cache.lookup_store(:memory_store, :expires_in => 60, :size => @record_size * 10)
B
Brian Durand 已提交
600 601 602 603 604 605 606 607 608 609 610 611 612 613
  end

  include CacheStoreBehavior
  include CacheDeleteMatchedBehavior
  include CacheIncrementDecrementBehavior

  def test_prune_size
    @cache.write(1, "aaaaaaaaaa") && sleep(0.001)
    @cache.write(2, "bbbbbbbbbb") && sleep(0.001)
    @cache.write(3, "cccccccccc") && sleep(0.001)
    @cache.write(4, "dddddddddd") && sleep(0.001)
    @cache.write(5, "eeeeeeeeee") && sleep(0.001)
    @cache.read(2) && sleep(0.001)
    @cache.read(4)
614
    @cache.prune(@record_size * 3)
615 616 617 618 619
    assert @cache.exist?(5)
    assert @cache.exist?(4)
    assert !@cache.exist?(3)
    assert @cache.exist?(2)
    assert !@cache.exist?(1)
B
Brian Durand 已提交
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635
  end

  def test_prune_size_on_write
    @cache.write(1, "aaaaaaaaaa") && sleep(0.001)
    @cache.write(2, "bbbbbbbbbb") && sleep(0.001)
    @cache.write(3, "cccccccccc") && sleep(0.001)
    @cache.write(4, "dddddddddd") && sleep(0.001)
    @cache.write(5, "eeeeeeeeee") && sleep(0.001)
    @cache.write(6, "ffffffffff") && sleep(0.001)
    @cache.write(7, "gggggggggg") && sleep(0.001)
    @cache.write(8, "hhhhhhhhhh") && sleep(0.001)
    @cache.write(9, "iiiiiiiiii") && sleep(0.001)
    @cache.write(10, "kkkkkkkkkk") && sleep(0.001)
    @cache.read(2) && sleep(0.001)
    @cache.read(4) && sleep(0.001)
    @cache.write(11, "llllllllll")
636 637 638 639 640 641 642 643 644 645 646
    assert @cache.exist?(11)
    assert @cache.exist?(10)
    assert @cache.exist?(9)
    assert @cache.exist?(8)
    assert @cache.exist?(7)
    assert !@cache.exist?(6)
    assert !@cache.exist?(5)
    assert @cache.exist?(4)
    assert !@cache.exist?(3)
    assert @cache.exist?(2)
    assert !@cache.exist?(1)
B
Brian Durand 已提交
647 648 649 650 651 652
  end

  def test_pruning_is_capped_at_a_max_time
    def @cache.delete_entry (*args)
      sleep(0.01)
      super
653
    end
B
Brian Durand 已提交
654 655 656 657 658 659
    @cache.write(1, "aaaaaaaaaa") && sleep(0.001)
    @cache.write(2, "bbbbbbbbbb") && sleep(0.001)
    @cache.write(3, "cccccccccc") && sleep(0.001)
    @cache.write(4, "dddddddddd") && sleep(0.001)
    @cache.write(5, "eeeeeeeeee") && sleep(0.001)
    @cache.prune(30, 0.001)
660 661 662 663 664
    assert @cache.exist?(5)
    assert @cache.exist?(4)
    assert @cache.exist?(3)
    assert @cache.exist?(2)
    assert !@cache.exist?(1)
B
Brian Durand 已提交
665 666
  end
end
667

B
Brian Durand 已提交
668 669 670 671 672 673 674 675
uses_memcached 'memcached backed store' do
  class MemCacheStoreTest < ActiveSupport::TestCase
    def setup
      @cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :expires_in => 60)
      @peek = ActiveSupport::Cache.lookup_store(:mem_cache_store)
      @data = @cache.instance_variable_get(:@data)
      @cache.clear
      @cache.silence!
676
      @cache.logger = ActiveSupport::Logger.new("/dev/null")
677 678
    end

B
Brian Durand 已提交
679 680 681
    include CacheStoreBehavior
    include LocalCacheBehavior
    include CacheIncrementDecrementBehavior
682
    include EncodedKeyCacheBehavior
B
Brian Durand 已提交
683 684 685 686 687 688

    def test_raw_values
      cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :raw => true)
      cache.clear
      cache.write("foo", 2)
      assert_equal "2", cache.read("foo")
689
    end
690

691 692 693 694
    def test_raw_values_with_marshal
      cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :raw => true)
      cache.clear
      cache.write("foo", Marshal.dump([]))
695
      assert_equal [], cache.read("foo")
696
    end
697

B
Brian Durand 已提交
698 699 700 701 702 703 704
    def test_local_cache_raw_values
      cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :raw => true)
      cache.clear
      cache.with_local_cache do
        cache.write("foo", 2)
        assert_equal "2", cache.read("foo")
      end
705
    end
706 707 708 709 710 711 712 713 714

    def test_local_cache_raw_values_with_marshal
      cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :raw => true)
      cache.clear
      cache.with_local_cache do
        cache.write("foo", Marshal.dump([]))
        assert_equal [], cache.read("foo")
      end
    end
715 716
  end
end
717

718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
class NullStoreTest < ActiveSupport::TestCase
  def setup
    @cache = ActiveSupport::Cache.lookup_store(:null_store)
  end

  def test_clear
    @cache.clear
  end

  def test_cleanup
    @cache.cleanup
  end

  def test_write
    assert_equal true, @cache.write("name", "value")
  end

  def test_read
    @cache.write("name", "value")
    assert_nil @cache.read("name")
  end

  def test_delete
    @cache.write("name", "value")
    assert_equal false, @cache.delete("name")
  end

  def test_increment
    @cache.write("name", 1, :raw => true)
    assert_nil @cache.increment("name")
  end

  def test_decrement
    @cache.write("name", 1, :raw => true)
    assert_nil @cache.increment("name")
  end

  def test_delete_matched
    @cache.write("name", "value")
    @cache.delete_matched(/name/)
  end

  def test_local_store_strategy
    @cache.with_local_cache do
      @cache.write("name", "value")
      assert_equal "value", @cache.read("name")
      @cache.delete("name")
      assert_nil @cache.read("name")
      @cache.write("name", "value")
    end
    assert_nil @cache.read("name")
  end

  def test_setting_nil_cache_store
    assert ActiveSupport::Cache.lookup_store.class.name, ActiveSupport::Cache::NullStore.name
  end
end

776 777 778 779 780
class CacheStoreLoggerTest < ActiveSupport::TestCase
  def setup
    @cache = ActiveSupport::Cache.lookup_store(:memory_store)

    @buffer = StringIO.new
781
    @cache.logger = ActiveSupport::Logger.new(@buffer)
782 783 784 785
  end

  def test_logging
    @cache.fetch('foo') { 'bar' }
786
    assert_present @buffer.string
787 788 789 790
  end

  def test_mute_logging
    @cache.mute { @cache.fetch('foo') { 'bar' } }
791
    assert_blank @buffer.string
792 793
  end
end
B
Brian Durand 已提交
794 795 796 797 798 799 800

class CacheEntryTest < ActiveSupport::TestCase
  def test_create_raw_entry
    time = Time.now
    entry = ActiveSupport::Cache::Entry.create("raw", time, :compress => false, :expires_in => 300)
    assert_equal "raw", entry.raw_value
    assert_equal time.to_f, entry.created_at
801
    assert !entry.compressed?
B
Brian Durand 已提交
802 803 804 805 806
    assert_equal 300, entry.expires_in
  end

  def test_expired
    entry = ActiveSupport::Cache::Entry.new("value")
807
    assert !entry.expired?, 'entry not expired'
B
Brian Durand 已提交
808
    entry = ActiveSupport::Cache::Entry.new("value", :expires_in => 60)
809
    assert !entry.expired?, 'entry not expired'
B
Brian Durand 已提交
810 811
    time = Time.now + 61
    Time.stubs(:now).returns(time)
812
    assert entry.expired?, 'entry is expired'
B
Brian Durand 已提交
813 814 815 816 817
  end

  def test_compress_values
    entry = ActiveSupport::Cache::Entry.new("value", :compress => true, :compress_threshold => 1)
    assert_equal "value", entry.value
818
    assert entry.compressed?
B
Brian Durand 已提交
819 820 821 822 823 824
    assert_equal "value", Marshal.load(Zlib::Inflate.inflate(entry.raw_value))
  end

  def test_non_compress_values
    entry = ActiveSupport::Cache::Entry.new("value")
    assert_equal "value", entry.value
825
    assert_equal "value", Marshal.load(entry.raw_value)
826
    assert !entry.compressed?
B
Brian Durand 已提交
827 828
  end
end