1// Copyright (c) 2013 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#import "chrome/browser/ui/cocoa/one_click_signin_dialog_controller.h" 6 7#include "chrome/browser/ui/browser.h" 8#include "chrome/browser/ui/browser_finder.h" 9#import "chrome/browser/ui/cocoa/one_click_signin_view_controller.h" 10#include "chrome/browser/ui/tabs/tab_strip_model.h" 11#include "chrome/test/base/in_process_browser_test.h" 12#import "testing/gtest_mac.h" 13 14class OneClickSigninDialogControllerTest : public InProcessBrowserTest { 15 public: 16 OneClickSigninDialogControllerTest() 17 : controller_(NULL), 18 sync_mode_(OneClickSigninSyncStarter::SYNC_WITH_DEFAULT_SETTINGS), 19 callback_count_(0) { 20 } 21 22 protected: 23 virtual void SetUpOnMainThread() OVERRIDE { 24 content::WebContents* web_contents = 25 browser()->tab_strip_model()->GetWebContentsAt(0); 26 BrowserWindow::StartSyncCallback callback = base::Bind( 27 &OneClickSigninDialogControllerTest::OnStartSyncCallback, 28 base::Unretained(this)); 29 controller_ = new OneClickSigninDialogController( 30 web_contents, callback, base::string16()); 31 EXPECT_NSEQ(@"OneClickSigninDialog", 32 [controller_->view_controller() nibName]); 33 } 34 35 // Weak pointer. Will delete itself when dialog closes. 36 OneClickSigninDialogController* controller_; 37 OneClickSigninSyncStarter::StartSyncMode sync_mode_; 38 int callback_count_; 39 40 private: 41 void OnStartSyncCallback(OneClickSigninSyncStarter::StartSyncMode mode) { 42 sync_mode_ = mode; 43 ++callback_count_; 44 } 45 46 DISALLOW_COPY_AND_ASSIGN(OneClickSigninDialogControllerTest); 47}; 48 49// Test that the dialog calls the callback if the OK button is clicked. 50// Callback should be called to setup sync with default settings. 51IN_PROC_BROWSER_TEST_F(OneClickSigninDialogControllerTest, OK) { 52 [controller_->view_controller() ok:nil]; 53 EXPECT_EQ(OneClickSigninSyncStarter::SYNC_WITH_DEFAULT_SETTINGS, sync_mode_); 54 EXPECT_EQ(1, callback_count_); 55} 56 57// Test that the dialog does call the callback if the Undo button 58// is clicked. Callback should be called to abort the sync. 59IN_PROC_BROWSER_TEST_F(OneClickSigninDialogControllerTest, Undo) { 60 [controller_->view_controller() onClickUndo:nil]; 61 EXPECT_EQ(OneClickSigninSyncStarter::UNDO_SYNC, sync_mode_); 62 EXPECT_EQ(1, callback_count_); 63} 64 65// Test that the advanced callback is run if its corresponding button 66// is clicked. 67IN_PROC_BROWSER_TEST_F(OneClickSigninDialogControllerTest, Advanced) { 68 [controller_->view_controller() onClickAdvancedLink:nil]; 69 EXPECT_EQ(OneClickSigninSyncStarter::CONFIGURE_SYNC_FIRST, sync_mode_); 70 EXPECT_EQ(1, callback_count_); 71} 72 73// Test that the dialog calls the callback if the bubble is closed. 74// Callback should be called to setup sync with default settings. 75IN_PROC_BROWSER_TEST_F(OneClickSigninDialogControllerTest, Close) { 76 controller_->constrained_window()->CloseWebContentsModalDialog(); 77 EXPECT_EQ(OneClickSigninSyncStarter::UNDO_SYNC, sync_mode_); 78 EXPECT_EQ(1, callback_count_); 79} 80 81// Test that clicking the learn more link opens a new window. 82IN_PROC_BROWSER_TEST_F(OneClickSigninDialogControllerTest, LearnMore) { 83 EXPECT_EQ(1u, chrome::GetTotalBrowserCount()); 84 [controller_->view_controller() textView:nil 85 clickedOnLink:nil 86 atIndex:0]; 87 EXPECT_EQ(2u, chrome::GetTotalBrowserCount()); 88} 89