def setUp(self):
self.storage = Storage(config=TConfig({"db.type": "sqlite", "db.db": ":memory:"}))
class TestStorage(unittest2.TestCase):
def setUp(self):
self.storage = Storage(config=TConfig({"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(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())