add flag call_nearby_identity_api and make api::SharingRpcClient base class of NearbyIdentityGrpcClient.

PiperOrigin-RevId: 698945381

sharing/internal/api/BUILD

@@ -40,6 +40,7 @@ cc_library(
     deps = [
         "//internal/platform:types",
         "//internal/platform/implementation:account_manager",
+        "//proto/identity/v1:rpcs_cc_proto",
         "//sharing/analytics",
         "//sharing/proto:share_cc_proto",
         "@com_google_absl//absl/functional:any_invocable",

sharing/internal/api/fake_nearby_share_client.cc

@@ -69,11 +69,38 @@ void FakeNearbyShareClient::ListPublicCertificates(
   std::move(callback)(response);
 }
 
+void FakeNearbyIdentityClient::QuerySharedCredentials(
+    const google::nearby::identity::v1::QuerySharedCredentialsRequest& request,
+    absl::AnyInvocable<
+        void(const absl::StatusOr<
+             google::nearby::identity::v1::QuerySharedCredentialsResponse>&
+                 response) &&>
+        callback) {
+  query_shared_credentials_requests_.emplace_back(request);
+  std::move(callback)(query_shared_credentials_response_);
+}
+
+void FakeNearbyIdentityClient::PublishDevice(
+    const google::nearby::identity::v1::PublishDeviceRequest& request,
+    absl::AnyInvocable<
+        void(const absl::StatusOr<
+             google::nearby::identity::v1::PublishDeviceResponse>& response) &&>
+        callback) {
+  publish_device_requests_.emplace_back(request);
+  std::move(callback)(publish_device_response_);
+}
+
 std::unique_ptr<nearby::sharing::api::SharingRpcClient>
 FakeNearbyShareClientFactory::CreateInstance() {
   auto instance = std::make_unique<FakeNearbyShareClient>();
   instances_.push_back(instance.get());
+  return instance;
+}
 
+std::unique_ptr<nearby::sharing::api::IdentityRpcClient>
+FakeNearbyShareClientFactory::CreateIdentityInstance() {
+  auto instance = std::make_unique<FakeNearbyIdentityClient>();
+  identity_instances_.push_back(instance.get());
   return instance;
 }
 

sharing/internal/api/fake_nearby_share_client.h

@@ -71,7 +71,6 @@ class FakeNearbyShareClient : public nearby::sharing::api::SharingRpcClient {
     list_public_certificates_responses_ = responses;
   }
 
- private:
   void UpdateDevice(
       const proto::UpdateDeviceRequest& request,
       absl::AnyInvocable<
@@ -103,6 +102,59 @@ class FakeNearbyShareClient : public nearby::sharing::api::SharingRpcClient {
       list_public_certificates_responses_;
 };
 
+// A fake implementation of the Nearby Identity RPC client that stores all
+// request data. Only use in unit tests.
+class FakeNearbyIdentityClient
+    : public nearby::sharing::api::IdentityRpcClient {
+ public:
+  FakeNearbyIdentityClient() = default;
+  ~FakeNearbyIdentityClient() override = default;
+
+  std::vector<google::nearby::identity::v1::PublishDeviceRequest>&
+  publish_device_requests() {
+    return publish_device_requests_;
+  }
+
+  void PublishDevice(
+      const google::nearby::identity::v1::PublishDeviceRequest& request,
+      absl::AnyInvocable<
+          void(const absl::StatusOr<google::nearby::identity::v1::
+                                        PublishDeviceResponse>& response) &&>
+          callback) override;
+
+  void SetPublishDeviceResponse(
+      absl::StatusOr<google::nearby::identity::v1::PublishDeviceResponse>
+          response) {
+    publish_device_response_ = response;
+  }
+
+  void QuerySharedCredentials(
+      const google::nearby::identity::v1::QuerySharedCredentialsRequest&
+          request,
+      absl::AnyInvocable<
+          void(const absl::StatusOr<
+               google::nearby::identity::v1::QuerySharedCredentialsResponse>&
+                   response) &&>
+          callback) override;
+
+  void SetQuerySharedCredentialsResponse(
+      absl::StatusOr<
+          google::nearby::identity::v1::QuerySharedCredentialsResponse>
+          response) {
+    query_shared_credentials_response_ = response;
+  }
+
+  std::vector<google::nearby::identity::v1::PublishDeviceRequest>
+      publish_device_requests_;
+  absl::StatusOr<google::nearby::identity::v1::PublishDeviceResponse>
+      publish_device_response_;
+
+  std::vector<google::nearby::identity::v1::QuerySharedCredentialsRequest>
+      query_shared_credentials_requests_;
+  absl::StatusOr<google::nearby::identity::v1::QuerySharedCredentialsResponse>
+      query_shared_credentials_response_;
+};
+
 class FakeNearbyShareClientFactory
     : public nearby::sharing::api::SharingRpcClientFactory {
  public:
@@ -112,6 +164,9 @@ class FakeNearbyShareClientFactory
  public:
   // Returns all FakeNearbyShareClient instances created by CreateInstance().
   std::vector<FakeNearbyShareClient*>& instances() { return instances_; }
+  std::vector<FakeNearbyIdentityClient*>& identity_instances() {
+    return identity_instances_;
+  }
   nearby::sharing::api::SharingRpcNotifier* GetRpcNotifier() const override {
     return nullptr;
   }
@@ -121,7 +176,11 @@ class FakeNearbyShareClientFactory
   std::unique_ptr<nearby::sharing::api::SharingRpcClient> CreateInstance()
       override;
 
+  std::unique_ptr<nearby::sharing::api::IdentityRpcClient>
+  CreateIdentityInstance() override;
+
   std::vector<FakeNearbyShareClient*> instances_;
+  std::vector<FakeNearbyIdentityClient*> identity_instances_;
 };
 
 }  // namespace sharing

sharing/internal/api/sharing_rpc_client.h

@@ -19,6 +19,7 @@
 
 #include "absl/functional/any_invocable.h"
 #include "absl/status/statusor.h"
+#include "proto/identity/v1/rpcs.pb.h"
 #include "sharing/internal/api/sharing_rpc_notifier.h"
 #include "sharing/proto/certificate_rpc.pb.h"
 #include "sharing/proto/contact_rpc.pb.h"
@@ -26,6 +27,29 @@
 
 namespace nearby::sharing::api {
 
+// IdentityRpcClient is used to access Nearby Identity backend APIs.
+class IdentityRpcClient {
+ public:
+  IdentityRpcClient() = default;
+  virtual ~IdentityRpcClient() = default;
+
+  virtual void QuerySharedCredentials(
+      const google::nearby::identity::v1::QuerySharedCredentialsRequest&
+          request,
+      absl::AnyInvocable<
+          void(const absl::StatusOr<
+               google::nearby::identity::v1::QuerySharedCredentialsResponse>&
+                   response) &&>
+          callback) = 0;
+
+  virtual void PublishDevice(
+      const google::nearby::identity::v1::PublishDeviceRequest& request,
+      absl::AnyInvocable<
+          void(const absl::StatusOr<google::nearby::identity::v1::
+                                        PublishDeviceResponse>& response) &&>
+          callback) = 0;
+};
+
 // SharingRpcClient is used to access Nearby Share backend APIs.
 class SharingRpcClient {
  public:
@@ -64,6 +88,7 @@ class SharingRpcClientFactory {
   virtual ~SharingRpcClientFactory() = default;
 
   virtual std::unique_ptr<SharingRpcClient> CreateInstance() = 0;
+  virtual std::unique_ptr<IdentityRpcClient> CreateIdentityInstance() = 0;
   virtual api::SharingRpcNotifier* GetRpcNotifier() const  = 0;
 };