From 103ac3f582665922678ff92e8b93f09bc82ebd98 Mon Sep 17 00:00:00 2001
From: Kevin Kuehler <kevin@segfault.fun>
Date: Tue, 21 Sep 2021 21:44:25 -0700
Subject: [PATCH] Add test to catch Speedreader Icon regressions

---
 .../speedreader/speedreader_browsertest.cc    | 39 +++++++++++++++++++
 test/data/articles/fake.html                  |  7 ++++
 2 files changed, 46 insertions(+)
 create mode 100644 test/data/articles/fake.html

diff --git a/browser/speedreader/speedreader_browsertest.cc b/browser/speedreader/speedreader_browsertest.cc
index fbef1eb7ba79..1a73bb4924e5 100644
--- a/browser/speedreader/speedreader_browsertest.cc
+++ b/browser/speedreader/speedreader_browsertest.cc
@@ -17,6 +17,10 @@
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/browser_list.h"
+#include "chrome/browser/ui/page_action/page_action_icon_type.h"
+#include "chrome/browser/ui/views/frame/browser_view.h"
+#include "chrome/browser/ui/views/frame/toolbar_button_provider.h"
+#include "chrome/browser/ui/views/page_action/page_action_icon_view.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/keep_alive_registry/keep_alive_types.h"
@@ -30,8 +34,19 @@
 #include "net/test/embedded_test_server/embedded_test_server.h"
 
 const char kTestHost[] = "theguardian.com";
+
+// The path is not readable. None of the URL components match the heuristic so
+// the SpeedreaderThrottle is never created.
 const char kTestPageSimple[] = "/simple.html";
+
+// The URL path appears readable and the page has text content. This is used to
+// test distillation in the common case.
 const char kTestPageReadable[] = "/articles/guardian.html";
+
+// The URL path appears readable but the page has no text content. This is used
+// to test the case where Speedreader aborts.
+const char kTestPageFakeReadable[] = "/articles/fake.html";
+
 const base::FilePath::StringPieceType kTestWhitelist =
     FILE_PATH_LITERAL("speedreader_whitelist.json");
 
@@ -116,6 +131,12 @@ class SpeedReaderBrowserTest : public InProcessBrowserTest {
         ui_test_utils::BROWSER_TEST_WAIT_FOR_LOAD_STOP);
   }
 
+  PageActionIconView* SpeedreaderIcon() {
+    return BrowserView::GetBrowserViewForBrowser(browser())
+        ->toolbar_button_provider()
+        ->GetPageActionIconView(PageActionIconType::kReaderMode);
+  }
+
  protected:
   base::test::ScopedFeatureList feature_list_;
   net::EmbeddedTestServer https_server_;
@@ -126,6 +147,7 @@ IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, RestoreSpeedreaderPage) {
   NavigateToPageSynchronously(kTestPageReadable);
   EXPECT_TRUE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_TRUE(SpeedreaderIcon()->GetVisible());
 
   Profile* profile = browser()->profile();
 
@@ -143,6 +165,7 @@ IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, RestoreSpeedreaderPage) {
   restore_observer.Wait();
   EXPECT_TRUE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_TRUE(SpeedreaderIcon()->GetVisible());
 }
 
 IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, NavigationNostickTest) {
@@ -150,16 +173,20 @@ IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, NavigationNostickTest) {
   NavigateToPageSynchronously(kTestPageSimple);
   EXPECT_FALSE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_FALSE(SpeedreaderIcon()->GetVisible());
+
   NavigateToPageSynchronously(kTestPageReadable,
                               WindowOpenDisposition::CURRENT_TAB);
   EXPECT_TRUE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_TRUE(SpeedreaderIcon()->GetVisible());
 
   // Ensure distill state doesn't stick when we back-navigate from a readable
   // page to a non-readable one.
   GoBack(browser());
   EXPECT_FALSE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_FALSE(SpeedreaderIcon()->GetVisible());
 }
 
 IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, DisableSiteWorks) {
@@ -167,10 +194,22 @@ IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, DisableSiteWorks) {
   NavigateToPageSynchronously(kTestPageReadable);
   EXPECT_TRUE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_TRUE(SpeedreaderIcon()->GetVisible());
+
   tab_helper()->MaybeToggleEnabledForSite(false);
   EXPECT_TRUE(WaitForLoadStop(ActiveWebContents()));
   EXPECT_FALSE(
       speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  // Disabled sites show the icon
+  EXPECT_TRUE(SpeedreaderIcon()->GetVisible());
+}
+
+IN_PROC_BROWSER_TEST_F(SpeedReaderBrowserTest, TestFakeReadable) {
+  ToggleSpeedreader();
+  NavigateToPageSynchronously(kTestPageFakeReadable);
+  EXPECT_FALSE(
+      speedreader::PageStateIsDistilled(tab_helper()->PageDistillState()));
+  EXPECT_FALSE(SpeedreaderIcon()->GetVisible());
 }
 
 // disabled in https://github.com/brave/brave-browser/issues/11328
diff --git a/test/data/articles/fake.html b/test/data/articles/fake.html
new file mode 100644
index 000000000000..5f6a842cbc60
--- /dev/null
+++ b/test/data/articles/fake.html
@@ -0,0 +1,7 @@
+<!-- We have this site under a readable path but it is just a video and has no text content.
+	   This is used in tests to propagate distill results to different browser services. -->
+
+<video width="256" height="240" autoplay>
+  <source src="video.mp4" type="video/mp4">
+Your browser does not support the video tag.
+</video>