aboutsummaryrefslogtreecommitdiff
path: root/tests/test___init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test___init__.py')
-rw-r--r--tests/test___init__.py35
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)