diff options
Diffstat (limited to 'distixlib/msg_id_extractor_tests.py')
| -rw-r--r-- | distixlib/msg_id_extractor_tests.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/distixlib/msg_id_extractor_tests.py b/distixlib/msg_id_extractor_tests.py index 6a7e259..b47b0f3 100644 --- a/distixlib/msg_id_extractor_tests.py +++ b/distixlib/msg_id_extractor_tests.py @@ -33,4 +33,18 @@ Subject: O, Romeo Wherefore art thou? ''') - self.assertEqual(distixlib.get_ids_from_message(msg), []) + self.assertEqual(distixlib.get_ids_from_message(msg), set()) + + def test_returns_id_from_message_id(self): + msg_text = '''\ +From: user@example.com +To: other@example.com +Subject: O, Romeo +Message-Id: <capulet123@verona> + +Wherefore art thou? +''' + + msg = email.message_from_string(msg_text) + self.assertEqual( + distixlib.get_ids_from_message(msg), set(['capulet123@verona'])) |