| // Copyright 2017 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/chromeos/extensions/active_tab_permission_granter_delegate_chromeos.h" |
| |
| #include <memory> |
| #include <string> |
| |
| #include "base/run_loop.h" |
| #include "chrome/browser/chromeos/extensions/public_session_permission_helper.h" |
| #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| #include "chromeos/login/login_state/scoped_test_public_session_login_state.h" |
| #include "extensions/browser/extension_dialog_auto_confirm.h" |
| #include "extensions/common/extension.h" |
| #include "extensions/common/extension_builder.h" |
| #include "extensions/common/value_builder.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace extensions { |
| |
| namespace { |
| |
| const char kWhitelistedId[] = "cbkkbcmdlboombapidmoeolnmdacpkch"; |
| // Use an extension ID that will never be whitelisted. |
| const char kNonWhitelistedId[] = "bogus"; |
| |
| scoped_refptr<const Extension> CreateExtension(const std::string& id) { |
| return ExtensionBuilder("test").SetID(id).Build(); |
| } |
| |
| } // namespace |
| |
| class ActiveTabPermissionGranterDelegateChromeOSTest |
| : public ChromeRenderViewHostTestHarness { |
| protected: |
| void SetUp() override; |
| void TearDown() override; |
| |
| ActiveTabPermissionGranterDelegateChromeOS delegate_; |
| std::unique_ptr<chromeos::ScopedTestPublicSessionLoginState> login_state_; |
| }; |
| |
| void ActiveTabPermissionGranterDelegateChromeOSTest::SetUp() { |
| ChromeRenderViewHostTestHarness::SetUp(); |
| login_state_.reset(new chromeos::ScopedTestPublicSessionLoginState()); |
| } |
| |
| void ActiveTabPermissionGranterDelegateChromeOSTest::TearDown() { |
| login_state_.reset(); |
| permission_helper::ResetPermissionsForTesting(); |
| ChromeRenderViewHostTestHarness::TearDown(); |
| } |
| |
| TEST_F(ActiveTabPermissionGranterDelegateChromeOSTest, GrantedForWhitelisted) { |
| auto extension = CreateExtension(kWhitelistedId); |
| EXPECT_TRUE(delegate_.ShouldGrantActiveTabOrPrompt(extension.get(), nullptr)); |
| } |
| |
| TEST_F(ActiveTabPermissionGranterDelegateChromeOSTest, |
| RejectedForNonWhitelisted) { |
| auto extension = CreateExtension(kNonWhitelistedId); |
| // Deny the permission request. |
| ScopedTestDialogAutoConfirm auto_confirm(ScopedTestDialogAutoConfirm::CANCEL); |
| // First request is always rejected (by design). |
| EXPECT_FALSE( |
| delegate_.ShouldGrantActiveTabOrPrompt(extension.get(), nullptr)); |
| // Spin the loop, allowing the dialog to be resolved. |
| base::RunLoop().RunUntilIdle(); |
| // Dialog result is propagated here, permission request is rejected. |
| EXPECT_FALSE( |
| delegate_.ShouldGrantActiveTabOrPrompt(extension.get(), nullptr)); |
| } |
| |
| TEST_F(ActiveTabPermissionGranterDelegateChromeOSTest, |
| GrantedForNonWhitelisted) { |
| auto extension = CreateExtension(kNonWhitelistedId); |
| // Allow the permission request. |
| ScopedTestDialogAutoConfirm auto_confirm(ScopedTestDialogAutoConfirm::ACCEPT); |
| EXPECT_FALSE( |
| delegate_.ShouldGrantActiveTabOrPrompt(extension.get(), nullptr)); |
| base::RunLoop().RunUntilIdle(); |
| // The permission request is granted now. |
| EXPECT_TRUE(delegate_.ShouldGrantActiveTabOrPrompt(extension.get(), nullptr)); |
| } |
| |
| } // namespace extensions |