def test_policies(self): self.celery.task(IgnoreArg()) p1 = self.mox.CreateMock(QueuePolicy) p2 = self.mox.CreateMock(QueuePolicy) p1.apply(self.env) p2.apply(self.env) self.mox.ReplayAll() queue = CeleryQueue(self.celery, self.relay) queue.add_policy(p1) queue.add_policy(p2) self.assertRaises(TypeError, queue.add_policy, None) queue._run_policies(self.env)