A ways better.

Tentatively fixes #7

1 files changed, 5 insertions, 14 deletions

diff --git a/test/test_pyg.py b/test/test_pyg.py
index e6b8fca..4076fc3 100755
--- a/test/test_pyg.py
+++ b/test/test_pyg.py
@@ -6,6 +6,7 @@ import subprocess
 import unittest
 
 import mail2news
+import news2mail
 
 
 class TestM2N(unittest.TestCase):
@@ -27,12 +28,7 @@ one line test
 """ % (mail2news.VERSION, mail2news.DESC)
 
     def test_m2n(self):
-        with open('examples/mail') as in_mail:
-            pid = subprocess.Popen(['python', 'pygm2n', '-Tv', '-n', 'pyg.test'],
-                                   stdin=subprocess.PIPE,
-                                   stdout=subprocess.PIPE,
-                                   universal_newlines=True)
-            out, _ = pid.communicate(in_mail.read())
+        out = mail2news.main(['-T', '-v', '-i', 'examples/mail', '-n', 'pyg.test'])
         self.assertEqual(out, self.expected_output)
 
 
@@ -59,22 +55,17 @@ Resent-Sender: sender@example.com
 
     def test_n2m(self):
         with open('examples/articletest.accepted') as in_mail:
-            pid = subprocess.Popen(['python', 'pygn2m', '-Tvt', 'test@example.com',
-                                    '-s', 'sender@example.com',
-                                    '-w', 'examples/whitelist.example'],
-                                   stdin=subprocess.PIPE,
-                                   stdout=subprocess.PIPE,
-                                   universal_newlines=True)
             in_message = in_mail.read().replace('pyg@pyg.server.tld',
                                                 'kame@inwind.it')
-            out, err = pid.communicate(in_message)
+            out = news2mail.main(['-T', '-v', '-t', 'test@example.com',
+                                    '-s', 'sender@example.com',
+                                    '-w', 'examples/whitelist.example'])
             out = re.sub(r'^Message-Id:.*$', '', out)
         # Not sure how to compare two email mesages (with different
         # times, etc.) so for now just to make sure the script doesn’t
         # blow up and gives some output
         # otherwise it would be
         # self.assertEqual(out, expected_output)
-        self.assertEqual(pid.returncode, 0)
         self.assertGreater(len(out), 0)