test suite: cut down testing the handling of flagged messages

Don't do entire test archiving runs, just call
archivemail.should_archive().
This commit is contained in:
Nikolaus Schulz 2008-09-30 13:27:33 +02:00
parent 0ca3f6ceee
commit 8ec8a00ffa
1 changed files with 10 additions and 29 deletions

View File

@ -917,53 +917,34 @@ class TestArchiveCopy(TestCaseInTempdir):
super(TestArchiveCopy, self).tearDown() super(TestArchiveCopy, self).tearDown()
class TestArchiveMboxFlagged(TestCaseInTempdir): class TestArchiveMboxFlagged(unittest.TestCase):
"""make sure the 'include_flagged' option works""" """make sure the 'include_flagged' option works"""
def setUp(self): def setUp(self):
super(TestArchiveMboxFlagged, self).setUp()
archivemail.options.include_flagged = 0 archivemail.options.include_flagged = 0
archivemail.options.quiet = 1 archivemail.options.quiet = 1
def testOld(self): def testOld(self):
"""by default, old flagged messages should not be archived""" """by default, old flagged messages should not be archived"""
self.mbox_name = make_mbox(messages=3, hours_old=(24 * 181), \ msg = make_message(default_headers={"X-Status": "F"},
headers={"X-Status":"F"}) hours_old=24*181, wantobj=True)
self.copy_name = tempfile.mkstemp()[1] assert(not archivemail.should_archive(msg))
shutil.copyfile(self.mbox_name, self.copy_name)
archivemail.archive(self.mbox_name)
assertEqualContent(self.mbox_name, self.copy_name)
archive_name = self.mbox_name + "_archive.gz"
assert(not os.path.exists(archive_name))
def testIncludeFlaggedNew(self): def testIncludeFlaggedNew(self):
"""new flagged messages should not be archived with include_flagged""" """new flagged messages should not be archived with include_flagged"""
self.mbox_name = make_mbox(messages=3, hours_old=(24 * 179), \ msg = make_message(default_headers={"X-Status": "F"},
headers={"X-Status":"F"}) hours_old=24*179, wantobj=True)
self.copy_name = tempfile.mkstemp()[1] assert(not archivemail.should_archive(msg))
shutil.copyfile(self.mbox_name, self.copy_name)
archivemail.options.include_flagged = 1
archivemail.archive(self.mbox_name)
assertEqualContent(self.mbox_name, self.copy_name)
archive_name = self.mbox_name + "_archive.gz"
assert(not os.path.exists(archive_name))
def testIncludeFlaggedOld(self): def testIncludeFlaggedOld(self):
"""old flagged messages should be archived with include_flagged""" """old flagged messages should be archived with include_flagged"""
self.mbox_name = make_mbox(messages=3, hours_old=(24 * 181), \
headers={"X-Status":"F"})
self.copy_name = tempfile.mkstemp()[1]
shutil.copyfile(self.mbox_name, self.copy_name)
archivemail.options.include_flagged = 1 archivemail.options.include_flagged = 1
archivemail.archive(self.mbox_name) msg = make_message(default_headers={"X-Status": "F"},
assert(os.path.exists(self.mbox_name)) hours_old=24*181, wantobj=True)
self.assertEqual(os.path.getsize(self.mbox_name), 0) assert(archivemail.should_archive(msg))
archive_name = self.mbox_name + "_archive.gz"
assertEqualContent(archive_name, self.copy_name, zipfirst=True)
def tearDown(self): def tearDown(self):
archivemail.options.include_flagged = 0 archivemail.options.include_flagged = 0
archivemail.options.quiet = 0 archivemail.options.quiet = 0
super(TestArchiveMboxFlagged, self).tearDown()
class TestArchiveMboxOutputDir(TestCaseInTempdir): class TestArchiveMboxOutputDir(TestCaseInTempdir):