From bf1c2ff0779f5eb972a404e9cededeaee4004676 Mon Sep 17 00:00:00 2001 From: Yuecheng Zhu Date: Thu, 2 Jan 2025 22:57:46 -0800 Subject: [PATCH] Internal change PiperOrigin-RevId: 711638282 --- .../devtools/mobileharness/infra/client/api/util/stub/BUILD | 2 ++ .../client/api/util/stub/GrpcFutureInterfaceFactory.java | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/BUILD b/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/BUILD index 8404aa7a9..71d50cb45 100644 --- a/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/BUILD +++ b/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/BUILD @@ -53,6 +53,8 @@ java_library( "StubFactory.java", ], deps = [ + "//java/com/google/wireless/qa/mobileharness/master/rpc/stub:lab_info_future", + "//java/com/google/wireless/qa/mobileharness/master/rpc/stub/grpc:lab_info_future", "//src/devtools/mobileharness/shared/util/comm/stub:stub_configuration_java_proto", "//src/java/com/google/devtools/mobileharness/infra/lab/rpc/stub:exec_test", "//src/java/com/google/devtools/mobileharness/infra/lab/rpc/stub:prepare_test", diff --git a/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/GrpcFutureInterfaceFactory.java b/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/GrpcFutureInterfaceFactory.java index dd498e331..2819fd27d 100644 --- a/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/GrpcFutureInterfaceFactory.java +++ b/src/java/com/google/devtools/mobileharness/infra/client/api/util/stub/GrpcFutureInterfaceFactory.java @@ -18,9 +18,13 @@ import com.google.devtools.mobileharness.infra.master.rpc.stub.grpc.JobSyncGrpcStub; import com.google.devtools.mobileharness.shared.util.comm.stub.StubConfigurationProto.StubConfiguration; +import com.google.wireless.qa.mobileharness.master.rpc.stub.grpc.LabInfoGrpcFutureStub; /** Interface to create future interface for different services. */ interface GrpcFutureInterfaceFactory { JobSyncGrpcStub.FutureInterface createJobSyncFutureInterface(StubConfiguration stubConfiguration); + + LabInfoGrpcFutureStub.FutureInterface createLabInfoFutureInterface( + StubConfiguration stubConfiguration); }