[Storage] Android - ManageSpace UI, Important Origins, and CBD Dialog

chrome/browser/android/preferences/important_sites_util.cc

 // Copyright 2016 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/preferences/important_sites_util.h"
 
 #include <algorithm>
 #include <map>
 #include <set>
 
@@ skipping 66 matching lines @@
     const std::vector<std::pair<GURL, double>>& sorted_new_origins,
     size_t max_important_domains,
     std::vector<std::string>* final_list) {
   for (const auto& pair : sorted_new_origins) {
     if (final_list->size() >= max_important_domains)
       return;
     std::string registerable_domain =
         net::registry_controlled_domains::GetDomainAndRegistry(
             pair.first,
             net::registry_controlled_domains::INCLUDE_PRIVATE_REGISTRIES);
+    if (registerable_domain.empty() && pair.first.HostIsIPAddress())
+      registerable_domain = pair.first.host();
     // Just iterate to find, as we assume our size is small.
     if (std::find(final_list->begin(), final_list->end(),
                   registerable_domain) == final_list->end()) {
       final_list->push_back(registerable_domain);
     }
   }
 }
 
 }  // namespace
 
@@ skipping 21 matching lines @@
   // Otherwise we can add a set later.
   std::vector<std::string> final_list;
   // We start with notifications.
   FillTopRegisterableDomains(sorted_notification_origins, max_results,
                              &final_list);
   // And now we fill the rest with high engagement sites.
   FillTopRegisterableDomains(sorted_engagement_origins, max_results,
                              &final_list);
   return final_list;
 }
+
+void ImportantSitesUtil::MarkOriginAsImportantForTesting(Profile* profile,
+                                                         const GURL& origin) {
+  // First get data from site engagement.
+  SiteEngagementService* site_engagement_service =
+      SiteEngagementService::Get(profile);
+  site_engagement_service->ResetScoreForURL(
+      origin, SiteEngagementScore::GetMediumEngagementBoundary());
+  DCHECK(site_engagement_service->IsEngagementAtLeast(
+      origin, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
+}