summary refs log tree commit diff
path: root/tests/util
diff options
context:
space:
mode:
authorMark Haines <mark.haines@matrix.org>2015-02-11 14:52:23 +0000
committerMark Haines <mark.haines@matrix.org>2015-02-11 14:52:23 +0000
commitd8324d5a2b9ffc6f3a426ecd240f6c2460630025 (patch)
tree4f305fb42994ea4f54ce6a1ddcce2dd32abf0677 /tests/util
parentMerge pull request #63 from matrix-org/homeserver_test_setup (diff)
downloadsynapse-d8324d5a2b9ffc6f3a426ecd240f6c2460630025.tar.xz
Add a lru cache class
Diffstat (limited to 'tests/util')
-rw-r--r--tests/util/test_lrucache.py56
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/util/test_lrucache.py b/tests/util/test_lrucache.py
new file mode 100644

index 0000000000..ab934bf928 --- /dev/null +++ b/tests/util/test_lrucache.py
@@ -0,0 +1,56 @@ +# -*- coding: utf-8 -*- +# Copyright 2015 OpenMarket Ltd +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +from .. import unittest + +from synapse.util.lrucache import LruCache + +class LruCacheTestCase(unittest.TestCase): + + 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) + + def test_pop(self): + cache = LruCache(1) + cache["key"] = 1 + self.assertEquals(cache.pop("key"), 1) + self.assertEquals(cache.pop("key"), None) + +