def test_non_json_ok_response(): with mock.patch("requests.Session.request") as request_mock: host_only = MlflowHostCreds("http://my-host") response_mock = mock.MagicMock() response_mock.status_code = 200 response_mock.text = "<html></html>" request_mock.return_value = response_mock response_proto = GetRun.Response() with pytest.raises( MlflowException, match= "API request to endpoint was successful but the response body was not " "in a valid JSON format", ): call_endpoint(host_only, "/api/2.0/fetch-model", "GET", "", response_proto)
def _call_endpoint(self, api, json_body, call_all_endpoints=False): response_proto = api.Response() if call_all_endpoints: endpoints = _METHOD_TO_ALL_INFO[api] return call_endpoints(self.get_host_creds(), endpoints, json_body, response_proto) else: endpoint, method = _METHOD_TO_INFO[api] return call_endpoint(self.get_host_creds(), endpoint, method, json_body, response_proto)
def _call_endpoint(self, api, json_body): endpoint, method = _METHOD_TO_INFO[api] response_proto = api.Response() return call_endpoint(self.get_host_creds(), endpoint, method, json_body, response_proto)
def _call_endpoint(self, service, api, json_body): db_creds = get_databricks_host_creds(self.databricks_profile_uri) endpoint, method = _SERVICE_AND_METHOD_TO_INFO[service][api] response_proto = api.Response() return call_endpoint(db_creds, endpoint, method, json_body, response_proto)
def _call_endpoint(self, service, api, json_body): db_profile = get_db_profile_from_uri(mlflow.tracking.get_tracking_uri()) db_creds = get_databricks_host_creds(db_profile) endpoint, method = _SERVICE_AND_METHOD_TO_INFO[service][api] response_proto = api.Response() return call_endpoint(db_creds, endpoint, method, json_body, response_proto)