Adding GCM support to the copresence component.

components/copresence/rpc/rpc_handler_unittest.cc

Index: components/copresence/rpc/rpc_handler_unittest.cc
diff --git a/components/copresence/rpc/rpc_handler_unittest.cc b/components/copresence/rpc/rpc_handler_unittest.cc
index 7efe2f64d8a607feedeae869b8875ca0075d8457..62fd56a9f9cba5ada50c540de91e2828870272e9 100644
--- a/components/copresence/rpc/rpc_handler_unittest.cc
+++ b/components/copresence/rpc/rpc_handler_unittest.cc
@@ -17,6 +17,7 @@
 #include "components/copresence/proto/data.pb.h"
 #include "components/copresence/proto/enums.pb.h"
 #include "components/copresence/proto/rpcs.pb.h"
+#include "components/copresence/test/fake_directive_handler.h"
 #include "components/copresence/test/stub_whispernet_client.h"
 #include "net/http/http_status_code.h"
 #include "testing/gmock/include/gmock/gmock.h"
@@ -43,45 +44,6 @@ void CreateSubscribedMessage(const std::vector<std::string>& subscription_ids,
   }
 }
 
-// TODO(ckehoe): Make DirectiveHandler an interface.
-class FakeDirectiveHandler final : public DirectiveHandler {
- public:
-  FakeDirectiveHandler() : DirectiveHandler(nullptr) {}
-
-  const std::vector<std::string>& added_directives() const {
-    return added_directives_;
-  }
-
-  const std::vector<std::string>& removed_directives() const {
-    return removed_directives_;
-  }
-
-  void Start(WhispernetClient* /* whispernet_client */,
-             const TokensCallback& /* tokens_cb */) override {
-    NOTREACHED();
-  }
-
-  void AddDirective(const Directive& directive) override {
-    added_directives_.push_back(directive.subscription_id());
-  }
-
-  void RemoveDirectives(const std::string& op_id) override {
-    removed_directives_.push_back(op_id);
-  }
-
-  const std::string GetCurrentAudioToken(AudioType type) const override {
-    return type == AUDIBLE ? "current audible" : "current inaudible";
-  }
-
-  bool IsAudioTokenHeard(AudioType type) const override { return true; }
-
- private:
-  std::vector<std::string> added_directives_;
-  std::vector<std::string> removed_directives_;
-
-  DISALLOW_COPY_AND_ASSIGN(FakeDirectiveHandler);
-};
-
 }  // namespace
 
 class RpcHandlerTest : public testing::Test, public CopresenceDelegate {
@@ -90,6 +52,7 @@ class RpcHandlerTest : public testing::Test, public CopresenceDelegate {
       : whispernet_client_(new StubWhispernetClient),
         rpc_handler_(this,
                      &directive_handler_,
+                     nullptr,
                      base::Bind(&RpcHandlerTest::CaptureHttpPost,
                                 base::Unretained(this))),
         status_(SUCCESS) {}
@@ -123,6 +86,11 @@ class RpcHandlerTest : public testing::Test, public CopresenceDelegate {
     return whispernet_client_.get();
   }
 
+  // TODO(ckehoe): Add GCM tests.
+  gcm::GCMDriver* GetGCMDriver() override {
+    return nullptr;
+  }
+
  protected:
 
   // Send test input to RpcHandler
@@ -140,7 +108,7 @@ class RpcHandlerTest : public testing::Test, public CopresenceDelegate {
     std::string serialized_response;
     response.SerializeToString(&serialized_response);
     rpc_handler_.RegisterResponseHandler(
-        auth_token, nullptr, net::HTTP_OK, serialized_response);
+        auth_token, false, nullptr, net::HTTP_OK, serialized_response);
   }
 
   void SendReport(scoped_ptr<ReportRequest> request,