Reset geolocation permission embargo for an origin when it becomes DSE

chrome/browser/android/search_geolocation/search_geolocation_service_unittest.cc

 // 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/android/search_geolocation/search_geolocation_service.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/callback.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/scoped_feature_list.h"
 #include "chrome/browser/android/search_geolocation/search_geolocation_disclosure_tab_helper.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/permissions/permission_decision_auto_blocker.h"
+#include "chrome/browser/permissions/permission_result.h"
 #include "chrome/common/chrome_features.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "components/content_settings/core/common/pref_names.h"
 #include "components/prefs/pref_service.h"
 #include "components/sync_preferences/testing_pref_service_syncable.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/origin.h"
@@ skipping 341 matching lines @@
   EXPECT_TRUE(SearchGeolocationDisclosureTabHelper::IsDisclosureResetForTests(
       profile()));
   SearchGeolocationDisclosureTabHelper::FakeShowingDisclosureForTests(
       profile());
 
   // Go back to google.com.au. The disclosure should again be reset.
   test_delegate()->SetDSEOrigin(kGoogleAusURL);
   EXPECT_TRUE(SearchGeolocationDisclosureTabHelper::IsDisclosureResetForTests(
       profile()));
 }
+
+TEST_F(SearchGeolocationServiceTest, Embargo) {
+  test_delegate()->SetDSEOrigin(kGoogleURL);
+
+  // Place another origin under embargo.
+  GURL google_aus_url(kGoogleAusURL);
+  PermissionDecisionAutoBlocker* auto_blocker =
+      PermissionDecisionAutoBlocker::GetForProfile(profile());
+  auto_blocker->RecordDismissAndEmbargo(google_aus_url,
+                                        CONTENT_SETTINGS_TYPE_GEOLOCATION);
+  auto_blocker->RecordDismissAndEmbargo(google_aus_url,
+                                        CONTENT_SETTINGS_TYPE_GEOLOCATION);
+  auto_blocker->RecordDismissAndEmbargo(google_aus_url,
+                                        CONTENT_SETTINGS_TYPE_GEOLOCATION);
+  PermissionResult result = auto_blocker->GetEmbargoResult(
+      GURL(kGoogleAusURL), CONTENT_SETTINGS_TYPE_GEOLOCATION);
+  EXPECT_EQ(result.source, PermissionStatusSource::MULTIPLE_DISMISSALS);
+  EXPECT_EQ(result.content_setting, CONTENT_SETTING_BLOCK);
+
+  // Now change the DSE to this origin and make sure the embargo is cleared.
+  test_delegate()->SetDSEOrigin(kGoogleAusURL);
+  result = auto_blocker->GetEmbargoResult(GURL(kGoogleAusURL),
+                                          CONTENT_SETTINGS_TYPE_GEOLOCATION);
+  EXPECT_EQ(result.source, PermissionStatusSource::UNSPECIFIED);
+  EXPECT_EQ(result.content_setting, CONTENT_SETTING_ASK);
+}