123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263 |
- from mock import Mock
- from synapse.util.caches.lrucache import LruCache
- from synapse.util.caches.treecache import TreeCache
- from tests import unittest
- from tests.unittest import override_config
- class LruCacheTestCase(unittest.HomeserverTestCase):
- def test_get_set(self):
- cache = LruCache(1)
- cache["key"] = "value"
- self.assertEquals(cache.get("key"), "value")
- self.assertEquals(cache["key"], "value")
- def test_eviction(self):
- cache = LruCache(2)
- cache[1] = 1
- cache[2] = 2
- self.assertEquals(cache.get(1), 1)
- self.assertEquals(cache.get(2), 2)
- cache[3] = 3
- self.assertEquals(cache.get(1), None)
- self.assertEquals(cache.get(2), 2)
- self.assertEquals(cache.get(3), 3)
- def test_setdefault(self):
- cache = LruCache(1)
- self.assertEquals(cache.setdefault("key", 1), 1)
- self.assertEquals(cache.get("key"), 1)
- self.assertEquals(cache.setdefault("key", 2), 1)
- self.assertEquals(cache.get("key"), 1)
- cache["key"] = 2
- self.assertEquals(cache.get("key"), 2)
- def test_pop(self):
- cache = LruCache(1)
- cache["key"] = 1
- self.assertEquals(cache.pop("key"), 1)
- self.assertEquals(cache.pop("key"), None)
- def test_del_multi(self):
- cache = LruCache(4, keylen=2, cache_type=TreeCache)
- cache[("animal", "cat")] = "mew"
- cache[("animal", "dog")] = "woof"
- cache[("vehicles", "car")] = "vroom"
- cache[("vehicles", "train")] = "chuff"
- self.assertEquals(len(cache), 4)
- self.assertEquals(cache.get(("animal", "cat")), "mew")
- self.assertEquals(cache.get(("vehicles", "car")), "vroom")
- cache.del_multi(("animal",))
- self.assertEquals(len(cache), 2)
- self.assertEquals(cache.get(("animal", "cat")), None)
- self.assertEquals(cache.get(("animal", "dog")), None)
- self.assertEquals(cache.get(("vehicles", "car")), "vroom")
- self.assertEquals(cache.get(("vehicles", "train")), "chuff")
-
- def test_clear(self):
- cache = LruCache(1)
- cache["key"] = 1
- cache.clear()
- self.assertEquals(len(cache), 0)
- @override_config({"caches": {"per_cache_factors": {"mycache": 10}}})
- def test_special_size(self):
- cache = LruCache(10, "mycache")
- self.assertEqual(cache.max_size, 100)
- class LruCacheCallbacksTestCase(unittest.HomeserverTestCase):
- def test_get(self):
- m = Mock()
- cache = LruCache(1)
- cache.set("key", "value")
- self.assertFalse(m.called)
- cache.get("key", callbacks=[m])
- self.assertFalse(m.called)
- cache.get("key", "value")
- self.assertFalse(m.called)
- cache.set("key", "value2")
- self.assertEquals(m.call_count, 1)
- cache.set("key", "value")
- self.assertEquals(m.call_count, 1)
- def test_multi_get(self):
- m = Mock()
- cache = LruCache(1)
- cache.set("key", "value")
- self.assertFalse(m.called)
- cache.get("key", callbacks=[m])
- self.assertFalse(m.called)
- cache.get("key", callbacks=[m])
- self.assertFalse(m.called)
- cache.set("key", "value2")
- self.assertEquals(m.call_count, 1)
- cache.set("key", "value")
- self.assertEquals(m.call_count, 1)
- def test_set(self):
- m = Mock()
- cache = LruCache(1)
- cache.set("key", "value", callbacks=[m])
- self.assertFalse(m.called)
- cache.set("key", "value")
- self.assertFalse(m.called)
- cache.set("key", "value2")
- self.assertEquals(m.call_count, 1)
- cache.set("key", "value")
- self.assertEquals(m.call_count, 1)
- def test_pop(self):
- m = Mock()
- cache = LruCache(1)
- cache.set("key", "value", callbacks=[m])
- self.assertFalse(m.called)
- cache.pop("key")
- self.assertEquals(m.call_count, 1)
- cache.set("key", "value")
- self.assertEquals(m.call_count, 1)
- cache.pop("key")
- self.assertEquals(m.call_count, 1)
- def test_del_multi(self):
- m1 = Mock()
- m2 = Mock()
- m3 = Mock()
- m4 = Mock()
- cache = LruCache(4, keylen=2, cache_type=TreeCache)
- cache.set(("a", "1"), "value", callbacks=[m1])
- cache.set(("a", "2"), "value", callbacks=[m2])
- cache.set(("b", "1"), "value", callbacks=[m3])
- cache.set(("b", "2"), "value", callbacks=[m4])
- self.assertEquals(m1.call_count, 0)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 0)
- self.assertEquals(m4.call_count, 0)
- cache.del_multi(("a",))
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 1)
- self.assertEquals(m3.call_count, 0)
- self.assertEquals(m4.call_count, 0)
- def test_clear(self):
- m1 = Mock()
- m2 = Mock()
- cache = LruCache(5)
- cache.set("key1", "value", callbacks=[m1])
- cache.set("key2", "value", callbacks=[m2])
- self.assertEquals(m1.call_count, 0)
- self.assertEquals(m2.call_count, 0)
- cache.clear()
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 1)
- def test_eviction(self):
- m1 = Mock(name="m1")
- m2 = Mock(name="m2")
- m3 = Mock(name="m3")
- cache = LruCache(2)
- cache.set("key1", "value", callbacks=[m1])
- cache.set("key2", "value", callbacks=[m2])
- self.assertEquals(m1.call_count, 0)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 0)
- cache.set("key3", "value", callbacks=[m3])
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 0)
- cache.set("key3", "value")
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 0)
- cache.get("key2")
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 0)
- cache.set("key1", "value", callbacks=[m1])
- self.assertEquals(m1.call_count, 1)
- self.assertEquals(m2.call_count, 0)
- self.assertEquals(m3.call_count, 1)
- class LruCacheSizedTestCase(unittest.HomeserverTestCase):
- def test_evict(self):
- cache = LruCache(5, size_callback=len)
- cache["key1"] = [0]
- cache["key2"] = [1, 2]
- cache["key3"] = [3]
- cache["key4"] = [4]
- self.assertEquals(cache["key1"], [0])
- self.assertEquals(cache["key2"], [1, 2])
- self.assertEquals(cache["key3"], [3])
- self.assertEquals(cache["key4"], [4])
- self.assertEquals(len(cache), 5)
- cache["key5"] = [5, 6]
- self.assertEquals(len(cache), 4)
- self.assertEquals(cache.get("key1"), None)
- self.assertEquals(cache.get("key2"), None)
- self.assertEquals(cache["key3"], [3])
- self.assertEquals(cache["key4"], [4])
- self.assertEquals(cache["key5"], [5, 6])
|