Add code to prompt for browser login during app notification setup

chrome/browser/extensions/app_notify_channel_setup_unittest.cc

 // Copyright (c) 2011 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.
 
+#include "base/bind.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop.h"
 #include "chrome/browser/extensions/app_notify_channel_setup.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/browser/browser_thread.h"
@@ skipping 42 matching lines @@
   // When our AppNotifyChannelSetupComplete method is called, we copy the
   // arguments into these member variables.
   std::string channel_id_;
   std::string error_;
   int route_id_;
   int callback_id_;
 
   DISALLOW_COPY_AND_ASSIGN(TestDelegate);
 };
 
+class TestUI : public AppNotifyChannelUI {
+ public:
+  TestUI() : delegate_(NULL) {}
+  ~TestUI() {}
+
+  // AppNotifyChannelUI.
+  virtual void PromptSyncSetup(Delegate* delegate) OVERRIDE {
+    CHECK(!delegate_);
+    delegate_ = delegate;
+
+    // If we have a result, post a task to call back the delegate with
+    // it. Otherwise ReportResult can be called manually at some later point.
+    if (setup_result_.get()) {
+      MessageLoop::current()->PostTask(
+          FROM_HERE,
+          base::Bind(&TestUI::ReportResult,
+                     base::Unretained(this),
+                     *setup_result_));
+    }
+  }
+
+  // This will make us automatically call back the delegate with |result| after
+  // PromptSyncSetup is called.
+  void SetSyncSetupResult(bool result) {
+    setup_result_.reset(new bool);
+    *setup_result_ = result;
+  }
+
+  void ReportResult(bool result) {
+    CHECK(delegate_);
+    delegate_->OnSyncSetupResult(result);
+  }
+
+ private:
+  AppNotifyChannelUI::Delegate* delegate_;
+  scoped_ptr<bool> setup_result_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestUI);
+};
+
 }  // namespace
 
 class AppNotifyChannelSetupTest : public testing::Test {
  public:
-  AppNotifyChannelSetupTest() : ui_thread_(BrowserThread::UI, &message_loop_) {}
+  AppNotifyChannelSetupTest() : ui_thread_(BrowserThread::UI, &message_loop_),
+                                ui_(new TestUI()) {
+  }
 
   virtual ~AppNotifyChannelSetupTest() {}
 
   virtual void SetChannelServerUrl(GURL channel_server_url) {
     CommandLine* command_line = CommandLine::ForCurrentProcess();
     command_line->AppendSwitchASCII(switches::kAppNotifyChannelServerURL,
                                     channel_server_url.spec());
   }
 
   virtual void SetLoggedInUser(const std::string username) {
@@ skipping 14 matching lines @@
 
     SetChannelServerUrl(server_url);
     SetLoggedInUser("user@gmail.com");
 
     scoped_refptr<AppNotifyChannelSetup > setup =
         new AppNotifyChannelSetup(&profile_,
                                   "1234",
                                   page_url,
                                   kRouteId,
                                   kCallbackId,
+                                  ui_.release(),
                                   delegate_.AsWeakPtr());
     setup->Start();
     message_loop_.Run();
     delegate_.ExpectWasCalled(expected_code, expected_error);
   }
 
  protected:
   MessageLoop message_loop_;
   BrowserThread ui_thread_;
   TestingProfile profile_;
   TestDelegate delegate_;
+  scoped_ptr<TestUI> ui_;
 };
 
-TEST_F(AppNotifyChannelSetupTest, NotAvailable) {
+TEST_F(AppNotifyChannelSetupTest, DidNotLogInToSync) {
   GURL url("http://www.google.com");
 
+  ui_->SetSyncSetupResult(false);
   scoped_refptr<AppNotifyChannelSetup > setup =
       new AppNotifyChannelSetup(&profile_,
                                 "1234",
                                 url,
                                 kRouteId,
                                 kCallbackId,
+                                ui_.release(),
                                 delegate_.AsWeakPtr());
   setup->Start();
   message_loop_.Run();
   delegate_.ExpectWasCalled(
       std::string(), std::string("not_available"));
 }
 
 TEST_F(AppNotifyChannelSetupTest, ServerSuccess) {
   RunServerTest(true, "dummy_do_not_use", "");
 }
 
 TEST_F(AppNotifyChannelSetupTest, ServerFailure) {
   RunServerTest(false, "", "channel_service_error");
 }