Python FirewallRulesSetCommand Examples

Example #1

File: test_firewall.py Project: pombredanne/spamfighter

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

Example #2

File: test_firewall.py Project: pombredanne/spamfighter

 def setUp(self):
     self.c = FirewallRulesSetCommand()
     self.firewall = MessageFirewall()
     getDefaultDomain().set('testFIREWALL', self.firewall)