From 18186a3c112195bab4d0b5476d310487d5eea6f2 Mon Sep 17 00:00:00 2001 From: "Stuart D. Gathman" <stuart@gathman.org> Date: Tue, 27 Aug 2019 19:24:06 -0400 Subject: [PATCH] Read header encoding tests as binary --- Milter/testctx.py | 1 + testsample.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Milter/testctx.py b/Milter/testctx.py index b50dbb8..05d3669 100644 --- a/Milter/testctx.py +++ b/Milter/testctx.py @@ -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 diff --git a/testsample.py b/testsample.py index ce8ebb1..8618cd8 100644 --- a/testsample.py +++ b/testsample.py @@ -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()) -- GitLab