def test_search_model_versions(self, mock_http): self.store.search_model_versions(filter_string="name='model_12'") self._verify_requests(mock_http, "model-versions/search", "GET", SearchModelVersions(filter="name='model_12'"))
def test_search_model_versions(mock_get_request_message, mock_model_registry_store): mock_get_request_message.return_value = SearchModelVersions( filter="source_path = 'A/B/CD'") mvds = [ ModelVersion( name="model_1", version="5", creation_timestamp=100, last_updated_timestamp=1200, description="v 5", user_id="u1", current_stage="Production", source="A/B/CD", run_id=uuid.uuid4().hex, status="READY", status_message=None, ), ModelVersion( name="model_1", version="12", creation_timestamp=110, last_updated_timestamp=2000, description="v 12", user_id="u2", current_stage="Production", source="A/B/CD", run_id=uuid.uuid4().hex, status="READY", status_message=None, ), ModelVersion( name="ads_model", version="8", creation_timestamp=200, last_updated_timestamp=2000, description="v 8", user_id="u1", current_stage="Staging", source="A/B/CD", run_id=uuid.uuid4().hex, status="READY", status_message=None, ), ModelVersion( name="fraud_detection_model", version="345", creation_timestamp=1000, last_updated_timestamp=1001, description="newest version", user_id="u12", current_stage="None", source="A/B/CD", run_id=uuid.uuid4().hex, status="READY", status_message=None, ), ] mock_model_registry_store.search_model_versions.return_value = mvds resp = _search_model_versions() args, _ = mock_model_registry_store.search_model_versions.call_args assert args == ("source_path = 'A/B/CD'", ) assert json.loads(resp.get_data()) == {"model_versions": jsonify(mvds)}