Partial setup flow with unittest.

chrome/browser/local_discovery/privetv3_setup_flow.cc

Index: chrome/browser/local_discovery/privetv3_setup_flow.cc
diff --git a/chrome/browser/local_discovery/privetv3_setup_flow.cc b/chrome/browser/local_discovery/privetv3_setup_flow.cc
index 5cafa932612b07d07d44d4c9e47352081b571a09..f15aaae566f1bb0efdd7312631469dfd4c91c20d 100644
--- a/chrome/browser/local_discovery/privetv3_setup_flow.cc
+++ b/chrome/browser/local_discovery/privetv3_setup_flow.cc
@@ -5,9 +5,85 @@
 #include "chrome/browser/local_discovery/privetv3_setup_flow.h"
 
 #include "base/logging.h"
+#include "chrome/browser/local_discovery/gcd_registration_ticket_request.h"
 
 namespace local_discovery {
 
+namespace {
+
+const char kSsidJsonKeyName[] = "wifi.ssid";
+const char kPasswordJsonKeyName[] = "wifi.passphrase";
+const char kTicketJsonKeyName[] = "registration.ticketID";
+const char kUserJsonKeyName[] = "registration.user";
+
+class SetupRequest : public PrivetV3Session::Request {
+ public:
+  explicit SetupRequest(PrivetV3SetupFlow* setup_flow);
+  virtual ~SetupRequest();
+
+  virtual std::string GetCallName() OVERRIDE {
+    return "/privet/v3/setup/start";
+  }
+  virtual const base::DictionaryValue& GetInput() OVERRIDE;
+
+  virtual void OnError(PrivetURLFetcher* fetcher,
+                       PrivetURLFetcher::ErrorType error) OVERRIDE;
+  virtual void OnParsedJson(PrivetURLFetcher* fetcher,
+                            const base::DictionaryValue& value,
+                            bool has_error) OVERRIDE;
+
+  void SetWiFiCridentials(const std::string& ssid, const std::string& password);
+
+  void SetRegistrationTicket(const std::string& ticket_id,
+                             const std::string& owner_email);
+
+ private:
+  base::DictionaryValue input_;
+  PrivetV3SetupFlow* setup_flow_;
+};
+
+SetupRequest::SetupRequest(PrivetV3SetupFlow* setup_flow)
+    : setup_flow_(setup_flow) {
+}
+
+SetupRequest::~SetupRequest() {
+}
+
+const base::DictionaryValue& SetupRequest::GetInput() {
+  return input_;
+}
+
+void SetupRequest::OnError(PrivetURLFetcher* fetcher,
+                           PrivetURLFetcher::ErrorType error) {
+  setup_flow_->OnSetupError();
+}
+
+void SetupRequest::OnParsedJson(PrivetURLFetcher* fetcher,
+                                const base::DictionaryValue& value,
+                                bool has_error) {
+  if (has_error)
+    return setup_flow_->OnSetupError();
+  setup_flow_->OnDeviceRegistered();
+}
+
+void SetupRequest::SetWiFiCridentials(const std::string& ssid,
+                                      const std::string& password) {
+  DCHECK(!ssid.empty());
+  DCHECK(!password.empty());
+  input_.SetString(kSsidJsonKeyName, ssid);
+  input_.SetString(kPasswordJsonKeyName, password);
+}
+
+void SetupRequest::SetRegistrationTicket(const std::string& ticket_id,
+                                         const std::string& owner_email) {
+  DCHECK(!ticket_id.empty());
+  DCHECK(!owner_email.empty());
+  input_.SetString(kTicketJsonKeyName, ticket_id);
+  input_.SetString(kUserJsonKeyName, owner_email);
+}
+
+}  // namespace
+
 PrivetV3SetupFlow::Delegate::~Delegate() {
 }
 
@@ -19,7 +95,17 @@ PrivetV3SetupFlow::~PrivetV3SetupFlow() {
 }
 
 void PrivetV3SetupFlow::Register(const std::string& service_name) {
-  NOTIMPLEMENTED();
+  service_name_ = service_name;
+  ticket_request_ = delegate_->CreateApiFlow();
+  if (!ticket_request_) {
+    OnSetupError();
+    return;
+  }
+  scoped_ptr<GCDApiFlow::Request> ticket_request(
+      new GCDRegistrationTicketRequest(
+          base::Bind(&PrivetV3SetupFlow::OnTicketCreated,
+                     weak_ptr_factory_.GetWeakPtr())));
+  ticket_request_->Start(ticket_request.Pass());
 }
 
 #if defined(ENABLE_WIFI_BOOTSTRAPPING)
@@ -28,4 +114,58 @@ void PrivetV3SetupFlow::SetupWifiAndRegister(const std::string& device_ssid) {
 }
 #endif  // ENABLE_WIFI_BOOTSTRAPPING
 
+void PrivetV3SetupFlow::OnSetupConfirmationNeeded(
+    const std::string& confirmation_code) {
+  delegate_->ConfirmSecurityCode(confirmation_code,
+                                 base::Bind(&PrivetV3SetupFlow::OnCodeConfirmed,
+                                            weak_ptr_factory_.GetWeakPtr()));
+}
+
+void PrivetV3SetupFlow::OnSessionEstablished() {
+  DCHECK(setup_request_);
+  session_->StartRequest(setup_request_.get());
+}
+
+void PrivetV3SetupFlow::OnCannotEstablishSession() {
+  OnSetupError();
+}
+
+void PrivetV3SetupFlow::OnSetupError() {
+  delegate_->OnSetupError();
+}
+
+void PrivetV3SetupFlow::OnDeviceRegistered() {
+  delegate_->OnSetupDone();
+}
+
+void PrivetV3SetupFlow::OnTicketCreated(const std::string& ticket_id) {
+  if (ticket_id.empty()) {
+    OnSetupError();
+    return;
+  }
+  SetupRequest* setup_request = new SetupRequest(this);
+  setup_request_.reset(setup_request);
+  setup_request->SetRegistrationTicket(ticket_id, "me");
+  delegate_->CreatePrivetV3Client(
+      service_name_,
+      base::Bind(&PrivetV3SetupFlow::OnPrivetClientCreated,
+                 weak_ptr_factory_.GetWeakPtr()));
+}
+
+void PrivetV3SetupFlow::OnPrivetClientCreated(
+    scoped_ptr<PrivetHTTPClient> privet_http_client) {
+  if (!privet_http_client) {
+    OnSetupError();
+    return;
+  }
+  session_.reset(new PrivetV3Session(privet_http_client.Pass(), this));
+  session_->Start();
+}
+
+void PrivetV3SetupFlow::OnCodeConfirmed(bool success) {
+  if (!success)
+    return OnSetupError();
+  session_->ConfirmCode();
+}
+
 }  // namespace local_discovery