From fe0c056c835861fd4edae1248282fa3ab2bf6d2a Mon Sep 17 00:00:00 2001 From: Steve Hill Date: Wed, 19 Oct 2016 18:08:56 -0400 Subject: [PATCH] Additional DomainLawyer testcases --- net/instaweb/rewriter/domain_lawyer_test.cc | 29 +++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/net/instaweb/rewriter/domain_lawyer_test.cc b/net/instaweb/rewriter/domain_lawyer_test.cc index 5a556ad4f..6c37aee54 100644 --- a/net/instaweb/rewriter/domain_lawyer_test.cc +++ b/net/instaweb/rewriter/domain_lawyer_test.cc @@ -1746,12 +1746,41 @@ TEST_F(DomainLawyerTest, MapNewUrlDomain) { ASSERT_TRUE(domain_lawyer_.AddTwoProtocolOriginDomainMapping( origin_host, from_host, "", &message_handler_)); GoogleString origin_url; + + ASSERT_TRUE(MapOrigin("http://www.foo.com/123/www.xyz.com/", &origin_url)); + EXPECT_STREQ("http://www.xyz.com/", origin_url); + ASSERT_TRUE(MapOrigin("http://www.foo.com/123/www.xyz.com/a/b", &origin_url)); EXPECT_STREQ("http://www.xyz.com/a/b", origin_url); ASSERT_TRUE( MapOrigin("https://www.foo.com/123/www.xyz.com/a/b", &origin_url)); EXPECT_STREQ("https://www.xyz.com/a/b", origin_url); + + ASSERT_TRUE( + MapOrigin("http://www.foo.com/123/www.xyz.com/#fragment", &origin_url)); + EXPECT_STREQ("http://www.xyz.com/#fragment", origin_url); } +TEST_F(DomainLawyerTest, MapNewUrlDomainWithoutDomainSuffix) { + StringPiece from_host("www.foo.com/www.baz.com/"); + StringPiece origin_host("www.baz.com"); + ASSERT_TRUE(domain_lawyer_.AddTwoProtocolOriginDomainMapping( + origin_host, from_host, "", &message_handler_)); + GoogleString origin_url; + + ASSERT_TRUE(MapOrigin("http://www.foo.com/www.baz.com/bar", &origin_url)); + EXPECT_STREQ("http://www.baz.com/bar", origin_url); +} + +TEST_F(DomainLawyerTest, MapUrlDomainWithLeaf) { + StringPiece from_host("www.foo.com"); + StringPiece origin_host("www.baz.com"); + ASSERT_TRUE(domain_lawyer_.AddTwoProtocolOriginDomainMapping( + origin_host, from_host, "", &message_handler_)); + GoogleString origin_url; + + ASSERT_TRUE(MapOrigin("http://www.foo.com/bar", &origin_url)); + EXPECT_STREQ("http://www.baz.com/bar", origin_url); +} } // namespace net_instaweb