Add base and bonus scores to site-engagement WebUI.

chrome/browser/ui/webui/engagement/site_engagement_ui_browsertest.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 "base/strings/string_util.h"
 #include "chrome/browser/engagement/site_engagement_service.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/webui/chrome_web_ui_controller_factory.h"
 #include "chrome/common/url_constants.h"
@@ skipping 78 matching lines @@
         browser()->tab_strip_model()->GetActiveWebContents(),
         base::JoinString({"window.domAutomationController.send("
                           "    document.getElementsByClassName('origin-cell')[",
                           base::IntToString(index), "].innerHTML);"},
                          ""),
         &origin_url));
 
     return origin_url;
   }
 
-  // Returns the stringified score at the specified zero-based row index.
-  std::string ScoreAtRow(int index) {
+  // Returns the stringified base score at the specified zero-based row index.
+  std::string BaseScoreAtRow(int index) {
     EXPECT_TRUE(page_is_populated_);
 
     std::string score_string;
     EXPECT_TRUE(content::ExecuteScriptAndExtractString(
         browser()->tab_strip_model()->GetActiveWebContents(),
-        base::JoinString({"window.domAutomationController.send("
-                          "    document.getElementsByClassName('score-input')[",
-                          base::IntToString(index), "].value);"},
-                         ""),
+        base::JoinString(
+            {"window.domAutomationController.send("
+             "    document.getElementsByClassName('base-score-input')[",
+             base::IntToString(index), "].value);"},
+            ""),
         &score_string));
 
     return score_string;
+  }
+
+  std::string BonusScoreAtRow(int index) {
+    EXPECT_TRUE(page_is_populated_);
+
+    std::string score_string;
+    EXPECT_TRUE(content::ExecuteScriptAndExtractString(
+        browser()->tab_strip_model()->GetActiveWebContents(),
+        base::JoinString(
+            {"window.domAutomationController.send("
+             "    document.getElementsByClassName('bonus-score-cell')[",
+             base::IntToString(index), "].innerHTML);"},
+            ""),
+        &score_string));
+
+    return score_string;
+  }
+
+  std::string TotalScoreAtRow(int index) {
+    EXPECT_TRUE(page_is_populated_);
+
+    std::string score_string;
+    EXPECT_TRUE(content::ExecuteScriptAndExtractString(
+        browser()->tab_strip_model()->GetActiveWebContents(),
+        base::JoinString(
+            {"window.domAutomationController.send("
+             "    document.getElementsByClassName('total-score-cell')[",
+             base::IntToString(index), "].innerHTML);"},
+            ""),
+        &score_string));
+
+    return score_string;
   }
 
  private:
   // True if the page contains site engagement data.
   bool page_is_populated_ = false;
 };
 
 IN_PROC_BROWSER_TEST_F(SiteEngagementUiBrowserTest, Basic) {
   ResetBaseScoreToMax(kExampleUrl);
 
   NavigateToWebUi();
   WaitUntilPagePopulated();
 
   EXPECT_EQ(1, NumberOfRows());
   EXPECT_EQ(kExampleUrl, OriginUrlAtRow(0));
 }
 
 IN_PROC_BROWSER_TEST_F(SiteEngagementUiBrowserTest,
                        ScoresHaveTwoDecimalPlaces) {
   ResetBaseScore(kExampleUrl, 3.14159);
 
   NavigateToWebUi();
   WaitUntilPagePopulated();
 
   EXPECT_EQ(1, NumberOfRows());
-  EXPECT_EQ("3.14", ScoreAtRow(0));
+  EXPECT_EQ("3.14", BaseScoreAtRow(0));
+  EXPECT_EQ("0", BonusScoreAtRow(0));
+  EXPECT_EQ("3.14", TotalScoreAtRow(0));
 }