1// Copyright 2014 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#include "base/bind.h" 6#include "base/memory/ref_counted.h" 7#include "base/run_loop.h" 8#include "chrome/browser/extensions/extension_install_prompt.h" 9#include "content/public/test/test_browser_thread_bundle.h" 10#include "extensions/common/extension.h" 11#include "extensions/common/extension_builder.h" 12#include "extensions/common/feature_switch.h" 13#include "extensions/common/permissions/api_permission.h" 14#include "extensions/common/permissions/api_permission_set.h" 15#include "extensions/common/permissions/manifest_permission_set.h" 16#include "extensions/common/permissions/permission_set.h" 17#include "extensions/common/url_pattern_set.h" 18#include "extensions/common/value_builder.h" 19#include "testing/gtest/include/gtest/gtest.h" 20 21namespace extensions { 22 23void VerifyPromptPermissionsCallback( 24 const base::Closure& quit_closure, 25 size_t regular_permissions_count, 26 size_t withheld_permissions_count, 27 const ExtensionInstallPrompt::ShowParams& params, 28 ExtensionInstallPrompt::Delegate* delegate, 29 scoped_refptr<ExtensionInstallPrompt::Prompt> install_prompt) { 30 ASSERT_TRUE(install_prompt.get()); 31 EXPECT_EQ(regular_permissions_count, 32 install_prompt->GetPermissionCount( 33 ExtensionInstallPrompt::REGULAR_PERMISSIONS)); 34 EXPECT_EQ(withheld_permissions_count, 35 install_prompt->GetPermissionCount( 36 ExtensionInstallPrompt::WITHHELD_PERMISSIONS)); 37 quit_closure.Run(); 38} 39 40TEST(ExtensionInstallPromptUnittest, PromptShowsPermissionWarnings) { 41 content::TestBrowserThreadBundle thread_bundle; 42 APIPermissionSet api_permissions; 43 api_permissions.insert(APIPermission::kTab); 44 scoped_refptr<PermissionSet> permission_set = 45 new PermissionSet(api_permissions, 46 ManifestPermissionSet(), 47 URLPatternSet(), 48 URLPatternSet()); 49 scoped_refptr<const Extension> extension = 50 ExtensionBuilder().SetManifest( 51 DictionaryBuilder().Set("name", "foo") 52 .Set("version", "1.0") 53 .Set("manifest_version", 2) 54 .Set("description", "Random Ext")).Build(); 55 ExtensionInstallPrompt prompt(NULL /* no web contents in this test */); 56 base::RunLoop run_loop; 57 prompt.set_callback_for_test( 58 base::Bind(&VerifyPromptPermissionsCallback, 59 run_loop.QuitClosure(), 60 1u, // |regular_permissions_count|. 61 0u)); // |withheld_permissions_count|. 62 prompt.ConfirmPermissions(NULL, // no delegate 63 extension.get(), 64 permission_set.get()); 65 run_loop.Run(); 66} 67 68TEST(ExtensionInstallPromptUnittest, PromptShowsWithheldPermissions) { 69 content::TestBrowserThreadBundle thread_bundle; 70 71 // Enable consent flag so that <all_hosts> permissions get withheld. 72 FeatureSwitch::ScopedOverride enable_scripts_switch( 73 FeatureSwitch::scripts_require_action(), true); 74 75 ListBuilder permissions; 76 permissions.Append("http://*/*"); 77 permissions.Append("http://www.google.com/"); 78 permissions.Append("tabs"); 79 scoped_refptr<const Extension> extension = 80 ExtensionBuilder().SetManifest( 81 DictionaryBuilder().Set("name", "foo") 82 .Set("version", "1.0") 83 .Set("manifest_version", 2) 84 .Set("description", "Random Ext") 85 .Set("permissions", permissions)).Build(); 86 ExtensionInstallPrompt prompt(NULL /* no web contents in this test */); 87 base::RunLoop run_loop; 88 89 // We expect <all_hosts> to be withheld, but http://www.google.com/ and tabs 90 // permissions should be granted as regular permissions. 91 prompt.ConfirmInstall( 92 NULL, 93 extension.get(), 94 base::Bind(&VerifyPromptPermissionsCallback, 95 run_loop.QuitClosure(), 96 2u, // |regular_permissions_count|. 97 1u)); // |withheld_permissions_count|. 98 run_loop.Run(); 99} 100 101} // namespace extensions 102