class MemcachedStorageTest(unittest.TestCase): def setUp(self): self.storage = MemcachedStorage('localhost:11211') if not self.storage.storage.set('test', 'test'): raise Exception('memcached is down') def tearDown(self): memcached = self.storage.storage memcached.delete('test') memcached.delete('key') memcached.disconnect_all() def test_set(self): '`MemcachedStorage` set method' self.assertEqual(self.storage.set('key', 'value'), True) self.assertEqual(self.storage.set('key', 'value1'), True) def test_get(self): '`MemcachedStorage` get method' self.assertEqual(self.storage.get('key'), None) self.assertEqual(self.storage.get('key', 'default'), 'default') self.storage.set('key', 'value') self.assertEqual(self.storage.get('key'), 'value') def test_delete(self): '`MemcachedStorage` delete method' self.assertEqual(self.storage.delete('key'), True) self.storage.set('key', 'value') self.assertEqual(self.storage.delete('key'), True) self.assertEqual(self.storage.get('key'), None)
class MemcachedStorageTest(unittest.TestCase): def setUp(self): with mock.patch.object(memcache, 'Client', Client): self.storage = MemcachedStorage('localhost:11211') if not self.storage.storage.set('test', 'test'): raise Exception('memcached is down') def tearDown(self): memcached = self.storage.storage memcached.delete('test') memcached.delete('key') memcached.disconnect_all() def test_set(self): '`MemcachedStorage` set method' self.assertEqual(self.storage.set('key', 'value'), True) self.assertEqual(self.storage.set('key', 'value1'), True) def test_get(self): '`MemcachedStorage` get method' self.assertEqual(self.storage.get('key'), None) self.assertEqual(self.storage.get('key', 'default'), 'default') self.storage.set('key', 'value') self.assertEqual(self.storage.get('key'), 'value') def test_delete(self): '`MemcachedStorage` delete method' self.storage.delete('key') self.storage.set('key', 'value') self.assertEqual(self.storage.delete('key'), True) self.assertEqual(self.storage.get('key'), None)