def test_connect_and_close_opens_and_closes_transport(self, amqp_client):
transport = Mock()
amqp_client.create_and_connect.return_value = transport
client = BusCtlClient()
client.connect()
client.close()
amqp_client.create_and_connect.assert_called_once_with()
transport.close.assert_called_once_with()
def test_connect_and_close_opens_and_closes_transport(self, amqp_client):
transport = Mock()
amqp_client.create_and_connect.return_value = transport
client = BusCtlClient()
client.connect()
client.close()
amqp_client.create_and_connect.assert_called_once_with()
transport.close.assert_called_once_with()
def test_close_transport_with_no_connection(self, amqp_client):
client = BusCtlClient()
client.close()
self.assertFalse(amqp_client.create_and_connect.called)
def test_close_transport_with_no_connection(self, amqp_client):
client = BusCtlClient()
client.close()
self.assertFalse(amqp_client.create_and_connect.called)
