Test to make sure mbox files contain unixfrom

Start a test suite for generated mbox files.

Signed-off-by: Konstantin Ryabitsev <konstantin@linuxfoundation.org>

1 files changed, 35 insertions, 0 deletions

diff --git a/tests/test___init__.py b/tests/test___init__.py
index 5a09584..d78667e 100644
--- a/tests/test___init__.py
+++ b/tests/test___init__.py
@@ -1,5 +1,7 @@
 import pytest  # noqa
 import b4
+import re
+import os
 
 
 @pytest.mark.parametrize('source,expected', [
@@ -13,3 +15,36 @@ def test_check_gpg_status(source, expected):
     with open(f'tests/samples/gpg-{source}.txt', 'r') as fh:
         status = fh.read()
     assert b4.check_gpg_status(status) == expected
+
+
+@pytest.mark.parametrize('source,regex,flags,ismbox', [
+    (None, r'^From git@z ', 0, False),
+    (None, r'\n\nFrom git@z ', 0, False),
+])
+def test_save_git_am_mbox(tmpdir, source, regex, flags, ismbox):
+    import re
+    if source is not None:
+        if ismbox:
+            import mailbox
+            mbx = mailbox.mbox(f'tests/samples/{source}.txt')
+            msgs = list(mbx)
+        else:
+            import email
+            with open(f'tests/samples/{source}.txt', 'rb') as fh:
+                msg = email.message_from_binary_file(fh)
+            msgs = [msg]
+    else:
+        import email.message
+        msgs = list()
+        for x in range(0, 3):
+            msg = email.message.EmailMessage()
+            msg.set_payload(f'Hello world {x}\n')
+            msg['Subject'] = f'Hello world {x}'
+            msg['From'] = f'Me{x} <me{x}@foo.bar>'
+            msgs.append(msg)
+    dest = os.path.join(tmpdir, 'out')
+    with open(dest, 'w') as fh:
+        b4.save_git_am_mbox(msgs, fh)
+    with open(dest, 'r') as fh:
+        res = fh.read()
+    assert re.search(regex, res, flags=flags)