mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-14 20:12:19 -05:00
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class TestMemoryCache(unittest.TestCase):
|
||
|
|
|
||
|
|
initstring = 'memory://'
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
from shove.cache.memory import MemoryCache
|
||
|
|
self.cache = MemoryCache(self.initstring)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
self.cache = None
|
||
|
|
|
||
|
|
def test_getitem(self):
|
||
|
|
self.cache['test'] = 'test'
|
||
|
|
self.assertEqual(self.cache['test'], 'test')
|
||
|
|
|
||
|
|
def test_setitem(self):
|
||
|
|
self.cache['test'] = 'test'
|
||
|
|
self.assertEqual(self.cache['test'], 'test')
|
||
|
|
|
||
|
|
def test_delitem(self):
|
||
|
|
self.cache['test'] = 'test'
|
||
|
|
del self.cache['test']
|
||
|
|
self.assertEqual('test' in self.cache, False)
|
||
|
|
|
||
|
|
def test_get(self):
|
||
|
|
self.assertEqual(self.cache.get('min'), None)
|
||
|
|
|
||
|
|
def test_timeout(self):
|
||
|
|
import time
|
||
|
|
from shove.cache.memory import MemoryCache
|
||
|
|
cache = MemoryCache(self.initstring, timeout=1)
|
||
|
|
cache['test'] = 'test'
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
def tmp():
|
||
|
|
cache['test']
|
||
|
|
self.assertRaises(KeyError, tmp)
|
||
|
|
|
||
|
|
def test_cull(self):
|
||
|
|
from shove.cache.memory import MemoryCache
|
||
|
|
cache = MemoryCache(self.initstring, max_entries=1)
|
||
|
|
cache['test'] = 'test'
|
||
|
|
cache['test2'] = 'test'
|
||
|
|
cache['test2'] = 'test'
|
||
|
|
self.assertEquals(len(cache), 1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|