From 98bb246ffe322037a96867e976ee629b5eaefe35 Mon Sep 17 00:00:00 2001 From: yanghua Date: Tue, 7 Jun 2022 12:30:54 +0800 Subject: [PATCH] fix: Refactor test to reuse LocalRegistryFile /assign @felixwang9817 Signed-off-by: yanghua --- .../java/feast/serving/it/ServingBaseTests.java | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/java/serving/src/test/java/feast/serving/it/ServingBaseTests.java b/java/serving/src/test/java/feast/serving/it/ServingBaseTests.java index c610d7df6b1..30cba0cb068 100644 --- a/java/serving/src/test/java/feast/serving/it/ServingBaseTests.java +++ b/java/serving/src/test/java/feast/serving/it/ServingBaseTests.java @@ -28,11 +28,9 @@ import feast.proto.serving.ServingAPIProto; import feast.proto.serving.ServingAPIProto.FieldStatus; import feast.proto.types.ValueProto; +import feast.serving.registry.LocalRegistryFile; import feast.serving.util.DataGenerator; import io.grpc.StatusRuntimeException; -import java.io.IOException; -import java.nio.file.Files; -import java.nio.file.Paths; import java.util.List; import java.util.Map; import java.util.concurrent.TimeUnit; @@ -62,14 +60,8 @@ protected ServingAPIProto.GetOnlineFeaturesRequest buildOnlineRequest(int driver static RegistryProto.Registry registryProto = readLocalRegistry(); private static RegistryProto.Registry readLocalRegistry() { - try { - return RegistryProto.Registry.parseFrom( - Files.readAllBytes(Paths.get("src/test/resources/docker-compose/feast10/registry.db"))); - } catch (IOException e) { - e.printStackTrace(); - } - - return null; + return new LocalRegistryFile("src/test/resources/docker-compose/feast10/registry.db") + .getContent(); } @Test