From 0ee512648668419a8237fdd397e62a66cfa009c2 Mon Sep 17 00:00:00 2001 From: Michał Górny Date: Thu, 2 Nov 2017 13:40:24 +0100 Subject: recursiveloader: Support setting timestamp --- tests/test_recursiveloader.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'tests/test_recursiveloader.py') diff --git a/tests/test_recursiveloader.py b/tests/test_recursiveloader.py index 40aeb2b..2835b7f 100644 --- a/tests/test_recursiveloader.py +++ b/tests/test_recursiveloader.py @@ -112,6 +112,17 @@ DATA test 0 MD5 d41d8cd98f00b204e9800998ecf8427e self.assertEqual(m.find_timestamp().ts, datetime.datetime(2017, 1, 1, 1, 1, 1)) + def test_set_timestamp(self): + m = gemato.recursiveloader.ManifestRecursiveLoader( + os.path.join(self.dir, 'Manifest')) + m.set_timestamp(datetime.datetime(2010, 7, 7, 7, 7, 7)) + self.assertEqual(m.find_timestamp().ts, + datetime.datetime(2010, 7, 7, 7, 7, 7)) + self.assertEqual( + len([x for x in m.loaded_manifests['Manifest'].entries + if x.tag == 'TIMESTAMP']), + 1) + def test_find_path_entry(self): m = gemato.recursiveloader.ManifestRecursiveLoader( os.path.join(self.dir, 'Manifest')) @@ -940,6 +951,14 @@ DATA test 0 MD5 d41d8cd98f00b204e9800998ecf8427e self.assertTrue(os.path.exists( os.path.join(self.dir, 'Manifest'))) + def test_set_timestamp(self): + m = gemato.recursiveloader.ManifestRecursiveLoader( + os.path.join(self.dir, 'Manifest')) + self.assertIsNone(m.find_timestamp()) + m.set_timestamp(datetime.datetime(2010, 7, 7, 7, 7, 7)) + self.assertEqual(m.find_timestamp().ts, + datetime.datetime(2010, 7, 7, 7, 7, 7)) + class DuplicateManifestFileEntryTest(TempDirTestCase): """ -- cgit v1.2.3