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_private_attempt(self): env = Envelope('*****@*****.**', ['*****@*****.**']) relay = Relay() self.mox.StubOutWithMock(relay, '_run_policies') self.mox.StubOutWithMock(relay, 'attempt') relay._run_policies(env) relay.attempt(env, 0) self.mox.ReplayAll() relay._attempt(env, 0)
def test_private_attempt(self): env = Envelope("*****@*****.**", ["*****@*****.**"]) relay = Relay() self.mox.StubOutWithMock(relay, "_run_policies") self.mox.StubOutWithMock(relay, "attempt") relay._run_policies(env) relay.attempt(env, 0) self.mox.ReplayAll() relay._attempt(env, 0)
def test_kill(self): relay = Relay() relay.kill() # no-op!
def test_public_attempt(self): env = Envelope('*****@*****.**', ['*****@*****.**']) relay = Relay() self.assertRaises(NotImplementedError, relay.attempt, env, 0)