| Index: chrome/browser/notifications/notification_permission_context_unittest.cc
|
| diff --git a/chrome/browser/notifications/notification_permission_context_unittest.cc b/chrome/browser/notifications/notification_permission_context_unittest.cc
|
| index ead50bf0ca0134143d47225b71603051d986333d..963d445f4eec1271b20c3e50ffb34290af2fabdc 100644
|
| --- a/chrome/browser/notifications/notification_permission_context_unittest.cc
|
| +++ b/chrome/browser/notifications/notification_permission_context_unittest.cc
|
| @@ -10,13 +10,17 @@
|
| #include "base/time/time.h"
|
| #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
|
| #include "chrome/browser/notifications/desktop_notification_profile_util.h"
|
| +#include "chrome/browser/permissions/permission_manager.h"
|
| +#include "chrome/browser/permissions/permission_manager_factory.h"
|
| #include "chrome/browser/permissions/permission_request_id.h"
|
| #include "chrome/browser/profiles/profile.h"
|
| #include "chrome/test/base/chrome_render_view_host_test_harness.h"
|
| #include "chrome/test/base/testing_profile.h"
|
| #include "components/content_settings/core/browser/host_content_settings_map.h"
|
| +#include "content/public/browser/permission_manager.h"
|
| #include "content/public/browser/render_process_host.h"
|
| #include "content/public/browser/web_contents.h"
|
| +#include "content/public/common/permission_status.mojom.h"
|
| #include "content/public/test/test_browser_thread_bundle.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| #include "url/gurl.h"
|
| @@ -24,6 +28,7 @@
|
| namespace {
|
|
|
| void DoNothing(ContentSetting content_setting) {}
|
| +void DoNothing2(content::PermissionStatus status) {}
|
|
|
| class TestNotificationPermissionContext : public NotificationPermissionContext {
|
| public:
|
| @@ -70,7 +75,28 @@ class TestNotificationPermissionContext : public NotificationPermissionContext {
|
| };
|
|
|
| class NotificationPermissionContextTest
|
| - : public ChromeRenderViewHostTestHarness {};
|
| + : public ChromeRenderViewHostTestHarness {
|
| + public:
|
| + scoped_refptr<base::TestMockTimeTaskRunner> SwitchToMockTime() {
|
| + old_task_runner_ = base::MessageLoop::current()->task_runner();
|
| + scoped_refptr<base::TestMockTimeTaskRunner> task_runner(
|
| + new base::TestMockTimeTaskRunner(base::Time::Now(),
|
| + base::TimeTicks::Now()));
|
| + base::MessageLoop::current()->SetTaskRunner(task_runner);
|
| + return task_runner;
|
| + }
|
| +
|
| + void TearDown() override {
|
| + if (old_task_runner_) {
|
| + base::MessageLoop::current()->SetTaskRunner(old_task_runner_);
|
| + old_task_runner_ = nullptr;
|
| + }
|
| + ChromeRenderViewHostTestHarness::TearDown();
|
| + }
|
| +
|
| + private:
|
| + scoped_refptr<base::SingleThreadTaskRunner> old_task_runner_;
|
| +};
|
|
|
| } // namespace
|
|
|
| @@ -127,12 +153,7 @@ TEST_F(NotificationPermissionContextTest, TestDenyInIncognitoAfterDelay) {
|
| web_contents()->GetMainFrame()->GetRoutingID(),
|
| -1);
|
|
|
| - scoped_refptr<base::SingleThreadTaskRunner> old_task_runner(
|
| - base::MessageLoop::current()->task_runner());
|
| - scoped_refptr<base::TestMockTimeTaskRunner> task_runner(
|
| - new base::TestMockTimeTaskRunner(base::Time::Now(),
|
| - base::TimeTicks::Now()));
|
| - base::MessageLoop::current()->SetTaskRunner(task_runner);
|
| + scoped_refptr<base::TestMockTimeTaskRunner> task_runner(SwitchToMockTime());
|
|
|
| ASSERT_EQ(0, permission_context.permission_set_count());
|
| ASSERT_FALSE(permission_context.last_permission_set_persisted());
|
| @@ -185,8 +206,39 @@ TEST_F(NotificationPermissionContextTest, TestDenyInIncognitoAfterDelay) {
|
| permission_context.last_permission_set_setting());
|
| EXPECT_EQ(CONTENT_SETTING_BLOCK,
|
| permission_context.GetContentSettingFromMap(url, url));
|
| +}
|
|
|
| - base::MessageLoop::current()->SetTaskRunner(old_task_runner);
|
| +// Tests that navigating cancels incognito permission requests without crashing.
|
| +TEST_F(NotificationPermissionContextTest, TestCancelledIncognitoRequest) {
|
| + TestNotificationPermissionContext permission_context(
|
| + profile()->GetOffTheRecordProfile());
|
| + GURL url("https://www.example.com");
|
| + NavigateAndCommit(url);
|
| +
|
| + const PermissionRequestID id(web_contents()->GetRenderProcessHost()->GetID(),
|
| + web_contents()->GetMainFrame()->GetRoutingID(),
|
| + -1);
|
| +
|
| + scoped_refptr<base::TestMockTimeTaskRunner> task_runner(SwitchToMockTime());
|
| +
|
| + content::PermissionManager* permission_manager =
|
| + PermissionManagerFactory::GetForProfile(
|
| + profile()->GetOffTheRecordProfile());
|
| +
|
| + // Request and cancel the permission via PermissionManager. That way if
|
| + // https://crbug.com/586944 regresses, then as well as the EXPECT_EQs below
|
| + // failing, PermissionManager::OnPermissionsRequestResponseStatus will crash.
|
| + int request_id = permission_manager->RequestPermission(
|
| + content::PermissionType::NOTIFICATIONS, web_contents()->GetMainFrame(),
|
| + url.GetOrigin(), true /* user_gesture */, base::Bind(&DoNothing2));
|
| +
|
| + permission_manager->CancelPermissionRequest(request_id);
|
| +
|
| + task_runner->FastForwardBy(base::TimeDelta::FromDays(1));
|
| +
|
| + EXPECT_EQ(0, permission_context.permission_set_count());
|
| + EXPECT_EQ(CONTENT_SETTING_ASK,
|
| + permission_context.GetContentSettingFromMap(url, url));
|
| }
|
|
|
| // Tests how multiple parallel permission requests get auto-denied in incognito.
|
| @@ -204,12 +256,7 @@ TEST_F(NotificationPermissionContextTest, TestParallelDenyInIncognito) {
|
| web_contents()->GetMainFrame()->GetRoutingID(),
|
| 1);
|
|
|
| - scoped_refptr<base::SingleThreadTaskRunner> old_task_runner(
|
| - base::MessageLoop::current()->task_runner());
|
| - scoped_refptr<base::TestMockTimeTaskRunner> task_runner(
|
| - new base::TestMockTimeTaskRunner(base::Time::Now(),
|
| - base::TimeTicks::Now()));
|
| - base::MessageLoop::current()->SetTaskRunner(task_runner);
|
| + scoped_refptr<base::TestMockTimeTaskRunner> task_runner(SwitchToMockTime());
|
|
|
| ASSERT_EQ(0, permission_context.permission_set_count());
|
| ASSERT_FALSE(permission_context.last_permission_set_persisted());
|
| @@ -252,6 +299,4 @@ TEST_F(NotificationPermissionContextTest, TestParallelDenyInIncognito) {
|
| permission_context.last_permission_set_setting());
|
| EXPECT_EQ(CONTENT_SETTING_BLOCK,
|
| permission_context.GetContentSettingFromMap(url, url));
|
| -
|
| - base::MessageLoop::current()->SetTaskRunner(old_task_runner);
|
| }
|
|
|