MD Settings: Amend SiteSettingsHandlerTest.Origins to test "Site Details" code.

chrome/browser/ui/webui/settings/site_settings_handler_unittest.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/ui/webui/settings/site_settings_handler.h"
 
 #include <memory>
 
 #include "base/test/histogram_tester.h"
 #include "chrome/browser/chrome_notification_types.h"
@@ skipping 81 matching lines @@
     const content::TestWebUI::CallData& data = *web_ui()->call_data().back();
     EXPECT_EQ("cr.webUIResponse", data.function_name());
 
     std::string callback_id;
     ASSERT_TRUE(data.arg1()->GetAsString(&callback_id));
     EXPECT_EQ(kCallbackId, callback_id);
     bool success = false;
     ASSERT_TRUE(data.arg2()->GetAsBoolean(&success));
     ASSERT_TRUE(success);
 
-    const base::ListValue* exceptions;
-    ASSERT_TRUE(data.arg3()->GetAsList(&exceptions));
-    EXPECT_EQ(1U, exceptions->GetSize());
     const base::DictionaryValue* exception;
-    ASSERT_TRUE(exceptions->GetDictionary(0, &exception));
+    if (data.arg3()->is_list()) {
+      const base::ListValue* exceptions;
+      ASSERT_TRUE(data.arg3()->GetAsList(&exceptions));
+      EXPECT_EQ(1U, exceptions->GetSize());
+      ASSERT_TRUE(exceptions->GetDictionary(0, &exception));
+    } else {
+      data.arg3()->GetAsDictionary(&exception);

[raymes, 2017/06/26 04:12:27]

Could you add a comment describing these 2 cases? It's quite subtle - I think
it's basically handling the 2 cases for the exception list and
getCategoryPermissionListForOrigin (unpacking the values of some structure in
the page in each case)?

[Patti Lor, 2017/06/26 08:10:54]

Done.

+    }
     std::string origin, embedding_origin, display_name, setting, source;
     ASSERT_TRUE(exception->GetString(site_settings::kOrigin, &origin));
     ASSERT_EQ(expected_origin, origin);
     ASSERT_TRUE(
         exception->GetString(site_settings::kDisplayName, &display_name));
     ASSERT_EQ(expected_display_name, display_name);
     ASSERT_TRUE(exception->GetString(
         site_settings::kEmbeddingOrigin, &embedding_origin));
     ASSERT_EQ(expected_embedding, embedding_origin);
     ASSERT_TRUE(exception->GetString(site_settings::kSetting, &setting));
@@ skipping 148 matching lines @@
     set_args.AppendBoolean(false);  // Incognito.
     base::HistogramTester histograms;
     handler()->HandleSetCategoryPermissionForOrigin(&set_args);
     EXPECT_EQ(1U, web_ui()->call_data().size());
     histograms.ExpectTotalCount(kUmaBase, 1);
     histograms.ExpectTotalCount(kUmaBase + ".Allowed", 0);
     histograms.ExpectTotalCount(kUmaBase + ".Blocked", 1);
     histograms.ExpectTotalCount(kUmaBase + ".Reset", 0);
   }
 
-  // Verify the change was successful.
-  base::ListValue listArgs;
-  listArgs.AppendString(kCallbackId);
-  listArgs.AppendString("notifications");
-  handler()->HandleGetExceptionList(&listArgs);
+  // If the change was successful, it should show up as an exception as well as
+  // in the response for a specific request to getCategoryPermissionForOrigin.
+  // Check getCategoryPermissionForOrigin.
+  base::ListValue getCategoryPermissionArgs;
+  getCategoryPermissionArgs.AppendString(kCallbackId);
+  getCategoryPermissionArgs.AppendString("notifications");
+  getCategoryPermissionArgs.AppendString(google);
+  handler()->HandleGetCategoryPermissionForOrigin(&getCategoryPermissionArgs);
   ValidateOrigin(google, google, google, "block", "preference", 2U);
 
+  // Check getExceptionList.
+  base::ListValue getExceptionListArgs;
+  getExceptionListArgs.AppendString(kCallbackId);
+  getExceptionListArgs.AppendString("notifications");
+  handler()->HandleGetExceptionList(&getExceptionListArgs);
+  ValidateOrigin(google, google, google, "block", "preference", 3U);
+
   {
     // Reset things back to how they were.
     base::ListValue reset_args;
     reset_args.AppendString(google);
     reset_args.AppendString(google);
     reset_args.AppendString("notifications");
     reset_args.AppendBoolean(false);  // Incognito.
     base::HistogramTester histograms;
     handler()->HandleResetCategoryPermissionForOrigin(&reset_args);
-    EXPECT_EQ(3U, web_ui()->call_data().size());
+    EXPECT_EQ(4U, web_ui()->call_data().size());
     histograms.ExpectTotalCount(kUmaBase, 1);
     histograms.ExpectTotalCount(kUmaBase + ".Allowed", 0);
     histograms.ExpectTotalCount(kUmaBase + ".Blocked", 0);
     histograms.ExpectTotalCount(kUmaBase + ".Reset", 1);
   }
 
   // Verify the reset was successful.
-  handler()->HandleGetExceptionList(&listArgs);
-  ValidateNoOrigin(4U);
+  handler()->HandleGetExceptionList(&getExceptionListArgs);
+  ValidateNoOrigin(5U);
+
+  handler()->HandleGetCategoryPermissionForOrigin(&getCategoryPermissionArgs);
+  // Ask is the currently the built-in Chrome default for notifications.
+  ValidateOrigin(google, google, google, "ask", "preference", 6U);
 }
 
 TEST_F(SiteSettingsHandlerTest, ExceptionHelpers) {
   ContentSettingsPattern pattern =
       ContentSettingsPattern::FromString("[*.]google.com");
   std::unique_ptr<base::DictionaryValue> exception =
       site_settings::GetExceptionForPage(pattern, pattern, pattern.ToString(),
                                          CONTENT_SETTING_BLOCK, "preference",
                                          false);
 
@@ skipping 98 matching lines @@
   args.AppendString("http://www.google.com");
   handler()->HandleRemoveZoomLevel(&args);
   ValidateZoom("", "", 3U);
 
   double default_level = host_zoom_map->GetDefaultZoomLevel();
   double level = host_zoom_map->GetZoomLevelForHostAndScheme("http", host);
   EXPECT_EQ(default_level, level);
 }
 
 }  // namespace settings