1 files changed, 110 insertions, 0 deletions
diff --git a/components/pairing/fake_controller_pairing_controller.h b/components/pairing/fake_controller_pairing_controller.h
new file mode 100644
index 0000000..ce6630a
--- /dev/null
+++ b/components/pairing/fake_controller_pairing_controller.h
@@ -0,0 +1,110 @@
+// 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.
+
+#ifndef COMPONENTS_PAIRING_FAKE_CONTROLLER_PAIRING_CONTROLLER_H_
+#define COMPONENTS_PAIRING_FAKE_CONTROLLER_PAIRING_CONTROLLER_H_
+
+#include <set>
+#include <utility>
+
+#include "base/macros.h"
+#include "base/observer_list.h"
+#include "base/time/time.h"
+#include "components/pairing/controller_pairing_controller.h"
+
+namespace pairing_chromeos {
+
+class FakeControllerPairingController
+    : public ControllerPairingController,
+      public ControllerPairingController::Observer {
+ public:
+  typedef ControllerPairingController::Observer Observer;
+
+  enum DiscoveryEventType { DEVICE_FOUND, DEVICE_LOST, NOTHING_FOUND };
+
+  typedef std::pair<DiscoveryEventType, std::string> DiscoveryEvent;
+  typedef std::vector<DiscoveryEvent> DiscoveryScenario;
+
+  // Config is a coma separated list of key-value pairs separated by colon.
+  // Supported options:
+  // * async_duration - integer. Default: 3000.
+  // * fail_connecting - {0,1}. Default: 0.
+  // * connection_lost - integer-integer. Default: 0-0.
+  // * fail_enrollment - {0,1}. Default: 0.
+  // * discovery - {F,L,N}-string(~{F,L,N}-string)*. Default:
+  //   F-Device_1~F-Device_5~F-Device_3~L-Device_3~L-Device_1~F-Device_1
+  // * code - 6 digits or empty string. Default: empty string. If strings is
+  // empty, random code is generated.
+  explicit FakeControllerPairingController(const std::string& config);
+  virtual ~FakeControllerPairingController();
+
+  // Applies given |config| to controller.
+  void ApplyConfig(const std::string& config);
+
+  // Sets delay for asynchronous operations. like device searching or host
+  // enrollment. Default value is 3 seconds.
+  void set_async_duration(base::TimeDelta async_duration) {
+    async_duration_ = async_duration;
+  }
+
+  // Causing an error on |STAGE_ESTABLISHING_CONNECTION| stage once.
+  void SetShouldFailOnConnecting();
+
+  // Causing a connection loss on |stage_begin| and a connection recovery
+  // on |stage_end| once.
+  void SetShouldLoseConnection(Stage stage_begin, Stage stage_end);
+
+  // Makes host enrollment to fail once.
+  void SetEnrollmentShouldFail();
+
+  // Sets devices discovery scenario for |STAGE_DEVICES_DESCOVERY| stage.
+  // Events are executed with interval of |async_duration_|.
+  // For default scenario refer to implementation.
+  void SetDiscoveryScenario(const DiscoveryScenario& discovery_scenario);
+
+  // Overridden from ControllerPairingController:
+  virtual void AddObserver(Observer* observer) OVERRIDE;
+  virtual void RemoveObserver(Observer* observer) OVERRIDE;
+  virtual Stage GetCurrentStage() OVERRIDE;
+  virtual void StartPairing() OVERRIDE;
+  virtual DeviceIdList GetDiscoveredDevices() OVERRIDE;
+  virtual void ChooseDeviceForPairing(const std::string& device_id) OVERRIDE;
+  virtual void RepeatDiscovery() OVERRIDE;
+  virtual std::string GetConfirmationCode() OVERRIDE;
+  virtual void SetConfirmationCodeIsCorrect(bool correct) OVERRIDE;
+  virtual void OnAuthenticationDone(
+      const chromeos::UserContext& user_context,
+      content::BrowserContext* browser_context) OVERRIDE;
+  virtual void StartSession() OVERRIDE;
+
+ private:
+  void ChangeStage(Stage new_stage);
+  void ChangeStageLater(Stage new_stage);
+  void ExecuteDiscoveryEvent(size_t event_position);
+  void DeviceFound(const std::string& device_id);
+  void DeviceLost(const std::string& device_id);
+
+  // Overridden from ui::ControllerPairingController::Observer:
+  virtual void PairingStageChanged(Stage new_stage) OVERRIDE;
+  virtual void DiscoveredDevicesListChanged() OVERRIDE;
+
+  ObserverList<ControllerPairingController::Observer> observers_;
+  Stage current_stage_;
+  std::string confirmation_code_;
+  std::string preset_confirmation_code_;
+  base::TimeDelta async_duration_;
+  DiscoveryScenario discovery_scenario_;
+  std::set<std::string> discovered_devices_;
+  std::string choosen_device_;
+  bool should_fail_on_connecting_;
+  Stage connection_lost_begin_;
+  Stage connection_lost_end_;
+  bool enrollment_should_fail_;
+
+  DISALLOW_COPY_AND_ASSIGN(FakeControllerPairingController);
+};
+
+}  // namespace pairing_chromeos
+
+#endif  // COMPONENTS_PAIRING_FAKE_CONTROLLER_PAIRING_CONTROLLER_H_