fix: Refactor test to reuse LocalRegistryFile (#2763)

yanghua · web-flow · commit 4339c0acc990 · 2022-06-07T08:58:21.000-07:00
Signed-off-by: yanghua &lt;yanghua1127@gmail.com&gt;
diff --git 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
