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