Partial setup flow with unittest.

chrome/browser/local_discovery/privetv3_setup_flow.h

 // 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 CHROME_BROWSER_LOCAL_DISCOVERY_PRIVETV3_SETUP_FLOW_H_
 #define CHROME_BROWSER_LOCAL_DISCOVERY_PRIVETV3_SETUP_FLOW_H_
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/local_discovery/gcd_api_flow.h"
 #include "chrome/browser/local_discovery/privet_http.h"
+#include "chrome/browser/local_discovery/privetv3_session.h"
 
 namespace local_discovery {
 
 // Provides complete flow for Privet v3 device setup.
-class PrivetV3SetupFlow {
+class PrivetV3SetupFlow : public PrivetV3Session::Delegate {
  public:
   // Delegate to be implemented by client code.
   class Delegate {
    public:
     typedef base::Callback<void(bool success)> ResultCallback;
     // If |ssid| is empty, call failed to get credentials.
     // If |key| is empty, network is open.
     typedef base::Callback<void(const std::string& ssid,
                                 const std::string& key)> CredentialsCallback;
 
@@ skipping 25 matching lines @@
     virtual void RestoreWifi(const ResultCallback& callback) = 0;
 
     // Notifies client that device is set up.
     virtual void OnSetupDone() = 0;
 
     // Notifies client setup failed.
     virtual void OnSetupError() = 0;
   };
 
   explicit PrivetV3SetupFlow(Delegate* delegate);
-  ~PrivetV3SetupFlow();
+  virtual ~PrivetV3SetupFlow();
 
   // Starts registration.
   void Register(const std::string& service_name);
 
 #if defined(ENABLE_WIFI_BOOTSTRAPPING)
   void SetupWifiAndRegister(const std::string& device_ssid);
 #endif  // ENABLE_WIFI_BOOTSTRAPPING
 
+  // PrivetV3Session::Delegate implementation.
+  virtual void OnSetupConfirmationNeeded(
+      const std::string& confirmation_code) OVERRIDE;
+  virtual void OnSessionEstablished() OVERRIDE;
+  virtual void OnCannotEstablishSession() OVERRIDE;
+
+  void OnSetupError();
+  void OnDeviceRegistered();
+
  private:
+  void OnTicketCreated(const std::string& ticket_id);
+  void OnPrivetClientCreated(scoped_ptr<PrivetHTTPClient> privet_http_client);
+  void OnCodeConfirmed(bool success);
+
   Delegate* delegate_;
   std::string service_name_;
+  scoped_ptr<GCDApiFlow> ticket_request_;
+  scoped_ptr<PrivetV3Session> session_;
+  scoped_ptr<PrivetV3Session::Request> setup_request_;
   base::WeakPtrFactory<PrivetV3SetupFlow> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(PrivetV3SetupFlow);
 };
 
 }  // namespace local_discovery
 
 #endif  // CHROME_BROWSER_LOCAL_DISCOVERY_PRIVETV3_SETUP_FLOW_H_