Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "build/build_config.h" | 5 #include "build/build_config.h" |
| 6 | 6 |
| 7 #if defined(OS_WIN) | 7 #if defined(OS_WIN) |
| 8 #include <shlobj.h> | 8 #include <shlobj.h> |
| 9 #include <windows.h> | 9 #include <windows.h> |
| 10 #endif | 10 #endif |
| 11 | 11 |
| 12 #include <algorithm> | 12 #include <algorithm> |
| 13 #include <string> | 13 #include <string> |
| 14 | 14 |
| 15 #include "base/basictypes.h" | 15 #include "base/basictypes.h" |
| 16 #include "base/bind.h" | |
| 16 #include "base/compiler_specific.h" | 17 #include "base/compiler_specific.h" |
| 17 #include "base/file_util.h" | 18 #include "base/file_util.h" |
| 18 #include "base/format_macros.h" | 19 #include "base/format_macros.h" |
| 19 #include "base/message_loop.h" | 20 #include "base/message_loop.h" |
| 20 #include "base/path_service.h" | 21 #include "base/path_service.h" |
| 21 #include "base/process_util.h" | 22 #include "base/process_util.h" |
| 22 #include "base/string_number_conversions.h" | 23 #include "base/string_number_conversions.h" |
| 23 #include "base/string_piece.h" | 24 #include "base/string_piece.h" |
| 24 #include "base/string_util.h" | 25 #include "base/string_util.h" |
| 25 #include "base/stringprintf.h" | 26 #include "base/stringprintf.h" |
| (...skipping 1599 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 1625 MessageLoop::current()->Run(); | 1626 MessageLoop::current()->Run(); |
| 1626 | 1627 |
| 1627 EXPECT_TRUE(d.data_received().find("user2/secret") != std::string::npos); | 1628 EXPECT_TRUE(d.data_received().find("user2/secret") != std::string::npos); |
| 1628 | 1629 |
| 1629 // Make sure we sent the cookie in the restarted transaction. | 1630 // Make sure we sent the cookie in the restarted transaction. |
| 1630 EXPECT_TRUE(d.data_received().find("Cookie: got_challenged=true") | 1631 EXPECT_TRUE(d.data_received().find("Cookie: got_challenged=true") |
| 1631 != std::string::npos); | 1632 != std::string::npos); |
| 1632 } | 1633 } |
| 1633 } | 1634 } |
| 1634 | 1635 |
| 1635 TEST_F(URLRequestTest, DelayedCookieCallback) { | 1636 // TEST_F(URLRequestTest, DelayedCookieCallback) { |
| 1636 TestServer test_server(TestServer::TYPE_HTTP, FilePath()); | 1637 // TestServer test_server(TestServer::TYPE_HTTP, FilePath()); |
| 1637 ASSERT_TRUE(test_server.Start()); | 1638 // ASSERT_TRUE(test_server.Start()); |
| 1638 | 1639 |
| 1639 scoped_refptr<URLRequestContext> context(new TestURLRequestContext()); | 1640 // scoped_refptr<URLRequestContext> context(new TestURLRequestContext()); |
| 1640 scoped_refptr<DelayedCookieMonster> delayed_cm = | 1641 // scoped_refptr<DelayedCookieMonster> delayed_cm = |
| 1641 new DelayedCookieMonster(); | 1642 // new DelayedCookieMonster(); |
| 1642 scoped_refptr<CookieStore> cookie_store = delayed_cm; | 1643 // scoped_refptr<CookieStore> cookie_store = delayed_cm; |
| 1643 context->set_cookie_store(delayed_cm); | 1644 // context->set_cookie_store(delayed_cm); |
| 1644 | 1645 |
| 1645 // Set up a cookie. | 1646 // // Set up a cookie. |
| 1646 { | 1647 // { |
| 1647 TestDelegate d; | 1648 // TestDelegate d; |
| 1648 URLRequest req(test_server.GetURL("set-cookie?CookieToNotSend=1"), &d); | 1649 // URLRequest req(test_server.GetURL("set-cookie?CookieToNotSend=1"), &d); |
| 1649 req.set_context(context); | 1650 // req.set_context(context); |
| 1650 req.Start(); | 1651 // req.Start(); |
| 1651 MessageLoop::current()->Run(); | 1652 // MessageLoop::current()->Run(); |
| 1652 EXPECT_EQ(0, d.blocked_get_cookies_count()); | 1653 // EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| 1653 EXPECT_EQ(0, d.blocked_set_cookie_count()); | 1654 // EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| 1654 EXPECT_EQ(1, d.set_cookie_count()); | 1655 // EXPECT_EQ(1, d.set_cookie_count()); |
| 1655 } | 1656 // } |
| 1656 | 1657 |
| 1657 // Verify that the cookie is set. | 1658 // // Verify that the cookie is set. |
| 1658 { | 1659 // { |
| 1659 TestDelegate d; | 1660 // TestDelegate d; |
| 1660 TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); | 1661 // TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| 1661 req.set_context(context); | 1662 // req.set_context(context); |
| 1662 req.Start(); | 1663 // req.Start(); |
| 1663 MessageLoop::current()->Run(); | 1664 // MessageLoop::current()->Run(); |
| 1664 | 1665 |
| 1665 EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") | 1666 // EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") |
| 1666 != std::string::npos); | 1667 // != std::string::npos); |
| 1667 EXPECT_EQ(0, d.blocked_get_cookies_count()); | 1668 // EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| 1668 EXPECT_EQ(0, d.blocked_set_cookie_count()); | 1669 // EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| 1669 } | 1670 // } |
| 1670 } | 1671 // } |
| 1671 | 1672 |
| 1672 TEST_F(URLRequestTest, DoNotSendCookies) { | 1673 TEST_F(URLRequestTest, DoNotSendCookies) { |
| 1673 TestServer test_server(TestServer::TYPE_HTTP, FilePath()); | 1674 TestServer test_server(TestServer::TYPE_HTTP, FilePath()); |
| 1674 ASSERT_TRUE(test_server.Start()); | 1675 ASSERT_TRUE(test_server.Start()); |
| 1675 | 1676 |
| 1676 // Set up a cookie. | 1677 // Set up a cookie. |
| 1677 { | 1678 { |
| 1678 TestDelegate d; | 1679 TestDelegate d; |
| 1679 URLRequest req(test_server.GetURL("set-cookie?CookieToNotSend=1"), &d); | 1680 URLRequest req(test_server.GetURL("set-cookie?CookieToNotSend=1"), &d); |
| 1680 req.set_context(default_context_); | 1681 req.set_context(default_context_); |
| (...skipping 294 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 1975 EXPECT_TRUE(d.data_received().find("CookieToNotSave=1") | 1976 EXPECT_TRUE(d.data_received().find("CookieToNotSave=1") |
| 1976 == std::string::npos); | 1977 == std::string::npos); |
| 1977 EXPECT_TRUE(d.data_received().find("CookieToNotUpdate=2") | 1978 EXPECT_TRUE(d.data_received().find("CookieToNotUpdate=2") |
| 1978 != std::string::npos); | 1979 != std::string::npos); |
| 1979 | 1980 |
| 1980 EXPECT_EQ(0, d.blocked_get_cookies_count()); | 1981 EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| 1981 EXPECT_EQ(0, d.blocked_set_cookie_count()); | 1982 EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| 1982 } | 1983 } |
| 1983 } | 1984 } |
| 1984 | 1985 |
| 1986 void CheckCookiePolicyCallback(const CookieList& cookies) { | |
| 1987 EXPECT_EQ(1U, cookies.size()); | |
| 1988 EXPECT_FALSE(cookies[0].IsPersistent()); | |
| 1989 MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask()); | |
| 1990 } | |
| 1991 | |
| 1985 TEST_F(URLRequestTest, CookiePolicy_ForceSession) { | 1992 TEST_F(URLRequestTest, CookiePolicy_ForceSession) { |
| 1986 TestServer test_server(TestServer::TYPE_HTTP, FilePath()); | 1993 TestServer test_server(TestServer::TYPE_HTTP, FilePath()); |
| 1987 ASSERT_TRUE(test_server.Start()); | 1994 ASSERT_TRUE(test_server.Start()); |
| 1988 | 1995 |
| 1989 // Set up a cookie. | 1996 // Set up a cookie. |
| 1990 { | 1997 { |
| 1991 TestDelegate d; | 1998 TestDelegate d; |
| 1992 d.set_cookie_options(TestDelegate::FORCE_SESSION); | 1999 d.set_cookie_options(TestDelegate::FORCE_SESSION); |
| 1993 URLRequest req(test_server.GetURL( | 2000 URLRequest req(test_server.GetURL( |
| 1994 "set-cookie?A=1;expires=\"Fri, 05 Feb 2010 23:42:01 GMT\""), &d); | 2001 "set-cookie?A=1;expires=\"Fri, 05 Feb 2010 23:42:01 GMT\""), &d); |
| 1995 req.set_context(default_context_); | 2002 req.set_context(default_context_); |
| 1996 req.Start(); // Triggers an asynchronous cookie policy check. | 2003 req.Start(); // Triggers an asynchronous cookie policy check. |
| 1997 | 2004 |
| 1998 MessageLoop::current()->Run(); | 2005 MessageLoop::current()->Run(); |
| 1999 | 2006 |
| 2000 EXPECT_EQ(0, d.blocked_get_cookies_count()); | 2007 EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| 2001 EXPECT_EQ(0, d.blocked_set_cookie_count()); | 2008 EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| 2002 } | 2009 } |
| 2003 | 2010 |
| 2004 // Now, check the cookie store. | 2011 // Now, check the cookie store. |
| 2005 CookieList cookies = | 2012 default_context_->cookie_store()->GetCookieMonster()->GetAllCookiesAsync( |
| 2006 default_context_->cookie_store()->GetCookieMonster()->GetAllCookies(); | 2013 base::Bind(&CheckCookiePolicyCallback)); |
| 2007 EXPECT_EQ(1U, cookies.size()); | 2014 MessageLoop::current()->RunAllPending(); |
|
erikwright (departed)
2011/08/11 15:13:52
Use a 'was_run' boolean to verify that the callbac
ycxiao
2011/08/12 02:35:24
Done.
| |
| 2008 EXPECT_FALSE(cookies[0].IsPersistent()); | |
| 2009 } | 2015 } |
| 2010 | 2016 |
| 2011 // In this test, we do a POST which the server will 302 redirect. | 2017 // In this test, we do a POST which the server will 302 redirect. |
| 2012 // The subsequent transaction should use GET, and should not send the | 2018 // The subsequent transaction should use GET, and should not send the |
| 2013 // Content-Type header. | 2019 // Content-Type header. |
| 2014 // http://code.google.com/p/chromium/issues/detail?id=843 | 2020 // http://code.google.com/p/chromium/issues/detail?id=843 |
| 2015 TEST_F(URLRequestTestHTTP, Post302RedirectGet) { | 2021 TEST_F(URLRequestTestHTTP, Post302RedirectGet) { |
| 2016 ASSERT_TRUE(test_server_.Start()); | 2022 ASSERT_TRUE(test_server_.Start()); |
| 2017 | 2023 |
| 2018 const char kData[] = "hello world"; | 2024 const char kData[] = "hello world"; |
| (...skipping 1114 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 3133 req.SetExtraRequestHeaders(headers); | 3139 req.SetExtraRequestHeaders(headers); |
| 3134 req.Start(); | 3140 req.Start(); |
| 3135 MessageLoop::current()->Run(); | 3141 MessageLoop::current()->Run(); |
| 3136 // If the net tests are being run with ChromeFrame then we need to allow for | 3142 // If the net tests are being run with ChromeFrame then we need to allow for |
| 3137 // the 'chromeframe' suffix which is added to the user agent before the | 3143 // the 'chromeframe' suffix which is added to the user agent before the |
| 3138 // closing parentheses. | 3144 // closing parentheses. |
| 3139 EXPECT_TRUE(StartsWithASCII(d.data_received(), "Lynx (textmode", true)); | 3145 EXPECT_TRUE(StartsWithASCII(d.data_received(), "Lynx (textmode", true)); |
| 3140 } | 3146 } |
| 3141 | 3147 |
| 3142 } // namespace net | 3148 } // namespace net |
| OLD | NEW |