src/chrome/browser/ui/cocoa/ssl_client_certificate_selector_cocoa_browsertest.mm

   1 // Copyright (c) 2012 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/ssl_client_certificate_selector_cocoa.h"
   6
   7 #import <SecurityInterface/SFChooseIdentityPanel.h>
   8
   9 #include "base/bind.h"
  10 #import "base/mac/mac_util.h"
  11 #include "chrome/browser/ssl/ssl_client_certificate_selector.h"
  12 #include "chrome/browser/ssl/ssl_client_certificate_selector_test.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "chrome/browser/ui/browser_commands.h"
  15 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  16 #include "components/web_modal/web_contents_modal_dialog_manager.h"
  17 #include "content/public/browser/web_contents.h"
  18 #include "content/public/test/test_utils.h"
  19 #include "ui/base/cocoa/window_size_constants.h"
  20
  21 using web_modal::WebContentsModalDialogManager;
  22
  23 namespace {
  24
  25 void OnCertificateSelected(net::X509Certificate** out_cert,
  26                            int* out_count,
  27                            net::X509Certificate* cert) {
  28   *out_cert = cert;
  29   ++(*out_count);
  30 }
  31
  32 }  // namespace
  33
  34 typedef SSLClientCertificateSelectorTestBase
  35     SSLClientCertificateSelectorCocoaTest;
  36
  37 // Flaky on 10.7; crbug.com/313243
  38 IN_PROC_BROWSER_TEST_F(SSLClientCertificateSelectorCocoaTest, DISABLED_Basic) {
  39   // TODO(kbr): re-enable: http://crbug.com/222296
  40   if (base::mac::IsOSMountainLionOrLater())
  41     return;
  42
  43   content::WebContents* web_contents =
  44       browser()->tab_strip_model()->GetActiveWebContents();
  45   WebContentsModalDialogManager* web_contents_modal_dialog_manager =
  46       WebContentsModalDialogManager::FromWebContents(web_contents);
  47   EXPECT_FALSE(web_contents_modal_dialog_manager->IsDialogActive());
  48
  49   net::X509Certificate* cert = NULL;
  50   int count = 0;
  51   SSLClientCertificateSelectorCocoa* selector =
  52       [[SSLClientCertificateSelectorCocoa alloc]
  53           initWithBrowserContext:web_contents->GetBrowserContext()
  54                  certRequestInfo:auth_requestor_->cert_request_info_.get()
  55                         callback:base::Bind(&OnCertificateSelected,
  56                                             &cert,
  57                                             &count)];
  58   [selector displayForWebContents:web_contents];
  59   content::RunAllPendingInMessageLoop();
  60   EXPECT_TRUE([selector panel]);
  61   EXPECT_TRUE(web_contents_modal_dialog_manager->IsDialogActive());
  62
  63   WebContentsModalDialogManager::TestApi test_api(
  64       web_contents_modal_dialog_manager);
  65   test_api.CloseAllDialogs();
  66   content::RunAllPendingInMessageLoop();
  67   EXPECT_FALSE(web_contents_modal_dialog_manager->IsDialogActive());
  68
  69   EXPECT_EQ(NULL, cert);
  70   EXPECT_EQ(1, count);
  71 }
  72
  73 // Test that switching to another tab correctly hides the sheet.
  74 IN_PROC_BROWSER_TEST_F(SSLClientCertificateSelectorCocoaTest, HideShow) {
  75   content::WebContents* web_contents =
  76       browser()->tab_strip_model()->GetActiveWebContents();
  77   SSLClientCertificateSelectorCocoa* selector =
  78       [[SSLClientCertificateSelectorCocoa alloc]
  79           initWithBrowserContext:web_contents->GetBrowserContext()
  80                  certRequestInfo:auth_requestor_->cert_request_info_.get()
  81                         callback:chrome::SelectCertificateCallback()];
  82   [selector displayForWebContents:web_contents];
  83   content::RunAllPendingInMessageLoop();
  84
  85   NSWindow* sheetWindow = [[selector overlayWindow] attachedSheet];
  86   NSRect sheetFrame = [sheetWindow frame];
  87   EXPECT_EQ(1.0, [sheetWindow alphaValue]);
  88
  89   // Switch to another tab and verify that the sheet is hidden.
  90   AddBlankTabAndShow(browser());
  91   EXPECT_EQ(0.0, [sheetWindow alphaValue]);
  92   EXPECT_TRUE(NSEqualRects(ui::kWindowSizeDeterminedLater,
  93                            [sheetWindow frame]));
  94
  95   // Switch back and verify that the sheet is shown.
  96   chrome::SelectNumberedTab(browser(), 0);
  97   EXPECT_EQ(1.0, [sheetWindow alphaValue]);
  98   EXPECT_TRUE(NSEqualRects(sheetFrame, [sheetWindow frame]));
  99 }