Read header encoding tests as binary

This commit is contained in:
Stuart D. Gathman
2019-08-27 19:24:06 -04:00
parent a01f598e37
commit 18186a3c11
2 changed files with 3 additions and 2 deletions
+1
View File
@@ -270,6 +270,7 @@ class TestCtx(object):
if rc != Milter.CONTINUE: return rc
# header
for h,val in msg.items():
# val is a Header object for invalid header values
rc = self._header(h,val)
if rc != Milter.CONTINUE: return rc
# eoh
+2 -2
View File
@@ -29,10 +29,10 @@ class BMSMilterTestCase(unittest.TestCase):
ctx._setsymval('j','mailhost')
rc = ctx._connect()
self.assertTrue(rc == Milter.CONTINUE)
with open('test/'+fname) as fp:
with open('test/'+fname,'rb') as fp:
rc = ctx._feedFile(fp)
milter = ctx.getpriv()
self.assertFalse(ctx._bodyreplaced,"Message body not replaced")
self.assertFalse(ctx._bodyreplaced,"Message body replaced")
fp = ctx._body
with open('test/'+fname+".tstout","wb") as ofp:
ofp.write(fp.getvalue())