| // 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 |