class TestLocalStorage(unittest.TestCase):
def setUp(self):
self.conf = get_test_config()
self.ls = LocalStorage(self.conf, self.conf.tempdir("localstorage"))
def tearDown(self):
self.conf.cleanup()
def testLocalStorage(self):
rsi = self.ls.get_stored_info(TEST_FILES_TGT)
self.assertEqual(len(rsi.all_trees), 0)
self.ls.save_tree(TEST_FILES_TGT, tree.Tree())
rsi = self.ls.get_stored_info(TEST_FILES_TGT)
self.assertEqual(len(rsi.all_trees), 1)
# TODO: load the tree again
def testLocalStorageKey(self):
newt = tree.Tree()
self.ls.save_tree(TEST_FILES_TGT, newt)
trees = self.ls.get_stored_info(TEST_FILES_TGT).all_trees
for t in trees:
self.assertEqual(t.get_metadata('pf:target'), TEST_FILES_TGT)
self.assertEqual(len(t.get_metadata('pf:saved_dt')), len(TREE_DT_FMT)+2)
with self.assertRaises(KeyError):
t.get_metadata('doesnt_exist')
sio = StringIO()
tree.save_tree(newt, sio)
self.assertEqual(len(t.get_contents_as_string()), len(sio.getvalue()))
def setUp(self):
self.conf = get_test_config()
self.ls = LocalStorage(self.conf, self.conf.tempdir("localstorage"))
