From a1641450ede4c7cfb4991d31b6232282c5673f6b Mon Sep 17 00:00:00 2001 From: Nikolaus Schulz Date: Fri, 2 Nov 2007 18:21:21 +0000 Subject: Added simple unittest for the --copy option. Mbox only. :-/ --- test_archivemail.py | 96 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 96 insertions(+) diff --git a/test_archivemail.py b/test_archivemail.py index 17c5846..55d229c 100755 --- a/test_archivemail.py +++ b/test_archivemail.py @@ -1039,6 +1039,102 @@ class TestArchiveDelete(TestCaseInTempdir): super(TestArchiveDelete, self).tearDown() +class TestArchiveCopy(TestCaseInTempdir): + """make sure the 'copy' option works""" + old_mbox = None + new_mbox = None + mbox_backup_name = None + mbox_name = None + + def setUp(self): + super(TestArchiveCopy, self).setUp() + archivemail.options.quiet = 1 + archivemail.options.copy_old_mail = 1 + + def testNew(self): + """archiving a new mailbox with the 'copy' option""" + for execute in ("package", "system"): + self.setUp() + self.mbox_name = make_mbox(messages=3, hours_old=(24 * 179)) + self.mbox_backup_name = tempfile.mkstemp()[1] + shutil.copyfile(self.mbox_name, self.mbox_backup_name) + if execute == "package": + archivemail.archive(self.mbox_name) + elif execute == "system": + run = "./archivemail.py --copy --quiet %s" % self.mbox_name + self.assertEqual(os.system(run), 0) + else: + sys.exit(1) + assert(os.path.exists(self.mbox_name)) + # mbox must not have changed: + assert(filecmp.cmp(self.mbox_name, self.mbox_backup_name, shallow=0)) + archive_name = self.mbox_name + "_archive.gz" + # There is no archive. + assert(not os.path.exists(archive_name)) + self.tearDown() + + def testMixed(self): + """archiving a mixed mailbox with the 'copy' option""" + for execute in ("package", "system"): + self.setUp() + self.new_mbox = make_mbox(messages=3, hours_old=(24 * 179)) + self.old_mbox = make_mbox(messages=3, hours_old=(24 * 181)) + self.mbox_name = tempfile.mkstemp()[1] + shutil.copyfile(self.new_mbox, self.mbox_name) + append_file(self.old_mbox, self.mbox_name) + self.mbox_backup_name = tempfile.mkstemp()[1] + shutil.copyfile(self.mbox_name, self.mbox_backup_name) + if execute == "package": + archivemail.archive(self.mbox_name) + elif execute == "system": + run = "./archivemail.py --copy --quiet %s" % self.mbox_name + self.assertEqual(os.system(run), 0) + else: + sys.exit(1) + assert(os.path.exists(self.mbox_name)) + # mbox must not have changed: + assert(filecmp.cmp(self.mbox_backup_name, self.mbox_name, shallow=0)) + # archive has the old messages. + archive_name = self.mbox_name + "_archive.gz" + assert(os.path.exists(archive_name)) + self.assertEqual(os.system("gzip -d %s" % archive_name), 0) + archive_name = self.mbox_name + "_archive" + assert(os.path.exists(archive_name)) + assert(filecmp.cmp(archive_name, self.old_mbox, shallow=0)) + self.tearDown() + + def testOld(self): + """archiving an old mailbox with the 'copy' option""" + for execute in ("package", "system"): + self.setUp() + self.mbox_name = make_mbox(messages=3, hours_old=(24 * 181)) + self.mbox_backup_name = tempfile.mkstemp()[1] + shutil.copyfile(self.mbox_name, self.mbox_backup_name) + if execute == "package": + archivemail.archive(self.mbox_name) + elif execute == "system": + run = "./archivemail.py --copy --quiet %s" % self.mbox_name + self.assertEqual(os.system(run), 0) + else: + sys.exit(1) + assert(os.path.exists(self.mbox_name)) + # mbox must not have changed: + assert(filecmp.cmp(self.mbox_backup_name, self.mbox_name, shallow=0)) + # archive has the old messages. + archive_name = self.mbox_name + "_archive.gz" + assert(os.path.exists(archive_name)) + self.assertEqual(os.system("gzip -d %s" % archive_name), 0) + archive_name = self.mbox_name + "_archive" + assert(os.path.exists(archive_name)) + assert(filecmp.cmp(archive_name, self.mbox_name, shallow=0)) + self.tearDown() + + def tearDown(self): + archivemail.options.copy_old_mail = 0 + archivemail.options.quiet = 0 + super(TestArchiveCopy, self).tearDown() + + class TestArchiveMboxFlagged(TestCaseInTempdir): """make sure the 'include_flagged' option works""" def setUp(self): -- cgit v1.2.3