def setUp(self): def fakemsg(body=False, headers=False): msg = [] if headers: msg.append("Header: Fake") if body: msg.append("\r\n") if body: msg.append("Fake Body") return "\r\n".join(msg) self.msg = DynamicIMAPMessage(fakemsg)
class DynamicIMAPMessageTest(unittest.TestCase): def setUp(self): def fakemsg(body=False, headers=False): msg = [] if headers: msg.append("Header: Fake") if body: msg.append("\r\n") if body: msg.append("Fake Body") return "\r\n".join(msg) self.msg = DynamicIMAPMessage(fakemsg) def testDate(self): date = imaplib.Time2Internaldate(time.time())[1:-1] self.assertEqual(self.msg.date, date) def testLoad(self): self.assertEqual(self.msg.as_string(), "Header: Fake\r\n\r\nFake Body")
class DynamicIMAPMessageTest (unittest.TestCase) : def setUp(self): def fakemsg(body=False, headers=False): msg = [] if headers: msg.append("Header: Fake") if body: msg.append("\r\n") if body: msg.append("Fake Body") return "\r\n".join(msg) self.msg = DynamicIMAPMessage(fakemsg) def testDate(self): date = imaplib.Time2Internaldate(time.time())[1:-1] self.assertEqual(self.msg.date, date) def testLoad(self): self.assertEqual(self.msg.as_string(), "Header: Fake\r\n\r\nFake Body") class IMAPFileMessageTest (unittest.TestCase) : def setUp(self): self.msg = IMAPFileMessage("filename", "directory") def testID(self): self.assertEqual(self.msg.id, "filename") def testDate(self): date = imaplib.Time2Internaldate(time.time())[1:-1] self.assertEqual(self.msg.date, date) class IMAPFileMessageFactoryTest (unittest.TestCase) : def testCreateNoContent(self): factory = IMAPFileMessageFactory() msg = factory.create("key", "directory") self.assertEqual(msg.id, key) self.assert_(isinstance(msg, type(IMAPFileMessage()))) def suite(): suite = unittest.TestSuite() for cls in (IMAPMessageTest, DynamicIMAPMessageTest, IMAPFileMessageTest, ): suite.addTest(unittest.makeSuite(cls)) return suite if __name__=='__main__': def runTestServer(): import asyncore asyncore.loop() TestListener() _thread.start_new_thread(runTestServer, ()) sb_test_support.unittest_main(argv=sys.argv + ['suite']) if __name__=='__main__': def runTestServer(): import asyncore asyncore.loop() TestListener() thread.start_new_thread(runTestServer, ()) sb_test_support.unittest_main(argv=sys.argv + ['suite'])