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)