def test_connect(self): service = FedoraMessagingService() service.factory = mock.Mock() service.connect() self.assertEqual(len(service.services), 1) serv = service.services[0] self.assertTrue(serv.factory is service.factory) self.assertTrue(serv.parent is service) self.assertIsInstance(serv, TCPClient)
def test_connect_tls(self): """Assert connecting with amqps starts an SSLClient.""" service = FedoraMessagingService("amqps://") service.factory = mock.Mock() service.connect() self.assertEqual(len(service.services), 1) serv = service.services[0] self.assertTrue(serv.factory is service.factory) self.assertTrue(serv.parent is service) self.assertIsInstance(serv, SSLClient)
def test_connect(self): callback = mock.Mock() service = FedoraMessagingService(callback) factory = mock.Mock() service.factoryClass = mock.Mock(side_effect=lambda *a: factory) service.connect() service.factoryClass.assert_called_once_with(service._parameters, service._bindings) self.assertEqual(len(service.services), 1) serv = service.services[0] self.assertTrue(serv.factory is factory) self.assertTrue(serv.parent is service)
def test_startService(self): service = FedoraMessagingService() serv = mock.Mock() service.addService(serv) service.connect = mock.Mock() service.startService() service.connect.assert_called_once()
def test_startService_no_consume(self): service = FedoraMessagingService(None) serv = mock.Mock() service.addService(serv) service.connect = mock.Mock() service.startService() service.connect.assert_called_once() serv.factory.consume.assert_not_called()
def test_startService(self): callback = mock.Mock() service = FedoraMessagingService(callback) serv = mock.Mock() service.addService(serv) service.connect = mock.Mock() service.startService() service.connect.assert_called_once() serv.factory.consume.assert_called_once_with(callback)