def test_specific_template(self):
providers = {
'spamhaus': '*****@*****.**',
'bitninja': '*****@*****.**',
'nfoservers': '*****@*****.**',
'lexsicom': '*****@*****.**',
}
for file_name, email in providers.iteritems():
sample = self.samples[file_name]
content = sample.read()
headers = self.email_parser.get_headers(content)
sender = parser.get_sender_from_headers(headers)
template = self.email_parser.get_template(sender)
self.assertEqual(self.email_parser._templates[email], template)
subject = parser.get_subject_from_headers(headers)
body, attachments = self.email_parser.get_body_and_attachments(content)
content = subject + '\n' + body
parsed_email = parser.ParsedEmail()
self.email_parser.update_parsed_email(parsed_email, content, template=template)
self.assertIn('1.2.3.4', parsed_email.ips)
def test_get_subject_sample2(self):
sample = self.samples['sample2']
headers = self.email_parser.get_headers(sample.read())
subject = parser.get_subject_from_headers(headers)
self.assertIn('[SpamCop (213.251.151.160) id:6301094139]{SPAM 07.7} =?UTF-8?Q?Impor', subject)
def test_get_subject_sample1(self):
sample = self.samples['sample1']
headers = self.email_parser.get_headers(sample.read())
subject = parser.get_subject_from_headers(headers)
self.assertEqual('Encoding test', subject)