def setUp(self):
self.storage = Storage(config=TConfig(
{'db.backend': 'pushsrv_ws.storage.memcache_sql.Storage',
'db.type': 'sqlite',
'db.db': ':memory:'}))
class TestStorage(unittest2.TestCase):
def setUp(self):
self.storage = Storage(config=TConfig(
{'db.backend': 'pushsrv_ws.storage.memcache_sql.Storage',
'db.type': 'sqlite',
'db.db': ':memory:'}))
def tearDown(self):
self.storage.purge()
del self.storage
def load(self, data=[]):
if not len(data):
data = [{'channelID': 'aaa', 'uaid': '111', 'version': 1,
'last_accessed': 10},
{'channelID': 'bbb', 'uaid': '111', 'version': 1,
'last_accessed': 20},
{'channelID': 'exp', 'uaid': '111', 'version': 0,
'state': 0, 'last_accessed': 20},
{'channelID': 'ccc', 'uaid': '222', 'version': 2}]
self.storage._load(data)
def test_update_child(self):
self.load()
self.storage.update_channel('111.aaa', 2, FakeLogger())
rec = self.storage._get_record('111.aaa')
self.assertEqual(int(rec.get('version')), 2)
def test_register_appids(self):
self.load()
self.storage.register_appid('444', 'ddd', FakeLogger())
rec = self.storage._get_record('444.ddd')
self.assertEqual(rec.get('uaid'), '444')
def test_delete_appid(self):
self.load()
self.storage.delete_appid('111', 'aaa', FakeLogger())
rec = self.storage._get_record('111.aaa')
assert (rec is not None)
self.assertEqual(rec.get('state'), 0)
def test_get_updates(self):
self.load()
data = self.storage.get_updates('111', last_accessed=None,
logger=FakeLogger())
self.assertEqual(data.get('expired')[0], 'exp')
self.assertEqual(len(data.get('updates')), 2)
data = self.storage.get_updates('111', last_accessed=20,
logger=FakeLogger())
self.assertEqual(len(data.get('updates')), 1)
def test_reload_data(self):
data = self.storage.reload_data('111',
[{'channelID': 'aaa', 'version': '5'},
{'channelID': 'bbb', 'version': '6'}],
FakeLogger())
self.assertEqual(data, 'aaa,bbb')
#TODO: check that subsequent updates are rejected.
with self.assertRaises(StorageException):
self.storage.reload_data('111',
[{'channelID': 'ccc', 'version': '7'}],
FakeLogger())
