class FirewallRulesSetCommandTestCase(unittest.TestCase): """ Тест на L{spamfighter.api.message.firewall.FirewallRulesSetCommand}. """ def setUp(self): self.c = FirewallRulesSetCommand() self.firewall = MessageFirewall() getDefaultDomain().set('testFIREWALL', self.firewall) def tearDown(self): getDefaultDomain().delete('testFIREWALL') def testRun(self): self.c.params.getUnserialized({'partner' : None, 'firewall' : 'testFIREWALL', 'rules' : 'stop as TEST'}) return self.c.run().addCallback(lambda _: self.assertEquals('stop as TEST', self.firewall.getRules())) def testRun(self): self.c.params.getUnserialized({'partner' : None, 'firewall' : 'testFIREWALL', 'rules' : 'YYYY'}) return self.c.run().addCallback(lambda _: self.assert_(False)).addErrback(lambda failure: failure.trap('spamfighter.core.commands.errors.FirewallSyntaxError'))
def setUp(self): self.c = FirewallRulesSetCommand() self.firewall = MessageFirewall() getDefaultDomain().set('testFIREWALL', self.firewall)