src/chrome/browser/ui/cocoa/extensions/extension_install_view_controller_unittest.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 <Cocoa/Cocoa.h>
   6
   7 #import "base/mac/scoped_nsobject.h"
   8 #include "base/strings/sys_string_conversions.h"
   9 #include "base/strings/utf_string_conversions.h"
  10 #import "chrome/browser/extensions/extension_install_prompt.h"
  11 #include "chrome/browser/ui/browser.h"
  12 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
  13 #import "chrome/browser/ui/cocoa/extensions/extension_install_prompt_test_utils.h"
  14 #import "chrome/browser/ui/cocoa/extensions/extension_install_view_controller.h"
  15 #include "extensions/common/extension.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17 #import "testing/gtest_mac.h"
  18
  19 using extensions::Extension;
  20
  21 // Base class for our tests.
  22 class ExtensionInstallViewControllerTest : public CocoaProfileTest {
  23  public:
  24   ExtensionInstallViewControllerTest() {
  25     extension_ = chrome::LoadInstallPromptExtension();
  26   }
  27
  28  protected:
  29   scoped_refptr<extensions::Extension> extension_;
  30 };
  31
  32 // Test that we can load the two kinds of prompts correctly, that the outlets
  33 // are hooked up, and that the dialog calls cancel when cancel is pressed.
  34 TEST_F(ExtensionInstallViewControllerTest, BasicsNormalCancel) {
  35   chrome::MockExtensionInstallPromptDelegate delegate;
  36
  37   scoped_refptr<ExtensionInstallPrompt::Prompt> prompt =
  38       chrome::BuildExtensionInstallPrompt(extension_.get());
  39   ExtensionInstallPrompt::PermissionsType type =
  40       ExtensionInstallPrompt::PermissionsType::REGULAR_PERMISSIONS;
  41
  42   std::vector<base::string16> permissions;
  43   permissions.push_back(base::UTF8ToUTF16("warning 1"));
  44   prompt->SetPermissions(permissions, type);
  45   // No details provided with this permission.
  46   std::vector<base::string16> details;
  47   details.push_back(base::string16());
  48   prompt->SetPermissionsDetails(details, type);
  49
  50   base::scoped_nsobject<ExtensionInstallViewController> controller(
  51       [[ExtensionInstallViewController alloc] initWithProfile:profile()
  52                                                     navigator:browser()
  53                                                      delegate:&delegate
  54                                                        prompt:prompt]);
  55
  56   [controller view];  // Force nib load.
  57
  58   // Test the right nib loaded.
  59   EXPECT_NSEQ(@"ExtensionInstallPrompt", [controller nibName]);
  60
  61   // Check all the controls.
  62   // Make sure everything is non-nil, and that the fields that are
  63   // auto-translated don't start with a caret (that would indicate that they
  64   // were not translated).
  65   EXPECT_TRUE([controller iconView]);
  66   EXPECT_TRUE([[controller iconView] image]);
  67
  68   EXPECT_TRUE([controller titleField]);
  69   EXPECT_NE(0u, [[[controller titleField] stringValue] length]);
  70
  71   NSOutlineView* outlineView = [controller outlineView];
  72   EXPECT_TRUE(outlineView);
  73   EXPECT_EQ(2, [outlineView numberOfRows]);
  74   EXPECT_NSEQ(base::SysUTF16ToNSString(prompt->GetPermission(0, type)),
  75               [[outlineView dataSource] outlineView:outlineView
  76                           objectValueForTableColumn:nil
  77                                              byItem:[outlineView itemAtRow:1]]);
  78
  79   EXPECT_TRUE([controller cancelButton]);
  80   EXPECT_NE(0u, [[[controller cancelButton] stringValue] length]);
  81   EXPECT_NE('^', [[[controller cancelButton] stringValue] characterAtIndex:0]);
  82
  83   EXPECT_TRUE([controller okButton]);
  84   EXPECT_NE(0u, [[[controller okButton] stringValue] length]);
  85   EXPECT_NE('^', [[[controller okButton] stringValue] characterAtIndex:0]);
  86
  87   // Test that cancel calls our delegate.
  88   [controller cancel:nil];
  89   EXPECT_EQ(1, delegate.abort_count());
  90   EXPECT_EQ(0, delegate.proceed_count());
  91 }
  92
  93 TEST_F(ExtensionInstallViewControllerTest, BasicsNormalOK) {
  94   chrome::MockExtensionInstallPromptDelegate delegate;
  95
  96   scoped_refptr<ExtensionInstallPrompt::Prompt> prompt =
  97       chrome::BuildExtensionInstallPrompt(extension_.get());
  98   ExtensionInstallPrompt::PermissionsType type =
  99       ExtensionInstallPrompt::PermissionsType::REGULAR_PERMISSIONS;
 100
 101   std::vector<base::string16> permissions;
 102   permissions.push_back(base::UTF8ToUTF16("warning 1"));
 103   prompt->SetPermissions(permissions, type);
 104   // No details provided with this permission.
 105   std::vector<base::string16> details;
 106   details.push_back(base::string16());
 107   prompt->SetPermissionsDetails(details, type);
 108
 109   base::scoped_nsobject<ExtensionInstallViewController> controller(
 110       [[ExtensionInstallViewController alloc] initWithProfile:profile()
 111                                                     navigator:browser()
 112                                                      delegate:&delegate
 113                                                        prompt:prompt]);
 114
 115   [controller view];  // Force nib load.
 116   [controller ok:nil];
 117
 118   EXPECT_EQ(0, delegate.abort_count());
 119   EXPECT_EQ(1, delegate.proceed_count());
 120 }
 121
 122 // Test that controls get repositioned when there are two warnings vs one
 123 // warning.
 124 TEST_F(ExtensionInstallViewControllerTest, MultipleWarnings) {
 125   chrome::MockExtensionInstallPromptDelegate delegate1;
 126   chrome::MockExtensionInstallPromptDelegate delegate2;
 127
 128   scoped_refptr<ExtensionInstallPrompt::Prompt> one_warning_prompt =
 129       chrome::BuildExtensionInstallPrompt(extension_.get());
 130   ExtensionInstallPrompt::PermissionsType type =
 131       ExtensionInstallPrompt::PermissionsType::REGULAR_PERMISSIONS;
 132
 133   std::vector<base::string16> permissions;
 134   permissions.push_back(base::UTF8ToUTF16("warning 1"));
 135   one_warning_prompt->SetPermissions(permissions, type);
 136   // No details provided with this permission.
 137   std::vector<base::string16> details;
 138   details.push_back(base::string16());
 139   one_warning_prompt->SetPermissionsDetails(details, type);
 140
 141   scoped_refptr<ExtensionInstallPrompt::Prompt> two_warnings_prompt =
 142       chrome::BuildExtensionInstallPrompt(extension_.get());
 143   permissions.push_back(base::UTF8ToUTF16("warning 2"));
 144   two_warnings_prompt->SetPermissions(permissions, type);
 145   // No details provided with this permission.
 146   details.push_back(base::string16());
 147   two_warnings_prompt->SetPermissionsDetails(details, type);
 148
 149   base::scoped_nsobject<ExtensionInstallViewController> controller1(
 150       [[ExtensionInstallViewController alloc]
 151           initWithProfile:profile()
 152                 navigator:browser()
 153                  delegate:&delegate1
 154                    prompt:one_warning_prompt]);
 155
 156   [controller1 view];  // Force nib load.
 157
 158   base::scoped_nsobject<ExtensionInstallViewController> controller2(
 159       [[ExtensionInstallViewController alloc]
 160           initWithProfile:profile()
 161                 navigator:browser()
 162                  delegate:&delegate2
 163                    prompt:two_warnings_prompt]);
 164
 165   [controller2 view];  // Force nib load.
 166
 167   // Test control positioning. We don't test exact positioning because we don't
 168   // want this to depend on string details and localization. But we do know the
 169   // relative effect that adding a second warning should have on the layout.
 170   ASSERT_LT([[controller1 view] frame].size.height,
 171             [[controller2 view] frame].size.height);
 172
 173   ASSERT_LT([[controller1 view] frame].size.height,
 174             [[controller2 view] frame].size.height);
 175 }
 176
 177 // Test that we can load the skinny prompt correctly, and that the outlets are
 178 // are hooked up.
 179 TEST_F(ExtensionInstallViewControllerTest, BasicsSkinny) {
 180   chrome::MockExtensionInstallPromptDelegate delegate;
 181
 182   // No warnings should trigger skinny prompt.
 183   scoped_refptr<ExtensionInstallPrompt::Prompt> no_warnings_prompt =
 184       chrome::BuildExtensionInstallPrompt(extension_.get());
 185
 186   base::scoped_nsobject<ExtensionInstallViewController> controller(
 187       [[ExtensionInstallViewController alloc]
 188           initWithProfile:profile()
 189                 navigator:browser()
 190                  delegate:&delegate
 191                    prompt:no_warnings_prompt]);
 192
 193   [controller view];  // Force nib load.
 194
 195   // Test the right nib loaded.
 196   EXPECT_NSEQ(@"ExtensionInstallPromptNoWarnings", [controller nibName]);
 197
 198   // Check all the controls.
 199   // In the skinny prompt, only the icon, title and buttons are non-nill.
 200   // Everything else is nil.
 201   EXPECT_TRUE([controller iconView]);
 202   EXPECT_TRUE([[controller iconView] image]);
 203
 204   EXPECT_TRUE([controller titleField]);
 205   EXPECT_NE(0u, [[[controller titleField] stringValue] length]);
 206
 207   EXPECT_TRUE([controller cancelButton]);
 208   EXPECT_NE(0u, [[[controller cancelButton] stringValue] length]);
 209   EXPECT_NE('^', [[[controller cancelButton] stringValue] characterAtIndex:0]);
 210
 211   EXPECT_TRUE([controller okButton]);
 212   EXPECT_NE(0u, [[[controller okButton] stringValue] length]);
 213   EXPECT_NE('^', [[[controller okButton] stringValue] characterAtIndex:0]);
 214
 215   EXPECT_FALSE([controller outlineView]);
 216 }
 217
 218
 219 // Test that we can load the inline prompt correctly, and that the outlets are
 220 // are hooked up.
 221 TEST_F(ExtensionInstallViewControllerTest, BasicsInline) {
 222   chrome::MockExtensionInstallPromptDelegate delegate;
 223
 224   // No warnings should trigger skinny prompt.
 225   scoped_refptr<ExtensionInstallPrompt::Prompt> inline_prompt =
 226       new ExtensionInstallPrompt::Prompt(
 227           ExtensionInstallPrompt::INLINE_INSTALL_PROMPT);
 228   inline_prompt->SetWebstoreData("1,000", true, 3.5, 200);
 229   inline_prompt->set_extension(extension_.get());
 230   inline_prompt->set_icon(chrome::LoadInstallPromptIcon());
 231
 232   base::scoped_nsobject<ExtensionInstallViewController> controller(
 233       [[ExtensionInstallViewController alloc] initWithProfile:profile()
 234                                                     navigator:browser()
 235                                                      delegate:&delegate
 236                                                        prompt:inline_prompt]);
 237
 238   [controller view];  // Force nib load.
 239
 240   // Test the right nib loaded.
 241   EXPECT_NSEQ(@"ExtensionInstallPromptWebstoreData", [controller nibName]);
 242
 243   // Check all the controls.
 244   EXPECT_TRUE([controller iconView]);
 245   EXPECT_TRUE([[controller iconView] image]);
 246
 247   EXPECT_TRUE([controller titleField]);
 248   EXPECT_NE(0u, [[[controller titleField] stringValue] length]);
 249
 250   EXPECT_TRUE([controller cancelButton]);
 251   EXPECT_NE(0u, [[[controller cancelButton] stringValue] length]);
 252   EXPECT_NE('^', [[[controller cancelButton] stringValue] characterAtIndex:0]);
 253
 254   EXPECT_TRUE([controller okButton]);
 255   EXPECT_NE(0u, [[[controller okButton] stringValue] length]);
 256   EXPECT_NE('^', [[[controller okButton] stringValue] characterAtIndex:0]);
 257
 258   EXPECT_TRUE([controller ratingStars]);
 259   EXPECT_EQ(5u, [[[controller ratingStars] subviews] count]);
 260
 261   EXPECT_TRUE([controller ratingCountField]);
 262   EXPECT_NE(0u, [[[controller ratingCountField] stringValue] length]);
 263
 264   EXPECT_TRUE([controller userCountField]);
 265   EXPECT_NE(0u, [[[controller userCountField] stringValue] length]);
 266
 267   EXPECT_TRUE([controller storeLinkButton]);
 268   EXPECT_NE(0u, [[[controller storeLinkButton] stringValue] length]);
 269   EXPECT_NE('^',
 270             [[[controller storeLinkButton] stringValue] characterAtIndex:0]);
 271
 272   // Though we have no permissions warnings, these should still be hooked up,
 273   // just invisible.
 274   EXPECT_TRUE([controller outlineView]);
 275   EXPECT_TRUE([[[controller outlineView] enclosingScrollView] isHidden]);
 276   EXPECT_TRUE([controller warningsSeparator]);
 277   EXPECT_TRUE([[controller warningsSeparator] isHidden]);
 278 }
 279
 280 TEST_F(ExtensionInstallViewControllerTest, PostInstallPermissionsPrompt) {
 281   chrome::MockExtensionInstallPromptDelegate delegate;
 282
 283   scoped_refptr<ExtensionInstallPrompt::Prompt> prompt =
 284       chrome::BuildExtensionPostInstallPermissionsPrompt(extension_.get());
 285   ExtensionInstallPrompt::PermissionsType type =
 286       ExtensionInstallPrompt::PermissionsType::REGULAR_PERMISSIONS;
 287
 288   std::vector<base::string16> permissions;
 289   permissions.push_back(base::UTF8ToUTF16("warning 1"));
 290   prompt->SetPermissions(permissions, type);
 291   // No details provided with this permission.
 292   std::vector<base::string16> details;
 293   details.push_back(base::string16());
 294   prompt->SetPermissionsDetails(details, type);
 295
 296   base::scoped_nsobject<ExtensionInstallViewController> controller(
 297       [[ExtensionInstallViewController alloc] initWithProfile:profile()
 298                                                     navigator:browser()
 299                                                      delegate:&delegate
 300                                                        prompt:prompt]);
 301
 302   [controller view];  // Force nib load.
 303
 304   EXPECT_TRUE([controller cancelButton]);
 305   EXPECT_FALSE([controller okButton]);
 306
 307   [controller cancel:nil];
 308   EXPECT_EQ(1, delegate.abort_count());
 309 }
 310
 311 // Test that permission details show up.
 312 TEST_F(ExtensionInstallViewControllerTest, PermissionsDetails) {
 313   chrome::MockExtensionInstallPromptDelegate delegate;
 314
 315   scoped_refptr<ExtensionInstallPrompt::Prompt> prompt =
 316       chrome::BuildExtensionInstallPrompt(extension_.get());
 317   ExtensionInstallPrompt::PermissionsType type =
 318       ExtensionInstallPrompt::PermissionsType::REGULAR_PERMISSIONS;
 319
 320   std::vector<base::string16> permissions;
 321   permissions.push_back(base::UTF8ToUTF16("warning 1"));
 322   std::vector<base::string16> permissions_details;
 323   permissions_details.push_back(base::UTF8ToUTF16("Detail 1"));
 324   prompt->SetPermissions(permissions, type);
 325   prompt->SetPermissionsDetails(permissions_details, type);
 326   prompt->SetIsShowingDetails(
 327       ExtensionInstallPrompt::PERMISSIONS_DETAILS, 0, true);
 328
 329   base::scoped_nsobject<ExtensionInstallViewController> controller(
 330       [[ExtensionInstallViewController alloc] initWithProfile:profile()
 331                                                     navigator:browser()
 332                                                      delegate:&delegate
 333                                                        prompt:prompt]);
 334
 335   [controller view];  // Force nib load.
 336
 337   NSOutlineView* outlineView = [controller outlineView];
 338   EXPECT_TRUE(outlineView);
 339   EXPECT_EQ(4, [outlineView numberOfRows]);
 340   EXPECT_NSEQ(base::SysUTF16ToNSString(prompt->GetPermissionsDetails(0, type)),
 341               [[outlineView dataSource] outlineView:outlineView
 342                           objectValueForTableColumn:nil
 343                                              byItem:[outlineView itemAtRow:2]]);
 344 }