def test_client_template_environment_mode_preview(self): tenv = MessageTemplateEnvironment() tenv.set_mode(MessageTemplateEnvironment.MODE_PREVIEW) self.assertTrue('inline_image' in tenv.globals) inline_image = tenv.globals['inline_image'] img_tag_result = inline_image(testing.TEST_MESSAGE_TEMPLATE_INLINE_IMAGE) img_tag_test = "<img src=\"file://{0}\">".format(testing.TEST_MESSAGE_TEMPLATE_INLINE_IMAGE) msg = 'The preview mode failed to properly format the img HTML tag' self.assertEqual(img_tag_result, img_tag_test, msg=msg)
def test_client_template_environment_mode_send(self): tenv = MessageTemplateEnvironment() tenv.set_mode(MessageTemplateEnvironment.MODE_SEND) self.assertTrue('inline_image' in tenv.globals) inline_image = tenv.globals['inline_image'] img_tag_result = inline_image(testing.TEST_MESSAGE_TEMPLATE_INLINE_IMAGE) img_tag_test = r'<img src="cid:' img_tag_test += self.image_cid_regex img_tag_test += r'">' msg = 'The send mode failed to properly format the img HTML tag' self.assertRegex(img_tag_result, img_tag_test, msg=msg)
def test_client_template_environment_mode_analyze(self): tenv = MessageTemplateEnvironment() self.assertTrue(hasattr(tenv, 'attachment_images')) self.assertIsInstance(tenv.attachment_images, dict) self.assertEqual(len(tenv.attachment_images), 0) tenv.set_mode(MessageTemplateEnvironment.MODE_ANALYZE) template = tenv.from_string(testing.TEST_MESSAGE_TEMPLATE) template.render(dict(client=dict(), url=dict())) msg = 'The analysis mode failed to identify the inline image' self.assertIn(testing.TEST_MESSAGE_TEMPLATE_INLINE_IMAGE, tenv.attachment_images, msg=msg) cid_value = tenv.attachment_images[testing.TEST_MESSAGE_TEMPLATE_INLINE_IMAGE] self.assertRegex(cid_value, self.image_cid_regex)