Audit the last usage of Geolocation and Notification permissions.

chrome/browser/geolocation/geolocation_permission_context_unittest.cc

 // Copyright (c) 2012 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/geolocation/geolocation_permission_context.h"
 
 #include <set>
 #include <string>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/containers/hash_tables.h"
 #include "base/memory/scoped_vector.h"
 #include "base/run_loop.h"
 #include "base/synchronization/waitable_event.h"
+#include "base/test/simple_test_clock.h"
+#include "base/time/clock.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/content_settings/permission_request_id.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
 #include "chrome/browser/geolocation/geolocation_permission_context_factory.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/infobars/core/confirm_infobar_delegate.h"
@@ skipping 663 matching lines @@
   EXPECT_FALSE(infobar_delegate->ShouldExpire(
       InfoBarService::NavigationDetailsFromLoadCommittedDetails(details)));
   // Ensure the infobar will expire when we commit the pending navigation.
   details.entry = web_contents()->GetController().GetActiveEntry();
   EXPECT_TRUE(infobar_delegate->ShouldExpire(
       InfoBarService::NavigationDetailsFromLoadCommittedDetails(details)));
 
   // Delete the tab contents.
   DeleteContents();
 }
+
+TEST_F(GeolocationPermissionContextTests, LastUsageAudited) {
+  GURL requesting_frame("http://www.example.com/geolocation");
+  NavigateAndCommit(requesting_frame);
+
+  base::SimpleTestClock* test_clock_ = new base::SimpleTestClock;
+  test_clock_->SetNow(base::Time::UnixEpoch() +
+                      base::TimeDelta::FromSeconds(10));
+  profile()->GetHostContentSettingsMap()->SetPrefClockForTesting(test_clock_);
+
+  // The permission shouldn't have been used yet.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame.GetOrigin(),
+                               requesting_frame.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            0);
+
+  EXPECT_EQ(0U, infobar_service()->infobar_count());
+  RequestGeolocationPermission(web_contents(), RequestID(0), requesting_frame);
+  ASSERT_EQ(1U, infobar_service()->infobar_count());
+  infobars::InfoBar* infobar = infobar_service()->infobar_at(0);
+  ConfirmInfoBarDelegate* infobar_delegate =
+      infobar->delegate()->AsConfirmInfoBarDelegate();
+  ASSERT_TRUE(infobar_delegate);
+  infobar_delegate->Accept();
+  CheckTabContentsState(requesting_frame, CONTENT_SETTING_ALLOW);
+  CheckPermissionMessageSent(0, true);
+
+  // Permission has been used at the starting time.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame.GetOrigin(),
+                               requesting_frame.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            10);
+
+  test_clock_->Advance(base::TimeDelta::FromSeconds(3));
+  RequestGeolocationPermission(web_contents(), RequestID(0), requesting_frame);
+
+  // Permission has been used three seconds later.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame.GetOrigin(),
+                               requesting_frame.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            13);
+}
+
+TEST_F(GeolocationPermissionContextTests, LastUsageAuditedMultipleFrames) {
+  base::SimpleTestClock* test_clock_ = new base::SimpleTestClock;
+  test_clock_->SetNow(base::Time::UnixEpoch() +
+                      base::TimeDelta::FromSeconds(10));
+  profile()->GetHostContentSettingsMap()->SetPrefClockForTesting(test_clock_);
+
+  GURL requesting_frame_0("http://www.example.com/geolocation");
+  GURL requesting_frame_1("http://www.example-2.com/geolocation");
+  // The permission shouldn't have been used yet.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_0.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            0);
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_1.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            0);
+
+  NavigateAndCommit(requesting_frame_0);
+  EXPECT_EQ(0U, infobar_service()->infobar_count());
+
+  // Request permission for two frames.
+  RequestGeolocationPermission(
+      web_contents(), RequestID(0), requesting_frame_0);
+  RequestGeolocationPermission(
+      web_contents(), RequestID(1), requesting_frame_1);
+
+  // Ensure only one infobar is created.
+  ASSERT_EQ(1U, infobar_service()->infobar_count());
+  infobars::InfoBar* infobar_0 = infobar_service()->infobar_at(0);
+  ConfirmInfoBarDelegate* infobar_delegate_0 =
+      infobar_0->delegate()->AsConfirmInfoBarDelegate();
+
+  // Accept the first frame.
+  infobar_delegate_0->Accept();
+  CheckTabContentsState(requesting_frame_0, CONTENT_SETTING_ALLOW);
+  CheckPermissionMessageSent(0, true);
+  infobar_service()->RemoveInfoBar(infobar_0);
+
+  // Verify that accepting the first didn't accept because it's embedder
+  // in the other.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_0.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            10);
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_1.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            0);
+
+  ASSERT_EQ(1U, infobar_service()->infobar_count());
+  infobars::InfoBar* infobar_1 = infobar_service()->infobar_at(0);
+  ConfirmInfoBarDelegate* infobar_delegate_1 =
+      infobar_1->delegate()->AsConfirmInfoBarDelegate();
+
+  test_clock_->Advance(base::TimeDelta::FromSeconds(1));
+
+  // Allow the second frame.
+  infobar_delegate_1->Accept();
+  CheckTabContentsState(requesting_frame_1, CONTENT_SETTING_ALLOW);
+  CheckPermissionMessageSent(0, true);
+  infobar_service()->RemoveInfoBar(infobar_1);
+
+  // Verify that the times are different.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_0.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            10);
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_1.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            11);
+
+  test_clock_->Advance(base::TimeDelta::FromSeconds(2));
+  RequestGeolocationPermission(
+      web_contents(), RequestID(0), requesting_frame_0);
+
+  // Verify that requesting permission in one frame doesn't update other where
+  // it is the embedder.
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_0.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            13);
+  EXPECT_EQ(profile()
+                ->GetHostContentSettingsMap()
+                ->GetLastUsage(requesting_frame_1.GetOrigin(),
+                               requesting_frame_0.GetOrigin(),
+                               CONTENT_SETTINGS_TYPE_GEOLOCATION)
+                .ToDoubleT(),
+            11);
+}