def test_policies(self): env = Envelope('*****@*****.**', ['*****@*****.**']) p1 = self.mox.CreateMock(RelayPolicy) p2 = self.mox.CreateMock(RelayPolicy) p1.apply(env) p2.apply(env) self.mox.ReplayAll() relay = Relay() relay.add_policy(p1) relay.add_policy(p2) self.assertRaises(TypeError, relay.add_policy, None) relay._run_policies(env)
def test_policies(self): env = Envelope('*****@*****.**', ['*****@*****.**']) p1 = self.mox.CreateMock(RelayPolicy) p2 = self.mox.CreateMock(RelayPolicy) p1.apply(env) p2.apply(env) self.mox.ReplayAll() relay = Relay() relay.add_policy(p1) relay.add_policy(p2) self.assertRaises(TypeError, relay.add_policy, None) relay._run_policies(env)