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