DandelionNStuff/pymilter
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Read header encoding tests as binary

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
Milter/testctx.py
+1
View File
@@ -270,6 +270,7 @@ class TestCtx(object):
if rc != Milter.CONTINUE: return rc if rc != Milter.CONTINUE: return rc
# header # header
for h,val in msg.items(): for h,val in msg.items():
# val is a Header object for invalid header values
rc = self._header(h,val) rc = self._header(h,val)
if rc != Milter.CONTINUE: return rc if rc != Milter.CONTINUE: return rc
# eoh # eoh
testsample.py
+2 -2
View File
@@ -29,10 +29,10 @@ class BMSMilterTestCase(unittest.TestCase):
ctx._setsymval('j','mailhost') ctx._setsymval('j','mailhost')
rc = ctx._connect() rc = ctx._connect()
self.assertTrue(rc == Milter.CONTINUE) self.assertTrue(rc == Milter.CONTINUE)
with open('test/'+fname) as fp: with open('test/'+fname,'rb') as fp:
rc = ctx._feedFile(fp) rc = ctx._feedFile(fp)
milter = ctx.getpriv() milter = ctx.getpriv()
self.assertFalse(ctx._bodyreplaced,"Message body not replaced") self.assertFalse(ctx._bodyreplaced,"Message body replaced")
fp = ctx._body fp = ctx._body
with open('test/'+fname+".tstout","wb") as ofp: with open('test/'+fname+".tstout","wb") as ofp:
ofp.write(fp.getvalue()) ofp.write(fp.getvalue())