def test_get_server_status_no_db_conn(self, mock_status, mock_resp): """ Test server status woth no connection to db """ mock_status.get_version.return_value = {"platform_version": '2.6.1'} mock_status.get_mongo_conn_status.return_value = {'connected': False} mock_status.get_broker_conn_status.return_value = {'connected': True} request = mock.MagicMock() status = StatusView() response = status.get(request) expected_cont = {'known_workers': [], 'messaging_connection': {'connected': True}, 'database_connection': {'connected': False}, 'api_version': '2', 'versions': {"platform_version": '2.6.1'}} mock_resp.assert_called_once_with(expected_cont) self.assertTrue(response is mock_resp.return_value)
def test_get_server_status_broker_conn(self, mock_status, mock_resp): """ Test server status with broker connection false. """ mock_status.get_version.return_value = {"platform_version": '2.6.1'} mock_status.get_mongo_conn_status.return_value = {'connected': True} mock_status.get_broker_conn_status.return_value = {'connected': False} mock_worker = mock.MagicMock() mock_worker.to_mongo.return_value.to_dict.return_value = { "last_heartbeat": "2015-03-19T13:55:36Z", "name": "*****@*****.**"} mock_status.get_workers.return_value = [mock_worker] request = mock.MagicMock() status = StatusView() response = status.get(request) expected_cont = {'known_workers': [{'last_heartbeat': '2015-03-19T13:55:36Z', 'name': '*****@*****.**'}], 'messaging_connection': {'connected': False}, 'database_connection': {'connected': True}, 'api_version': '2', 'versions': {"platform_version": '2.6.1'}} mock_resp.assert_called_once_with(expected_cont) self.assertTrue(response is mock_resp.return_value)