def test_check_client_not_found(self):
clients_manager = ClientsManager({'test': ClientCollectionTest.TestClient()})
clients_manager.check_connection = MagicMock(side_effect=KeyError)
client = ClientCheck(clients_manager)
client.__no_auth__ = True
self.api.add_route('/api/clients/{client}/check', client)
self.simulate_request('/api/clients/{0}/check'.format('tracker.org'))
self.assertEqual(self.srmock.status, falcon.HTTP_NOT_FOUND)
def test_check_client_error(self):
clients_manager = ClientsManager({'test': ClientCollectionTest.TestClient()})
clients_manager.check_connection = MagicMock(side_effect=Exception)
client = ClientCheck(clients_manager)
client.__no_auth__ = True
self.api.add_route('/api/clients/{client}/check', client)
self.simulate_request('/api/clients/{0}/check'.format('tracker.org'))
self.assertEqual(self.srmock.status, falcon.HTTP_INTERNAL_SERVER_ERROR)
def test_check_client(self, value):
clients_manager = ClientsManager({'test': ClientCollectionTest.TestClient()})
clients_manager.check_connection = MagicMock(return_value=value)
client = ClientCheck(clients_manager)
client.__no_auth__ = True
self.api.add_route('/api/clients/{client}/check', client)
body = self.simulate_request('/api/clients/{0}/check'.format('tracker.org'), decode="utf-8")
self.assertEqual(self.srmock.status, falcon.HTTP_OK)
self.assertTrue('application/json' in self.srmock.headers_dict['Content-Type'])
result = json.loads(body)
self.assertIsInstance(result, dict)
self.assertEqual(result, {'status': value})