recursiveloader: Support setting timestamp

1 files changed, 19 insertions, 0 deletions

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):
     """