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

suport chgfrom() in TestMilter

Browse Source 
see #3
This commit is contained in:
Yudai Kato
2016-08-10 15:00:20 +09:00
parent ea84943f29
commit 4c9c168096
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Milter/test.py
+8
View File
@@ -32,6 +32,8 @@ class TestBase(object):
self._bodyreplaced = False self._bodyreplaced = False
## True if the %milter changed any headers. ## True if the %milter changed any headers.
self._headerschanged = False self._headerschanged = False
## True if the %milter changed the envelope from.
self._envfromchanged = False
## Reply codes and messages set by the %milter ## Reply codes and messages set by the %milter
self._reply = None self._reply = None
## The rfc822 message object for the current email being fed to the %milter. ## The rfc822 message object for the current email being fed to the %milter.
@@ -60,6 +62,12 @@ class TestBase(object):
else: else:
raise IOError,"replacebody not called from eom()" 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 # FIXME: rfc822 indexing does not really reflect the way chg/add header
# work for a %milter # work for a %milter
def chgheader(self,field,idx,value): def chgheader(self,field,idx,value):