| /* |
| * Copyright 2013 Google Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| // Author: bharathbhushan@google.com (Bharath Bhushan Kowshik Raghupathi) |
| |
| #include "net/instaweb/rewriter/public/critical_line_info_finder.h" |
| |
| #include "net/instaweb/rewriter/critical_line_info.pb.h" |
| #include "net/instaweb/rewriter/public/rewrite_driver.h" |
| #include "net/instaweb/rewriter/public/rewrite_options.h" |
| #include "net/instaweb/rewriter/public/rewrite_test_base.h" |
| #include "net/instaweb/rewriter/public/server_context.h" |
| #include "net/instaweb/util/public/mock_property_page.h" |
| #include "net/instaweb/util/public/property_cache.h" |
| #include "pagespeed/kernel/base/basictypes.h" |
| #include "pagespeed/kernel/base/gtest.h" |
| #include "pagespeed/kernel/base/proto_util.h" |
| #include "pagespeed/kernel/base/string.h" |
| #include "pagespeed/kernel/http/http_names.h" |
| #include "pagespeed/kernel/http/request_headers.h" |
| |
| namespace net_instaweb { |
| |
| class CriticalLineInfoFinderTest : public RewriteTestBase { |
| protected: |
| GoogleString panel_start(int index) { |
| return rewrite_driver()->critical_line_info()->panels(index).start_xpath(); |
| } |
| |
| GoogleString panel_end(int index) { |
| return rewrite_driver()->critical_line_info() |
| ->panels(index).end_marker_xpath(); |
| } |
| |
| CriticalLineInfo CreateCriticalLineInfo( |
| const GoogleString& start_xpath1, |
| const GoogleString& end_xpath1, |
| const GoogleString& start_xpath2, |
| const GoogleString& end_xpath2) { |
| CriticalLineInfo config; |
| Panel* panel = config.add_panels(); |
| panel->set_start_xpath(start_xpath1); |
| panel->set_end_marker_xpath(end_xpath1); |
| panel = config.add_panels(); |
| panel->set_start_xpath(start_xpath2); |
| panel->set_end_marker_xpath(end_xpath2); |
| return config; |
| } |
| |
| void WriteToPcache(CriticalLineInfo config) { |
| server_context()->page_property_cache()->set_enabled(true); |
| |
| const PropertyCache::Cohort* cohort = finder_->cohort(); |
| rewrite_driver()->set_property_page(NewMockPage("http://www.test.com")); |
| server_context()->page_property_cache()->Read( |
| rewrite_driver()->property_page()); |
| |
| GoogleString buf; |
| StringOutputStream sstream(&buf); |
| ASSERT_TRUE(config.SerializeToZeroCopyStream(&sstream)); |
| PropertyPage* page = rewrite_driver()->property_page(); |
| ASSERT_TRUE(page != NULL); |
| ASSERT_EQ("beacon_cohort", cohort->name()); |
| page->GetProperty(cohort, |
| CriticalLineInfoFinder::kCriticalLineInfoPropertyName); |
| page->UpdateValue( |
| cohort, CriticalLineInfoFinder::kCriticalLineInfoPropertyName, buf); |
| rewrite_driver()->property_page()->WriteCohort(cohort); |
| } |
| |
| protected: |
| virtual void SetUp() { |
| RewriteTestBase::SetUp(); |
| const PropertyCache::Cohort* beacon_cohort = |
| SetupCohort(page_property_cache(), RewriteDriver::kBeaconCohort); |
| server_context()->set_beacon_cohort(beacon_cohort); |
| finder_ = new CriticalLineInfoFinder(beacon_cohort); |
| server_context()->set_critical_line_info_finder(finder_); |
| } |
| |
| CriticalLineInfoFinder* finder_; |
| RequestHeaders request_headers_; |
| }; |
| |
| TEST_F(CriticalLineInfoFinderTest, BasicTest) { |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| } |
| |
| TEST_F(CriticalLineInfoFinderTest, ConfigInHeader) { |
| request_headers_.Add(HttpAttributes::kXPsaSplitConfig, |
| "div[@id='b']:div[4]"); |
| rewrite_driver()->SetRequestHeaders(request_headers_); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(1, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[@id='b']", panel_start(0)); |
| EXPECT_EQ("div[4]", panel_end(0)); |
| } |
| |
| TEST_F(CriticalLineInfoFinderTest, ConfigInDomainOptions) { |
| options()->set_critical_line_config("div[@id='b']:div[4]"); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(1, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[@id='b']", panel_start(0)); |
| EXPECT_EQ("div[4]", panel_end(0)); |
| } |
| |
| TEST_F(CriticalLineInfoFinderTest, ConfigError) { |
| options()->set_critical_line_config("div[1]:div[2]:div[3]"); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| } |
| |
| TEST_F(CriticalLineInfoFinderTest, MultipleXpathPairs) { |
| options()->set_critical_line_config("div[1]:div[2],div[3]:div[4],div[5]"); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(3, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[1]", panel_start(0)); |
| EXPECT_EQ("div[2]", panel_end(0)); |
| EXPECT_EQ("div[3]", panel_start(1)); |
| EXPECT_EQ("div[4]", panel_end(1)); |
| EXPECT_EQ("div[5]", panel_start(2)); |
| EXPECT_EQ("", panel_end(2)); |
| } |
| |
| TEST_F(CriticalLineInfoFinderTest, ConfigInPcache) { |
| CriticalLineInfo config = CreateCriticalLineInfo( |
| "div[1]", "div[2]", "div[3]", ""); |
| WriteToPcache(config); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(2, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[1]", panel_start(0)); |
| EXPECT_EQ("div[2]", panel_end(0)); |
| EXPECT_EQ("div[3]", panel_start(1)); |
| EXPECT_EQ("", panel_end(1)); |
| } |
| |
| // Pcache config is preferred over domain config. |
| TEST_F(CriticalLineInfoFinderTest, ConfigInMultipleSources1) { |
| CriticalLineInfo config = CreateCriticalLineInfo( |
| "div[1]", "div[2]", "div[3]", ""); |
| WriteToPcache(config); |
| options()->set_critical_line_config("div[10]:div[11]"); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(2, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[1]", panel_start(0)); |
| EXPECT_EQ("div[2]", panel_end(0)); |
| EXPECT_EQ("div[3]", panel_start(1)); |
| EXPECT_EQ("", panel_end(1)); |
| } |
| |
| // HTTP header is preferred over domain config. |
| TEST_F(CriticalLineInfoFinderTest, ConfigInMultipleSources2) { |
| request_headers_.Add(HttpAttributes::kXPsaSplitConfig, |
| "div[1]:div[2]"); |
| rewrite_driver()->SetRequestHeaders(request_headers_); |
| options()->set_critical_line_config("div[10]:div[11]"); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(1, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[1]", panel_start(0)); |
| EXPECT_EQ("div[2]", panel_end(0)); |
| } |
| |
| // HTTP header is preferred over Pcache. |
| TEST_F(CriticalLineInfoFinderTest, ConfigInMultipleSources3) { |
| request_headers_.Add(HttpAttributes::kXPsaSplitConfig, |
| "div[1]:div[2]"); |
| rewrite_driver()->SetRequestHeaders(request_headers_); |
| CriticalLineInfo config = CreateCriticalLineInfo( |
| "div[10]", "div[11]", "div[12]", ""); |
| WriteToPcache(config); |
| |
| EXPECT_TRUE(rewrite_driver()->critical_line_info() == NULL); |
| server_context()->critical_line_info_finder() |
| ->GetCriticalLine(rewrite_driver()); |
| EXPECT_EQ(1, rewrite_driver()->critical_line_info()->panels_size()); |
| EXPECT_EQ("div[1]", panel_start(0)); |
| EXPECT_EQ("div[2]", panel_end(0)); |
| } |
| |
| } // namespace net_instaweb |