def test_valid_search(self, mock_search): dr = dummy_request() expected_return_code = 200 mock_search.return_value = [] actual_return_code, _ = log_source_search(dr) assert actual_return_code == expected_return_code
def test_search_errors(self, mock_search, error_instance, expected_return): dr = dummy_request() mock_search.side_effect = error_instance actual_return_code, _ = log_source_search(dr) assert actual_return_code == expected_return