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 |