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..cd6c38f8f22e4c41c36ccd41085f73fc1717d53e
--- /dev/null
+++ b/content/renderer/previews_state_helper_unittest.cc
@@ -0,0 +1,146 @@
+// 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 "content/public/common/previews_state.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/WebURLResponse.h"
+
+namespace content {
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateNoServerTransformRequested) {
+  blink::WebURLResponse response_no_headers;
+
+  // No transforms specified.
+  EXPECT_EQ(PREVIEWS_OFF, GetPreviewsStateFromMainFrameResponse(
+                              PREVIEWS_UNSPECIFIED, response_no_headers));
+
+  // Client Lo-Fi preserved if no Server Lo-Fi nor Lite Page.
+  EXPECT_EQ(CLIENT_LOFI_ON, GetPreviewsStateFromMainFrameResponse(
+                                CLIENT_LOFI_ON, response_no_headers));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateNoTransformResponseHeaders) {
+  blink::WebURLResponse response_no_headers;
+  // Lite Page enabled but no CPCT nor page-polices => all cleared.
+  EXPECT_EQ(PREVIEWS_OFF,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_no_headers));
+  EXPECT_EQ(PREVIEWS_OFF, GetPreviewsStateFromMainFrameResponse(
+                              SERVER_LITE_PAGE_ON, response_no_headers));
+  EXPECT_EQ(PREVIEWS_OFF,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | CLIENT_LOFI_ON, response_no_headers));
+  EXPECT_EQ(PREVIEWS_OFF,
+            GetPreviewsStateFromMainFrameResponse(
+                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, GetPreviewsStateFromMainFrameResponse(
+                                SERVER_LOFI_ON, response_no_headers));
+  EXPECT_EQ(SERVER_LOFI_ON | CLIENT_LOFI_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LOFI_ON | CLIENT_LOFI_ON, response_no_headers));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateLitePageTransform) {
+  blink::WebURLResponse response_with_lite_page;
+  response_with_lite_page.AddHTTPHeaderField("chrome-proxy-content-transform",
+                                             "lite-page");
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            GetPreviewsStateFromMainFrameResponse(SERVER_LITE_PAGE_ON,
+                                                  response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | CLIENT_LOFI_ON, response_with_lite_page));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON, response_with_lite_page));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateLitePageAndLoFiTransforms) {
+  blink::WebURLResponse response_with_lite_page_and_page_policy;
+  response_with_lite_page_and_page_policy.AddHTTPHeaderField(
+      "chrome-proxy-content-transform", "lite-page");
+  response_with_lite_page_and_page_policy.AddHTTPHeaderField(
+      "Chrome-Proxy", "Page-Policies=Empty-Image");
+
+  response_with_lite_page_and_page_policy.AddHTTPHeaderField(
+      "Chrome-Proxy", "Page-Policies=Empty-Image");
+  EXPECT_EQ(SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_lite_page_and_page_policy));
+  EXPECT_EQ(SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                response_with_lite_page_and_page_policy));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateLoFiFallbackPagePolicy) {
+  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_OFF, GetPreviewsStateFromMainFrameResponse(
+                              SERVER_LITE_PAGE_ON, response_with_page_policy));
+  EXPECT_EQ(PREVIEWS_OFF, GetPreviewsStateFromMainFrameResponse(
+                              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,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(SERVER_LOFI_ON, GetPreviewsStateFromMainFrameResponse(
+                                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                                response_with_page_policy));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateLoFiFallbackAndOtherPolicies) {
+  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,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(SERVER_LOFI_ON, GetPreviewsStateFromMainFrameResponse(
+                                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                                response_with_page_policy));
+}
+
+TEST(PreviewsStateHelperTest, GetPreviewsStateNoKnownPolicies) {
+  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_OFF,
+            GetPreviewsStateFromMainFrameResponse(
+                SERVER_LITE_PAGE_ON | SERVER_LOFI_ON | CLIENT_LOFI_ON,
+                response_with_page_policy));
+  EXPECT_EQ(PREVIEWS_OFF, GetPreviewsStateFromMainFrameResponse(
+                              SERVER_LITE_PAGE_ON | SERVER_LOFI_ON,
+                              response_with_page_policy));
+}
+
+}  // namespace content