def setUp(self):
super(CastWithConsumerTest, self).setUp()
self.api = api.API(Mock, Mock)
self.origin_get_routing_key = self.api._get_routing_key
self.origin_create_consumer = kombu.Connection.create_consumer
self.origin_close = kombu.Connection.close
self.origin_create_connection = rpc.create_connection
def setUp(self):
super(ApiTest, self).setUp()
self.FAKE_ID = 'instance-id-x23d2d'
self.api = api.API(mock(), self.FAKE_ID)
when(rpc).call(any(), any(), any(), any(int)).thenRaise(
ValueError('Unexpected Rpc Invocation'))
when(rpc).cast(any(), any(), any()).thenRaise(
ValueError('Unexpected Rpc Invocation'))
def setUp(self):
super(ApiTest, self).setUp()
self.api = api.API(Mock, Mock)
self.origin_rpc_call = proxy.RpcProxy.call
proxy.RpcProxy.call = Mock()
self.rpc_call = proxy.RpcProxy.call
self.origin_rpc_cast = proxy.RpcProxy.cast
proxy.RpcProxy.cast = Mock()
self.rpc_cast = proxy.RpcProxy.cast
self.origin_object = agent_models.AgentHeartBeat.find_by
agent_models.AgentHeartBeat.find_by = Mock()
self.origin_is_active = agent_models.AgentHeartBeat.is_active
self.origin_api_id = self.api.id
def setUp(self):
super(OtherTests, self).setUp()
self.api = api.API(Mock, Mock)
self.origin_cast_with_consumer = self.api._cast_with_consumer
def setUp(self):
super(CastWithConsumerTest, self).setUp()
self.api = api.API(mock(), 'instance-id-x23d2d')
