def test_get_text_body(self): template = MagicMock() template.render.side_effect = str email = Email(sender='ben', recipients=['bob', 'betty'], subject='email', template=template, data={'foo': 'bar'}) self.assertEqual(email.get_text_body(), "{'foo': 'bar'}")
def test_as_string(self): email = Email(sender='ben', recipients=['bob', 'betty'], subject='email', template=None, data=None) email.get_text_body = MagicMock() email.get_text_body.return_value = 'Some text' expected_string = """\ Content-Type: text/plain; charset="us-ascii" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit Subject: email From: ben To: bob, betty Some text""" self.assertEqual(email.as_string(), expected_string)