def testSetRules(self):
firewall = MessageFirewall()
message = TaggedMessage([])
firewall.setRules("stop as NEW")
return firewall.analyze(message, getDefaultDomain()).addCallback(
lambda result: self.assertEquals("NEW", result)
)
def testPickling(self):
import pickle
fw = MessageFirewall()
self.assertEquals("", pickle.loads(pickle.dumps(fw)).getRules())
fw.setRules("stop as OK")
self.assertEquals("stop as OK", pickle.loads(pickle.dumps(fw)).getRules())
