diff --git a/Milter/test.py b/Milter/test.py index 77edbc1..5643f54 100644 --- a/Milter/test.py +++ b/Milter/test.py @@ -62,8 +62,8 @@ class TestBase(object): raise IOError,"addrcpt not called from eom()" self._addrcpt.append(rcpt) - def setreply(self,rcode,xcode,msg): - self.reply = (rcode,xcode,msg) + def setreply(self,rcode,xcode,*msg): + self.reply = (rcode,xcode) + msg def feedFile(self,fp,sender="spam@adv.com",rcpt="victim@lamb.com"): self._body = None