1 files changed, 64 insertions, 0 deletions
diff --git a/components/proximity_auth/cryptauth/fake_cryptauth_gcm_manager.h b/components/proximity_auth/cryptauth/fake_cryptauth_gcm_manager.h
new file mode 100644
index 0000000..b25eb83
--- /dev/null
+++ b/components/proximity_auth/cryptauth/fake_cryptauth_gcm_manager.h
@@ -0,0 +1,64 @@
+// Copyright 2015 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.
+
+#ifndef COMPONENTS_PROXIMITY_AUTH_FAKE_CRYPTAUTH_FAKE_CRYPTAUTH_GCM_MANAGER_H
+#define COMPONENTS_PROXIMITY_AUTH_FAKE_CRYPTAUTH_FAKE_CRYPTAUTH_GCM_MANAGER_H
+
+#include "base/macros.h"
+#include "base/observer_list.h"
+#include "components/proximity_auth/cryptauth/cryptauth_gcm_manager.h"
+
+namespace proximity_auth {
+
+// CryptAuthGCMManager implementation for testing purposes.
+class FakeCryptAuthGCMManager : public CryptAuthGCMManager {
+ public:
+  // Creates the instance:
+  // |registration_id|: The GCM registration id from a previous successful
+  //     enrollment. Pass in an empty |registration_id| to simulate never having
+  //     registered successfully.
+  explicit FakeCryptAuthGCMManager(const std::string& registration_id);
+
+  ~FakeCryptAuthGCMManager() override;
+
+  // CryptAuthGCMManager:
+  void StartListening() override;
+  void RegisterWithGCM() override;
+  std::string GetRegistrationId() override;
+  void AddObserver(Observer* observer) override;
+  void RemoveObserver(Observer* observer) override;
+
+  // Simulates completing a GCM registration with the resulting
+  // |registration_id|. Passing an empty |registration_id| will simulate a
+  // registration failure.
+  // A registration attempt must currently be in progress.
+  void CompleteRegistration(const std::string& registration_id);
+
+  // Simulates receiving a re-enroll push message from GCM.
+  void PushReenrollMessage();
+
+  // Simulates receiving a re-sync push message from GCM.
+  void PushResyncMessage();
+
+  bool registration_in_progress() { return registration_in_progress_; }
+
+  void set_registration_id(const std::string& registration_id) {
+    registration_id_ = registration_id;
+  }
+
+ private:
+  // Whether a registration attempt is currently in progress.
+  bool registration_in_progress_;
+
+  // The registration id obtained from the last successful registration.
+  std::string registration_id_;
+
+  base::ObserverList<Observer> observers_;
+
+  DISALLOW_COPY_AND_ASSIGN(FakeCryptAuthGCMManager);
+};
+
+}  // namespace proximity_auth
+
+#endif  // COMPONENTS_PROXIMITY_FAKE_CRYPTAUTH_FAKE_CRYPTAUTH_GCM_MANAGER_H