Add tests
This commit is contained in:
parent
fb7299800f
commit
4fce59f274
|
@ -19,6 +19,7 @@ from .. import unittest
|
||||||
from synapse.util.caches.lrucache import LruCache
|
from synapse.util.caches.lrucache import LruCache
|
||||||
from synapse.util.caches.treecache import TreeCache
|
from synapse.util.caches.treecache import TreeCache
|
||||||
|
|
||||||
|
|
||||||
class LruCacheTestCase(unittest.TestCase):
|
class LruCacheTestCase(unittest.TestCase):
|
||||||
|
|
||||||
def test_get_set(self):
|
def test_get_set(self):
|
||||||
|
@ -72,3 +73,9 @@ class LruCacheTestCase(unittest.TestCase):
|
||||||
self.assertEquals(cache.get(("vehicles", "car")), "vroom")
|
self.assertEquals(cache.get(("vehicles", "car")), "vroom")
|
||||||
self.assertEquals(cache.get(("vehicles", "train")), "chuff")
|
self.assertEquals(cache.get(("vehicles", "train")), "chuff")
|
||||||
# Man from del_multi say "Yes".
|
# Man from del_multi say "Yes".
|
||||||
|
|
||||||
|
def test_clear(self):
|
||||||
|
cache = LruCache(1)
|
||||||
|
cache["key"] = 1
|
||||||
|
cache.clear()
|
||||||
|
self.assertEquals(len(cache), 0)
|
||||||
|
|
|
@ -25,6 +25,7 @@ class TreeCacheTestCase(unittest.TestCase):
|
||||||
cache[("b",)] = "B"
|
cache[("b",)] = "B"
|
||||||
self.assertEquals(cache.get(("a",)), "A")
|
self.assertEquals(cache.get(("a",)), "A")
|
||||||
self.assertEquals(cache.get(("b",)), "B")
|
self.assertEquals(cache.get(("b",)), "B")
|
||||||
|
self.assertEquals(len(cache), 2)
|
||||||
|
|
||||||
def test_pop_onelevel(self):
|
def test_pop_onelevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
|
@ -33,6 +34,7 @@ class TreeCacheTestCase(unittest.TestCase):
|
||||||
self.assertEquals(cache.pop(("a",)), "A")
|
self.assertEquals(cache.pop(("a",)), "A")
|
||||||
self.assertEquals(cache.pop(("a",)), None)
|
self.assertEquals(cache.pop(("a",)), None)
|
||||||
self.assertEquals(cache.get(("b",)), "B")
|
self.assertEquals(cache.get(("b",)), "B")
|
||||||
|
self.assertEquals(len(cache), 1)
|
||||||
|
|
||||||
def test_get_set_twolevel(self):
|
def test_get_set_twolevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
|
@ -42,6 +44,7 @@ class TreeCacheTestCase(unittest.TestCase):
|
||||||
self.assertEquals(cache.get(("a", "a")), "AA")
|
self.assertEquals(cache.get(("a", "a")), "AA")
|
||||||
self.assertEquals(cache.get(("a", "b")), "AB")
|
self.assertEquals(cache.get(("a", "b")), "AB")
|
||||||
self.assertEquals(cache.get(("b", "a")), "BA")
|
self.assertEquals(cache.get(("b", "a")), "BA")
|
||||||
|
self.assertEquals(len(cache), 3)
|
||||||
|
|
||||||
def test_pop_twolevel(self):
|
def test_pop_twolevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
|
@ -53,6 +56,7 @@ class TreeCacheTestCase(unittest.TestCase):
|
||||||
self.assertEquals(cache.get(("a", "b")), "AB")
|
self.assertEquals(cache.get(("a", "b")), "AB")
|
||||||
self.assertEquals(cache.pop(("b", "a")), "BA")
|
self.assertEquals(cache.pop(("b", "a")), "BA")
|
||||||
self.assertEquals(cache.pop(("b", "a")), None)
|
self.assertEquals(cache.pop(("b", "a")), None)
|
||||||
|
self.assertEquals(len(cache), 1)
|
||||||
|
|
||||||
def test_pop_mixedlevel(self):
|
def test_pop_mixedlevel(self):
|
||||||
cache = TreeCache()
|
cache = TreeCache()
|
||||||
|
@ -64,3 +68,11 @@ class TreeCacheTestCase(unittest.TestCase):
|
||||||
self.assertEquals(cache.get(("a", "a")), None)
|
self.assertEquals(cache.get(("a", "a")), None)
|
||||||
self.assertEquals(cache.get(("a", "b")), None)
|
self.assertEquals(cache.get(("a", "b")), None)
|
||||||
self.assertEquals(cache.get(("b", "a")), "BA")
|
self.assertEquals(cache.get(("b", "a")), "BA")
|
||||||
|
self.assertEquals(len(cache), 1)
|
||||||
|
|
||||||
|
def test_clear(self):
|
||||||
|
cache = TreeCache()
|
||||||
|
cache[("a",)] = "A"
|
||||||
|
cache[("b",)] = "B"
|
||||||
|
cache.clear()
|
||||||
|
self.assertEquals(len(cache), 0)
|
||||||
|
|
Loading…
Reference in New Issue