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())