def init_mailer_from_name(self, name): element = self.cfgtree.find('mailer[@name="%s"]' % name) if element is None: return None mail = Mail() mail.set_sender(element.find('from').text) mail.set_to(element.find('to').text) mail.set_subject(element.find('subject').text) mail.set_body(element.find('body').text) server = element.find('server').text return Mailer(server, mail)
class MailTest(unittest.TestCase): CORRELATE = Mail def setUp(self): self.mail = Mail(sender='test') def testConstructor(self): mail = Mail() self.assertNotEqual(mail.get_sender(), None) self.assertNotEqual(mail.get_sender(), '') user = getuser() self.assertTrue(mail.get_sender().startswith(user + '@')) def testSetFromTemplateString(self): self.mail.set_from_template_string(template) tmpl = self.mail.get_smtp_mail().strip() self.assertEqual(tmpl, smtp) head = self.mail.get_smtp_header().strip() self.assertTrue(tmpl.startswith(head)) def testSetSender(self): self.assertEqual(self.mail.get_sender(), 'test') self.mail.set_sender('test2') self.assertEqual(self.mail.get_sender(), 'test2') def testGetSender(self): pass # see testSetSender() def checkSetAddr(self, set_method, get_method): set_method('test') self.assertEqual(get_method(), ['test']) set_method(['test1', 'test2']) self.assertEqual(get_method(), ['test1', 'test2']) set_method('test1, test2') self.assertEqual(get_method(), ['test1', 'test2']) def checkAddAddr(self, add_method, get_method): add_method(['test1', 'test2']) self.assertEqual(get_method(), ['test1', 'test2']) add_method('test3, test4') self.assertEqual(get_method(), ['test1', 'test2', 'test3', 'test4']) def testSetTo(self): self.checkSetAddr(self.mail.set_to, self.mail.get_to) def testAddTo(self): self.checkAddAddr(self.mail.add_to, self.mail.get_to) def testGetTo(self): pass # see testSetTo() def testSetCc(self): self.checkSetAddr(self.mail.set_cc, self.mail.get_cc) def testAddCc(self): self.checkAddAddr(self.mail.add_cc, self.mail.get_cc) def testGetCc(self): pass # see testSetCc() def testSetBcc(self): self.checkSetAddr(self.mail.set_bcc, self.mail.get_bcc) def testAddBcc(self): self.checkAddAddr(self.mail.add_bcc, self.mail.get_bcc) def testGetBcc(self): pass # see testSetBcc() def testGetReceipients(self): self.mail.set_to('test1') self.mail.set_cc('test2') self.mail.set_bcc('test3') self.assertEqual(self.mail.get_receipients(), ['test1', 'test2', 'test3']) def testSetSubject(self): self.assertEqual(self.mail.get_subject(), '') self.mail.set_subject('test') self.assertEqual(self.mail.get_subject(), 'test') def testGetSubject(self): pass # see testSetSubject() def testSetBody(self): self.assertEqual(self.mail.get_body(), '') self.mail.set_body('test') self.assertEqual(self.mail.get_body(), 'test') def testGetBody(self): pass # see testSetBody() def testAddAttachment(self): self.assertEqual(self.mail.get_attachments(), []) self.mail.add_attachment('foo') self.assertEqual(self.mail.get_attachments(), ['foo']) self.mail.add_attachment('bar') self.assertEqual(self.mail.get_attachments(), ['foo', 'bar']) def testGetAttachments(self): self.testAddAttachment() def testGetSmtpHeader(self): pass # see testSetFromTemplateString() def testGetSmtpMail(self): pass # see testSetFromTemplateString()
class MailTest(unittest.TestCase): CORRELATE = Mail def setUp(self): self.mail = Mail(sender = 'test') def testConstructor(self): mail = Mail() self.failIfEqual(mail.get_sender(), None) self.failIfEqual(mail.get_sender(), '') user = getuser() self.assert_(mail.get_sender().startswith(user + '@')) def testSetFromTemplateString(self): self.mail.set_from_template_string(template) tmpl = self.mail.get_smtp_mail().strip() self.assertEqual(tmpl, smtp) head = self.mail.get_smtp_header().strip() self.assert_(tmpl.startswith(head)) def testSetSender(self): self.assertEqual(self.mail.get_sender(), 'test') self.mail.set_sender('test2') self.assertEqual(self.mail.get_sender(), 'test2') def testGetSender(self): pass # see testSetSender() def checkSetAddr(self, set_method, get_method): set_method('test') self.assertEqual(get_method(), ['test']) set_method(['test1', 'test2']) self.assertEqual(get_method(), ['test1', 'test2']) set_method('test1, test2') self.assertEqual(get_method(), ['test1', 'test2']) def checkAddAddr(self, add_method, get_method): add_method(['test1', 'test2']) self.assertEqual(get_method(), ['test1', 'test2']) add_method('test3, test4') self.assertEqual(get_method(), ['test1', 'test2', 'test3', 'test4']) def testSetTo(self): self.checkSetAddr(self.mail.set_to, self.mail.get_to) def testAddTo(self): self.checkAddAddr(self.mail.add_to, self.mail.get_to) def testGetTo(self): pass # see testSetTo() def testSetCc(self): self.checkSetAddr(self.mail.set_cc, self.mail.get_cc) def testAddCc(self): self.checkAddAddr(self.mail.add_cc, self.mail.get_cc) def testGetCc(self): pass # see testSetCc() def testSetBcc(self): self.checkSetAddr(self.mail.set_bcc, self.mail.get_bcc) def testAddBcc(self): self.checkAddAddr(self.mail.add_bcc, self.mail.get_bcc) def testGetBcc(self): pass # see testSetBcc() def testGetReceipients(self): self.mail.set_to('test1') self.mail.set_cc('test2') self.mail.set_bcc('test3') self.assertEqual(self.mail.get_receipients(), ['test1', 'test2', 'test3']) def testSetSubject(self): self.assertEqual(self.mail.get_subject(), '') self.mail.set_subject('test') self.assertEqual(self.mail.get_subject(), 'test') def testGetSubject(self): pass # see testSetSubject() def testSetBody(self): self.assertEqual(self.mail.get_body(), '') self.mail.set_body('test') self.assertEqual(self.mail.get_body(), 'test') def testGetBody(self): pass # see testSetBody() def testAddAttachment(self): self.assertEqual(self.mail.get_attachments(), []) self.mail.add_attachment('foo') self.assertEqual(self.mail.get_attachments(), ['foo']) self.mail.add_attachment('bar') self.assertEqual(self.mail.get_attachments(), ['foo', 'bar']) def testGetAttachments(self): self.testAddAttachment() def testGetSmtpHeader(self): pass # see testSetFromTemplateString() def testGetSmtpMail(self): pass # see testSetFromTemplateString()