From 4c9c1680961edc7b0b90b9cca5372891958562d0 Mon Sep 17 00:00:00 2001
From: Yudai Kato <grandeur09@gmail.com>
Date: Wed, 10 Aug 2016 15:00:20 +0900
Subject: [PATCH] suport chgfrom() in TestMilter

see #3
---
 Milter/test.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Milter/test.py b/Milter/test.py
index 8c9167d..4b04e6a 100644
--- a/Milter/test.py
+++ b/Milter/test.py
@@ -32,6 +32,8 @@ class TestBase(object):
     self._bodyreplaced = False
     ## True if the %milter changed any headers.
     self._headerschanged = False
+    ## True if the %milter changed the envelope from.
+    self._envfromchanged = False
     ## Reply codes and messages set by the %milter
     self._reply = None
     ## The rfc822 message object for the current email being fed to the %milter.
@@ -60,6 +62,12 @@ class TestBase(object):
     else:
       raise IOError,"replacebody not called from eom()"
 
+  def chgfrom(self,sender,params=None):
+    if not self._body:
+      raise IOError,"chgheader not called from eom()"
+    self.log('chgfrom: sender=%s' % (sender))
+    self._envfromchanged = True
+
   # FIXME: rfc822 indexing does not really reflect the way chg/add header
   # work for a %milter
   def chgheader(self,field,idx,value):