Adds Lo-Fi fallback support for new Data Reduction Proxy protocol.

content/renderer/previews_state_helper_unittest.cc

Index: content/renderer/previews_state_helper_unittest.cc
diff --git a/content/renderer/previews_state_helper_unittest.cc b/content/renderer/previews_state_helper_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..614758cde6846a6fba0c2de720ecfcbcbe213684
--- /dev/null
+++ b/content/renderer/previews_state_helper_unittest.cc
@@ -0,0 +1,121 @@
+// 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 "content/renderer/previews_state_helper.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+namespace {}  // namespaces
+
+TEST(PreviewsStateHelperTest, UpdatePreviewsStateNoTransformResponseHeaders) {
+  blink::WebURLResponse response_no_headers;
+  // Lite Page enabled but no CPCT nor page-polices => all cleared.
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_no_headers));
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(SERVER_LITE_PAGE_ON,
+                                                     response_no_headers));
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | CLIENT_LOFI_ON, response_no_headers));
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON, response_no_headers));
+
+  // Legacy Lo-Fi path (Lite Pages not enabled but Server Lo-Fi is).
+  EXPECT_EQ(SERVER_LOFI_ON, UpdatePreviewsStateFromMainFrameResponse(
+                                SERVER_LOFI_ON, response_no_headers));
+  EXPECT_EQ(SERVER_LOFI_ON | CLIENT_LOFI_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LOFI_ON | CLIENT_LOFI_ON, response_no_headers));
+
+  // Client Lo-Fi should be cleared if no Server Lo-Fi nor Lite Page.
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED, UpdatePreviewsStateFromMainFrameResponse(
+                                      CLIENT_LOFI_ON, response_no_headers));
+}
+
+TEST(PreviewsStateHelperTest, UpdatePreviewsStateLitePageTransform) {
+  blink::WebURLResponse response_with_lite_page;
+  response_with_lite_page.AddHTTPHeaderField("chrome-proxy-content-transform",
+                                             "lite-page");
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            UpdatePreviewsStateFromMainFrameResponse(SERVER_LITE_PAGE_ON,
+                                                     response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | CLIENT_LOFI_ON, response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON, response_with_lite_page));
+}
+
+TEST(PreviewsStateHelperTest, UpdatePreviewsStateLoFiFallbackPagePolicy) {
+  blink::WebURLResponse response_with_page_policy;
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy", "SomeNoise");
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy",
+                                               "page-policies=empty-image");
+
+  // No fallbacks if Server Lo-Fi not enabled.
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON, response_with_page_policy));
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED, UpdatePreviewsStateFromMainFrameResponse(
+                                      SERVER_LITE_PAGE_ON | CLIENT_LOFI_ON,
+                                      response_with_page_policy));
+
+  // Lo-Fi fallbacks if Server Lo-Fi enabled and get empty-image directive.
+  EXPECT_EQ(SERVER_LOFI_ON | CLIENT_LOFI_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(SERVER_LOFI_ON, UpdatePreviewsStateFromMainFrameResponse(
+                                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                                response_with_page_policy));
+}
+
+TEST(PreviewsStateHelperTest, UpdatePreviewsStateLoFiFallbackAndOtherPolicies) {
+  blink::WebURLResponse response_with_page_policy;
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy", "Prefix=Noise");
+  response_with_page_policy.AddHTTPHeaderField(
+      "chrome-proxy", "page-policies=new-hotness|empty-image|newer-hotness");
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy",
+                                               "Suffix=More|Noise");
+
+  // Lo-Fi fallbacks if Server Lo-Fi enabled and get empty-image directive.
+  EXPECT_EQ(SERVER_LOFI_ON | CLIENT_LOFI_ON,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(SERVER_LOFI_ON, UpdatePreviewsStateFromMainFrameResponse(
+                                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                                response_with_page_policy));
+}
+
+TEST(PreviewsStateHelperTest, UpdatePreviewsStateNoKnownPolicies) {
+  blink::WebURLResponse response_with_page_policy;
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy", "Prefix=Noise");
+  response_with_page_policy.AddHTTPHeaderField(
+      "chrome-proxy",
+      "page-policies=new-hotness|new-empty-image|newer-hotness");
+  response_with_page_policy.AddHTTPHeaderField("chrome-proxy",
+                                               "Suffix=More|Noise");
+
+  // Lo-Fi fallbacks if Server Lo-Fi enabled and get empty-image directive.
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED,
+            UpdatePreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(PREVIEWS_UNSPECIFIED, UpdatePreviewsStateFromMainFrameResponse(
+                                      SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                                      response_with_page_policy));
+}
+
+}  // namespace content