minor test improvements

This commit is contained in:
phail 2022-11-20 20:12:41 +01:00
parent af8a6c3764
commit 3599bb52c0

View File

@ -92,125 +92,130 @@ class TestParser(TestCase):
"message/rfc822", "message/rfc822",
) )
self.assertTrue( self.assertIn(
{"namespace": "", "prefix": "", "key": "attachments", "value": ""} {"namespace": "", "prefix": "", "key": "attachments", "value": ""},
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "", "prefix": "",
"key": "date", "key": "date",
"value": "2022-10-12 21:40:43 UTC+02:00", "value": "2022-10-12 21:40:43 UTC+02:00",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "content-language", "key": "content-language",
"value": "en-US", "value": "en-US",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "content-type", "key": "content-type",
"value": "text/plain; charset=UTF-8; format=flowed", "value": "text/plain; charset=UTF-8; format=flowed",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "date", "key": "date",
"value": "Wed, 12 Oct 2022 21:40:43 +0200", "value": "Wed, 12 Oct 2022 21:40:43 +0200",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "delivered-to", "key": "delivered-to",
"value": "mail@someserver.de", "value": "mail@someserver.de",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "from", "key": "from",
"value": "Some One <mail@someserver.de>", "value": "Some One <mail@someserver.de>",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "message-id", "key": "message-id",
"value": "<6e99e34d-e20a-80c4-ea61-d8234b612be9@someserver.de>", "value": "<6e99e34d-e20a-80c4-ea61-d8234b612be9@someserver.de>",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{"namespace": "", "prefix": "header", "key": "mime-version", "value": "1.0"} {
in metadata, "namespace": "",
"prefix": "header",
"key": "mime-version",
"value": "1.0",
},
metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "received", "key": "received",
"value": "from mail.someserver.org ([::1])\n\tby e1acdba3bd07 with LMTP\n\tid KBKZGD2YR2NTCgQAjubtDA\n\t(envelope-from <mail@someserver.de>)\n\tfor <mail@someserver.de>; Wed, 10 Oct 2022 11:40:46 +0200, from [127.0.0.1] (localhost [127.0.0.1]) by localhost (Mailerdaemon) with ESMTPSA id 2BC9064C1616\n\tfor <some@one.de>; Wed, 12 Oct 2022 21:40:46 +0200 (CEST)", "value": "from mail.someserver.org ([::1])\n\tby e1acdba3bd07 with LMTP\n\tid KBKZGD2YR2NTCgQAjubtDA\n\t(envelope-from <mail@someserver.de>)\n\tfor <mail@someserver.de>; Wed, 10 Oct 2022 11:40:46 +0200, from [127.0.0.1] (localhost [127.0.0.1]) by localhost (Mailerdaemon) with ESMTPSA id 2BC9064C1616\n\tfor <some@one.de>; Wed, 12 Oct 2022 21:40:46 +0200 (CEST)",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "return-path", "key": "return-path",
"value": "<mail@someserver.de>", "value": "<mail@someserver.de>",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "subject", "key": "subject",
"value": "Simple Text Mail", "value": "Simple Text Mail",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{"namespace": "", "prefix": "header", "key": "to", "value": "some@one.de"} {"namespace": "", "prefix": "header", "key": "to", "value": "some@one.de"},
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "user-agent", "key": "user-agent",
"value": "Mozilla/5.0 (X11; Linux x86_64; rv:102.0) Gecko/20100101\n Thunderbird/102.3.1", "value": "Mozilla/5.0 (X11; Linux x86_64; rv:102.0) Gecko/20100101\n Thunderbird/102.3.1",
} },
in metadata, metadata,
) )
self.assertTrue( self.assertIn(
{ {
"namespace": "", "namespace": "",
"prefix": "header", "prefix": "header",
"key": "x-last-tls-session-version", "key": "x-last-tls-session-version",
"value": "TLSv1.3", "value": "TLSv1.3",
} },
in metadata, metadata,
) )
def test_parse_na(self): def test_parse_na(self):
@ -438,8 +443,8 @@ class TestParser(TestCase):
resulting_html = result[-1][1].read() resulting_html = result[-1][1].read()
self.assertTrue(result[-1][0] == "index.html") self.assertTrue(result[-1][0] == "index.html")
self.assertTrue(result[0][0] in resulting_html) self.assertIn(result[0][0], resulting_html)
self.assertFalse("<script" in resulting_html.lower()) self.assertNotIn("<script", resulting_html.lower())
@mock.patch("paperless_mail.parsers.requests.post") @mock.patch("paperless_mail.parsers.requests.post")
def test_generate_pdf_from_html(self, mock_post: mock.MagicMock): def test_generate_pdf_from_html(self, mock_post: mock.MagicMock):