async def test_get_all(self, *args): mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Service.get_all(mock_db) mock_db.find.assert_called_with({}) mock_cursor.to_list.assert_called()
async def test_get_by_secure(self, *args): mock_secure = True mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Service.get_by_secure(mock_secure, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'secure': mock_secure}) mock_cursor.to_list.assert_called()
async def test_get_by_path(self, *args): mock_path = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Service.get_by_path(mock_path, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'path': mock_path}) mock_cursor.to_list.assert_called()
async def test_get_by_status_code(self, *args): mock_status_code = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Insights.get_by_status_code(mock_status_code, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'status_code': mock_status_code}) mock_cursor.to_list.assert_called()
async def test_get_by_method(self, *args): mock_method = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await RequestValidator.get_by_method(mock_method, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'method': mock_method}) mock_cursor.to_list.assert_called()
async def test_get_by_remote_ip(self, *args): mock_remote_ip = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Insights.get_by_remote_ip(mock_remote_ip, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'remote_ip': mock_remote_ip}) mock_cursor.to_list.assert_called()
async def test_get_by_service_id(self, *args): mock_service_id = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Insights.get_by_service_id(mock_service_id, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'service_id': mock_service_id}) mock_cursor.to_list.assert_called()
async def test_get_by_target(self, *args): mock_target = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Event.get_by_target(mock_target, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'target': mock_target}) mock_cursor.to_list.assert_called()
async def test_get_by_threshold(self, *args): mock_thresold = 0.0 mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await CircuitBreaker.get_by_threshold(mock_thresold, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with({'threshold': mock_thresold}) mock_cursor.to_list.assert_called()
async def test_get_by_circuit_breaker_id(self, *args): mock_circuit_breaker_mock = 'some-value' mock_db = CoroutineMock() mock_cursor = MagicMock() mock_cursor.to_list = CoroutineMock() mock_db.find = MagicMock() mock_db.find.return_value = mock_cursor await Event.get_by_circuit_breaker_id(mock_circuit_breaker_mock, mock_db) mock_db.find.assert_called() mock_db.find.assert_called_with( {'circuit_breaker_id': mock_circuit_breaker_mock}) mock_cursor.to_list.assert_called()