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