| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/memory/ref_counted.h" | 6 #include "base/memory/ref_counted.h" |
| 7 #include "base/run_loop.h" | 7 #include "base/run_loop.h" |
| 8 #include "chrome/browser/extensions/extension_install_prompt.h" | 8 #include "chrome/browser/extensions/extension_install_prompt.h" |
| 9 #include "content/public/test/test_browser_thread_bundle.h" | 9 #include "content/public/test/test_browser_thread_bundle.h" |
| 10 #include "extensions/common/extension.h" | 10 #include "extensions/common/extension.h" |
| 11 #include "extensions/common/extension_builder.h" | 11 #include "extensions/common/extension_builder.h" |
| 12 #include "extensions/common/permissions/api_permission.h" | 12 #include "extensions/common/permissions/api_permission.h" |
| 13 #include "extensions/common/permissions/api_permission_set.h" | 13 #include "extensions/common/permissions/api_permission_set.h" |
| 14 #include "extensions/common/permissions/manifest_permission_set.h" | 14 #include "extensions/common/permissions/manifest_permission_set.h" |
| 15 #include "extensions/common/permissions/permission_set.h" | 15 #include "extensions/common/permissions/permission_set.h" |
| 16 #include "extensions/common/url_pattern_set.h" | 16 #include "extensions/common/url_pattern_set.h" |
| 17 #include "extensions/common/value_builder.h" | 17 #include "extensions/common/value_builder.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 19 |
| 20 namespace extensions { | 20 namespace extensions { |
| 21 | 21 |
| 22 void VerifyPromptPermissionsCallback( | 22 void VerifyPromptPermissionsCallback( |
| 23 const base::Closure& quit_closure, | 23 const base::Closure& quit_closure, |
| 24 const ExtensionInstallPrompt::ShowParams& params, | 24 const ExtensionInstallPrompt::ShowParams& params, |
| 25 ExtensionInstallPrompt::Delegate* delegate, | 25 ExtensionInstallPrompt::Delegate* delegate, |
| 26 scoped_refptr<ExtensionInstallPrompt::Prompt> install_prompt) { | 26 scoped_refptr<ExtensionInstallPrompt::Prompt> install_prompt) { |
| 27 ASSERT_TRUE(install_prompt.get()); | 27 ASSERT_TRUE(install_prompt.get()); |
| 28 EXPECT_EQ(1u, install_prompt->GetPermissionCount()); | 28 EXPECT_EQ(1u, |
| 29 install_prompt->GetPermissionCount( |
| 30 ExtensionInstallPrompt::ALL_PERMISSIONS)); |
| 29 quit_closure.Run(); | 31 quit_closure.Run(); |
| 30 } | 32 } |
| 31 | 33 |
| 32 TEST(ExtensionInstallPromptUnittest, PromptShowsPermissionWarnings) { | 34 TEST(ExtensionInstallPromptUnittest, PromptShowsPermissionWarnings) { |
| 33 content::TestBrowserThreadBundle thread_bundle; | 35 content::TestBrowserThreadBundle thread_bundle; |
| 34 APIPermissionSet api_permissions; | 36 APIPermissionSet api_permissions; |
| 35 api_permissions.insert(APIPermission::kTab); | 37 api_permissions.insert(APIPermission::kTab); |
| 36 scoped_refptr<PermissionSet> permission_set = | 38 scoped_refptr<PermissionSet> permission_set = |
| 37 new PermissionSet(api_permissions, | 39 new PermissionSet(api_permissions, |
| 38 ManifestPermissionSet(), | 40 ManifestPermissionSet(), |
| 39 URLPatternSet(), | 41 URLPatternSet(), |
| 40 URLPatternSet()); | 42 URLPatternSet()); |
| 41 scoped_refptr<const Extension> extension = | 43 scoped_refptr<const Extension> extension = |
| 42 ExtensionBuilder().SetManifest( | 44 ExtensionBuilder().SetManifest( |
| 43 DictionaryBuilder().Set("name", "foo") | 45 DictionaryBuilder().Set("name", "foo") |
| 44 .Set("version", "1.0") | 46 .Set("version", "1.0") |
| 45 .Set("manifest_version", 2) | 47 .Set("manifest_version", 2) |
| 46 .Set("description", "Random Ext")).Build(); | 48 .Set("description", "Random Ext")).Build(); |
| 47 ExtensionInstallPrompt prompt(NULL /* no web contents in this test */); | 49 ExtensionInstallPrompt prompt(NULL /* no web contents in this test */); |
| 48 base::RunLoop run_loop; | 50 base::RunLoop run_loop; |
| 49 prompt.set_callback_for_test(base::Bind(&VerifyPromptPermissionsCallback, | 51 prompt.set_callback_for_test(base::Bind(&VerifyPromptPermissionsCallback, |
| 50 run_loop.QuitClosure())); | 52 run_loop.QuitClosure())); |
| 51 prompt.ConfirmPermissions(NULL, // no delegate | 53 prompt.ConfirmPermissions(NULL, // no delegate |
| 52 extension.get(), | 54 extension.get(), |
| 53 permission_set.get()); | 55 permission_set.get()); |
| 54 run_loop.Run(); | 56 run_loop.Run(); |
| 55 } | 57 } |
| 56 | 58 |
| 57 } // namespace extensions | 59 } // namespace extensions |
| OLD | NEW |