caching_test.rb 23.5 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
B
Brian Durand 已提交
7
    assert_equal '1/2/true', ActiveSupport::Cache.expand_cache_key([1, '2', true])
8 9
    assert_equal 'name/1/2/true', ActiveSupport::Cache.expand_cache_key([1, '2', true], :name)
  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 58 59 60
    key = 'foo'
    def key.cache_key
      :foo_key
    end
    assert_equal 'foo_key', ActiveSupport::Cache.expand_cache_key([key])
  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 393 394 395 396
# 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
  if defined?(Encoding)
    Encoding.list.each do |encoding|
      define_method "test_#{encoding.name.underscore}_encoded_values" do
        key = "foo".force_encoding(encoding)
397
        assert @cache.write(key, "1", :raw => true)
398 399
        assert_equal "1", @cache.read(key)
        assert_equal "1", @cache.fetch(key)
400
        assert @cache.delete(key)
401 402 403 404 405 406 407 408
        assert_equal "2", @cache.fetch(key, :raw => true) { "2" }
        assert_equal 3, @cache.increment(key)
        assert_equal 2, @cache.decrement(key)
      end
    end

    def test_common_utf8_values
      key = "\xC3\xBCmlaut".force_encoding(Encoding::UTF_8)
409
      assert @cache.write(key, "1", :raw => true)
410 411
      assert_equal "1", @cache.read(key)
      assert_equal "1", @cache.fetch(key)
412
      assert @cache.delete(key)
413 414 415 416 417 418 419
      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)
420
      assert @cache.write(key, "1", :raw => true)
421 422 423 424 425
      assert_equal Encoding::UTF_8, key.encoding
    end
  end
end

B
Brian Durand 已提交
426 427 428 429
module CacheDeleteMatchedBehavior
  def test_delete_matched
    @cache.write("foo", "bar")
    @cache.write("fu", "baz")
430 431
    @cache.write("foo/bar", "baz")
    @cache.write("fu/baz", "bar")
B
Brian Durand 已提交
432
    @cache.delete_matched(/oo/)
433 434 435 436
    assert !@cache.exist?("foo")
    assert @cache.exist?("fu")
    assert !@cache.exist?("foo/bar")
    assert @cache.exist?("fu/baz")
B
Brian Durand 已提交
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 462 463
  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')
464
    end
465
    assert retval
B
Brian Durand 已提交
466 467
    assert_equal 'bar', @cache.read('foo')
  end
468

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

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

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

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

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

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

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

B
Brian Durand 已提交
519 520 521 522 523 524
  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')
525
    end
B
Brian Durand 已提交
526
  end
527

B
Brian Durand 已提交
528 529 530 531 532 533
  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')
534
    end
B
Brian Durand 已提交
535
  end
536

B
Brian Durand 已提交
537 538 539 540 541 542 543 544 545 546
  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
547

B
Brian Durand 已提交
548 549 550 551 552
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)
553
    @cache_with_pathname = ActiveSupport::Cache.lookup_store(:file_store, Pathname.new(cache_dir), :expires_in => 60)
B
Brian Durand 已提交
554 555 556 557 558 559 560 561 562 563 564 565 566 567 568
  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

569 570 571 572
  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
573 574 575 576 577 578

  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
579 580 581 582 583 584 585 586 587
  
  # 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"
    path = @cache.send(:key_file_path, key)    
    assert path.split('/').all? { |dir_name| dir_name.size <= ActiveSupport::Cache::FileStore::FILENAME_MAX_SIZE}
    assert_equal 'B', File.basename(path)
  end
588

589 590 591 592 593
  # 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/)
594 595
    end
  end
B
Brian Durand 已提交
596
end
597

B
Brian Durand 已提交
598 599
class MemoryStoreTest < ActiveSupport::TestCase
  def setup
600 601
    @record_size = Marshal.dump("aaaaaaaaaa").bytesize
    @cache = ActiveSupport::Cache.lookup_store(:memory_store, :expires_in => 60, :size => @record_size * 10)
B
Brian Durand 已提交
602 603 604 605 606 607 608 609 610 611 612 613 614 615
  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)
616
    @cache.prune(@record_size * 3)
617 618 619 620 621
    assert @cache.exist?(5)
    assert @cache.exist?(4)
    assert !@cache.exist?(3)
    assert @cache.exist?(2)
    assert !@cache.exist?(1)
B
Brian Durand 已提交
622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637
  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")
638 639 640 641 642 643 644 645 646 647 648
    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 已提交
649 650 651 652 653 654
  end

  def test_pruning_is_capped_at_a_max_time
    def @cache.delete_entry (*args)
      sleep(0.01)
      super
655
    end
B
Brian Durand 已提交
656 657 658 659 660 661
    @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)
662 663 664 665 666
    assert @cache.exist?(5)
    assert @cache.exist?(4)
    assert @cache.exist?(3)
    assert @cache.exist?(2)
    assert !@cache.exist?(1)
B
Brian Durand 已提交
667 668
  end
end
669

B
Brian Durand 已提交
670 671 672 673 674 675 676 677 678
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!
      @cache.logger = Logger.new("/dev/null")
679 680
    end

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

    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")
691
    end
692 693 694 695 696 697 698 699
    
    def test_raw_values_with_marshal
      cache = ActiveSupport::Cache.lookup_store(:mem_cache_store, :raw => true)
      cache.clear
      cache.write("foo", Marshal.dump([]))
      assert_equal [], cache.read("foo")      
    end
    
B
Brian Durand 已提交
700 701 702 703 704 705 706
    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
707
    end
708 709 710 711 712 713 714 715 716

    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
717 718
  end
end
719 720 721 722 723 724 725 726 727 728 729

class CacheStoreLoggerTest < ActiveSupport::TestCase
  def setup
    @cache = ActiveSupport::Cache.lookup_store(:memory_store)

    @buffer = StringIO.new
    @cache.logger = Logger.new(@buffer)
  end

  def test_logging
    @cache.fetch('foo') { 'bar' }
730
    assert_present @buffer.string
731 732 733 734
  end

  def test_mute_logging
    @cache.mute { @cache.fetch('foo') { 'bar' } }
735
    assert_blank @buffer.string
736 737
  end
end
B
Brian Durand 已提交
738 739 740 741 742 743 744

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
745
    assert !entry.compressed?
B
Brian Durand 已提交
746 747 748 749 750
    assert_equal 300, entry.expires_in
  end

  def test_expired
    entry = ActiveSupport::Cache::Entry.new("value")
751
    assert !entry.expired?, 'entry not expired'
B
Brian Durand 已提交
752
    entry = ActiveSupport::Cache::Entry.new("value", :expires_in => 60)
753
    assert !entry.expired?, 'entry not expired'
B
Brian Durand 已提交
754 755
    time = Time.now + 61
    Time.stubs(:now).returns(time)
756
    assert entry.expired?, 'entry is expired'
B
Brian Durand 已提交
757 758 759 760 761
  end

  def test_compress_values
    entry = ActiveSupport::Cache::Entry.new("value", :compress => true, :compress_threshold => 1)
    assert_equal "value", entry.value
762
    assert entry.compressed?
B
Brian Durand 已提交
763 764 765 766 767 768
    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
769
    assert_equal "value", Marshal.load(entry.raw_value)
770
    assert !entry.compressed?
B
Brian Durand 已提交
771 772
  end
end