class ShelveCacheTestCase(unittest.TestCase):
def setUp(self):
self.cache_dir = tempfile.mkdtemp()
self.cache_path = os.path.join(self.cache_dir, 'shelf')
self.cache = ShelveCache(self.cache_path)
def tearDown(self):
self.cache.cache.close()
try:
os.remove(self.cache_path)
except OSError:
pass
try:
os.rmdir(self.cache_dir)
except OSError:
pass
def test_cache(self):
self.cache.put('foo', 'bar', 3600)
self.assertEqual(self.cache.get('foo'), 'bar')
def test_expire(self):
self.cache.put('baz', 'qux', -1)
self.assertEqual(self.cache.get('baz'), None)
class ShelveCacheTestCase(unittest.TestCase):
def setUp(self):
self.cache_dir = tempfile.mkdtemp()
self.cache_path = os.path.join(self.cache_dir, "shelf")
self.cache = ShelveCache(self.cache_path)
def tearDown(self):
self.cache.cache.close()
os.remove(self.cache_path)
os.rmdir(self.cache_dir)
def test_cache(self):
self.cache.put("foo", "bar", 3600)
self.assertEqual(self.cache.get("foo"), "bar")
def test_expire(self):
self.cache.put("baz", "qux", -1)
self.assertEqual(self.cache.get("baz"), None)
