From edef64a4226889aa10a362d6153abac865be6297 Mon Sep 17 00:00:00 2001 From: "Stuart D. Gathman" Date: Mon, 26 Sep 2016 18:55:25 -0400 Subject: [PATCH] Binary file output for emails --- testsample.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/testsample.py b/testsample.py index 5bff76a..6259257 100644 --- a/testsample.py +++ b/testsample.py @@ -19,7 +19,7 @@ class BMSMilterTestCase(unittest.TestCase): self.failUnless(rc == Milter.ACCEPT) self.failUnless(milter._bodyreplaced,"Message body not replaced") fp = milter._body - open('test/'+fname+".tstout","w").write(fp.getvalue()) + open('test/'+fname+".tstout","wb").write(fp.getvalue()) #self.failUnless(fp.getvalue() == open("test/virus1.out","r").read()) fp.seek(0) msg = mime.message_from_file(fp) @@ -34,7 +34,7 @@ class BMSMilterTestCase(unittest.TestCase): self.failUnless(rc == Milter.ACCEPT) self.failIf(milter._bodyreplaced,"Milter needlessly replaced body.") fp = milter._body - open('test/'+fname+".tstout","w").write(fp.getvalue()) + open('test/'+fname+".tstout","wb").write(fp.getvalue()) milter.close() def testDefang2(self): @@ -47,14 +47,14 @@ class BMSMilterTestCase(unittest.TestCase): self.failUnless(rc == Milter.ACCEPT) self.failUnless(milter._bodyreplaced,"Message body not replaced") fp = milter._body - open("test/virus3.tstout","w").write(fp.getvalue()) + open("test/virus3.tstout","wb").write(fp.getvalue()) #self.failUnless(fp.getvalue() == open("test/virus3.out","r").read()) rc = milter.feedMsg("virus6") self.failUnless(rc == Milter.ACCEPT) self.failUnless(milter._bodyreplaced,"Message body not replaced") self.failUnless(milter._headerschanged,"Message headers not adjusted") fp = milter._body - open("test/virus6.tstout","w").write(fp.getvalue()) + open("test/virus6.tstout","wb").write(fp.getvalue()) milter.close() def suite(): return unittest.makeSuite(BMSMilterTestCase,'test')