def test_version(self, mock_client, mocker): mock_client._core_service_stub = Core.CoreServiceStub(grpc.insecure_channel("")) mock_client._serving_service_stub = Serving.ServingServiceStub( grpc.insecure_channel("") ) mocker.patch.object( mock_client._core_service_stub, "GetFeastCoreVersion", return_value=GetFeastCoreVersionResponse(version="0.3.2"), ) mocker.patch.object( mock_client._serving_service_stub, "GetFeastServingInfo", return_value=GetFeastServingInfoResponse(version="0.3.2"), ) status = mock_client.version() assert ( status["core"]["url"] == CORE_URL and status["core"]["version"] == "0.3.2" and status["serving"]["url"] == SERVING_URL and status["serving"]["version"] == "0.3.2" )
def GetFeastCoreVersion(self, request, context): return GetFeastCoreVersionResponse(version="0.3.0")