Adding support for authenticated copresence calls

components/copresence/rpc/rpc_handler_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/copresence/rpc/rpc_handler.h"
 
 #include <map>
 #include <string>
 #include <vector>
 
@@ skipping 50 matching lines @@
  private:
   std::vector<Directive> added_directives_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeDirectiveHandler);
 };
 
 }  // namespace
 
 class RpcHandlerTest : public testing::Test, public CopresenceDelegate {
  public:
-  RpcHandlerTest() : rpc_handler_(this), status_(SUCCESS), api_key_("API key") {
+  RpcHandlerTest() : rpc_handler_(this), status_(SUCCESS) {
     rpc_handler_.server_post_callback_ =
         base::Bind(&RpcHandlerTest::CaptureHttpPost, base::Unretained(this));
     rpc_handler_.device_id_ = "Device ID";
   }
 
   void CaptureHttpPost(
       net::URLRequestContextGetter* url_context_getter,
       const std::string& rpc_name,
+      const std::string& app_id,
       scoped_ptr<MessageLite> request_proto,
       const RpcHandler::PostCleanupCallback& response_callback) {
     rpc_name_ = rpc_name;
+    app_id_ = app_id;
     request_proto_ = request_proto.Pass();
   }
 
   void CaptureStatus(CopresenceStatus status) {
     status_ = status;
   }
 
   inline const ReportRequest* GetReportSent() {
     return static_cast<ReportRequest*>(request_proto_.get());
   }
@@ skipping 53 matching lines @@
   }
 
   virtual net::URLRequestContextGetter* GetRequestContext() const override {
     return NULL;
   }
 
   virtual const std::string GetPlatformVersionString() const override {
     return kChromeVersion;
   }
 
-  virtual const std::string GetAPIKey() const override {
-    return api_key_;
+  virtual const std::string GetAPIKey(const std::string& app_id) const
+      override {
+    NOTREACHED();
+    return "";
+  }
+
+  virtual const std::string GetAuthToken(const std::string& app_id) const
+      override {
+    NOTREACHED();
+    return "";
   }
 
   virtual WhispernetClient* GetWhispernetClient() override {
     return NULL;
   }
 
  protected:
   // For rpc_handler_.invalid_audio_token_cache_
   base::MessageLoop message_loop_;
 
   RpcHandler rpc_handler_;
   CopresenceStatus status_;
-  std::string api_key_;
 
   std::string rpc_name_;
+  std::string app_id_;
   scoped_ptr<MessageLite> request_proto_;
   std::map<std::string, std::vector<Message>> messages_by_subscription_;
 };
 
 TEST_F(RpcHandlerTest, Initialize) {
   SetDeviceId("");
   rpc_handler_.Initialize(RpcHandler::SuccessCallback());
   RegisterDeviceRequest* registration =
       static_cast<RegisterDeviceRequest*>(request_proto_.get());
   Identity identity = registration->device_identifiers().registrant();
   EXPECT_EQ(CHROME, identity.type());
   EXPECT_FALSE(identity.chrome_id().empty());
 }
 
 TEST_F(RpcHandlerTest, CreateRequestHeader) {
   SetDeviceId("CreateRequestHeader Device ID");
   rpc_handler_.SendReportRequest(make_scoped_ptr(new ReportRequest),
                                  "CreateRequestHeader App ID",
                                  StatusCallback());
   EXPECT_EQ(RpcHandler::kReportRequestRpcName, rpc_name_);
+  EXPECT_EQ("CreateRequestHeader App ID", app_id_);
   ReportRequest* report = static_cast<ReportRequest*>(request_proto_.get());
   EXPECT_EQ(kChromeVersion,
             report->header().framework_version().version_name());
   EXPECT_EQ("CreateRequestHeader App ID",
             report->header().client_version().client());
   EXPECT_EQ("CreateRequestHeader Device ID",
             report->header().registered_device_id());
   EXPECT_EQ(CHROME_PLATFORM_TYPE,
             report->header().device_fingerprint().type());
 }
 
 TEST_F(RpcHandlerTest, ReportTokens) {
   std::vector<AudioToken> test_tokens;
   test_tokens.push_back(AudioToken("token 1", false));
   test_tokens.push_back(AudioToken("token 2", true));
   test_tokens.push_back(AudioToken("token 3", false));
   AddInvalidToken("token 2");
 
   rpc_handler_.ReportTokens(test_tokens);
   EXPECT_EQ(RpcHandler::kReportRequestRpcName, rpc_name_);
+  EXPECT_EQ("", app_id_);
   ReportRequest* report = static_cast<ReportRequest*>(request_proto_.get());
   RepeatedPtrField<TokenObservation> tokens_sent =
       report->update_signals_request().token_observation();
   ASSERT_EQ(2, tokens_sent.size());
   EXPECT_EQ("token 1", tokens_sent.Get(0).token_id());
   EXPECT_EQ("token 3", tokens_sent.Get(1).token_id());
 }
 
 TEST_F(RpcHandlerTest, ReportResponseHandler) {
   // Fail on HTTP status != 200.
@@ skipping 50 matching lines @@
             messages_by_subscription_["Subscription 2"][1].payload());
 
   ASSERT_EQ(2U, directive_handler->added_directives().size());
   EXPECT_EQ("Subscription 1",
             directive_handler->added_directives()[0].subscription_id());
   EXPECT_EQ("Subscription 2",
             directive_handler->added_directives()[1].subscription_id());
 }
 
 }  // namespace copresence