Chromium Code Reviews| Index: net/url_request/url_request_unittest.cc |
| diff --git a/net/url_request/url_request_unittest.cc b/net/url_request/url_request_unittest.cc |
| index caeaf71cbc6df57b4297df21e05f15b971668819..29f2d257aaef30723177bd8604dd9fc1f32ad7e6 100644 |
| --- a/net/url_request/url_request_unittest.cc |
| +++ b/net/url_request/url_request_unittest.cc |
| @@ -2561,7 +2561,9 @@ TEST_F(URLRequestTest, DelayedCookieCallback) { |
| LocalHttpTestServer test_server; |
| ASSERT_TRUE(test_server.Start()); |
| + TestNetworkDelegate network_delegate; // must outlive URLRequests |
| scoped_refptr<URLRequestContext> context(new TestURLRequestContext()); |
| + context->set_network_delegate(&network_delegate); |
| scoped_refptr<DelayedCookieMonster> delayed_cm = |
| new DelayedCookieMonster(); |
| scoped_refptr<CookieStore> cookie_store = delayed_cm; |
| @@ -2574,13 +2576,14 @@ TEST_F(URLRequestTest, DelayedCookieCallback) { |
| req.set_context(context); |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| - EXPECT_EQ(1, d.set_cookie_count()); |
| + EXPECT_EQ(0, network_delegate.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, network_delegate.blocked_set_cookie_count()); |
| + EXPECT_EQ(1, network_delegate.set_cookie_count()); |
| } |
| // Verify that the cookie is set. |
| { |
| + network_delegate.ResetCookieCounters(); |
|
willchan no longer on Chromium
2012/03/01 20:30:10
I don't know that I like this as much. Can we just
|
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(context); |
| @@ -2589,8 +2592,8 @@ TEST_F(URLRequestTest, DelayedCookieCallback) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, network_delegate.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, network_delegate.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2605,12 +2608,13 @@ TEST_F(URLRequestTest, DoNotSendCookies) { |
| req.set_context(default_context_); |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie is set. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2619,12 +2623,13 @@ TEST_F(URLRequestTest, DoNotSendCookies) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie isn't sent when LOAD_DO_NOT_SEND_COOKIES is set. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_load_flags(LOAD_DO_NOT_SEND_COOKIES); |
| @@ -2636,8 +2641,8 @@ TEST_F(URLRequestTest, DoNotSendCookies) { |
| == std::string::npos); |
| // LOAD_DO_NOT_SEND_COOKIES does not trigger OnGetCookies. |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2653,13 +2658,14 @@ TEST_F(URLRequestTest, DoNotSaveCookies) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| - EXPECT_EQ(1, d.set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| + EXPECT_EQ(1, default_network_delegate_.set_cookie_count()); |
| } |
| // Try to set-up another cookie and update the previous cookie. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| URLRequest req(test_server.GetURL( |
| "set-cookie?CookieToNotSave=1&CookieToNotUpdate=1"), &d); |
| @@ -2670,13 +2676,14 @@ TEST_F(URLRequestTest, DoNotSaveCookies) { |
| MessageLoop::current()->Run(); |
| // LOAD_DO_NOT_SAVE_COOKIES does not trigger OnSetCookie. |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| - EXPECT_EQ(0, d.set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.set_cookie_count()); |
| } |
| // Verify the cookies weren't saved or updated. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2688,9 +2695,9 @@ TEST_F(URLRequestTest, DoNotSaveCookies) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotUpdate=2") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| - EXPECT_EQ(0, d.set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.set_cookie_count()); |
| } |
| } |
| @@ -2706,12 +2713,13 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie is set. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2721,14 +2729,16 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie isn't sent. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| - d.set_cookie_options(TestDelegate::NO_GET_COOKIES); |
| + default_network_delegate_.set_cookie_options( |
| + TestNetworkDelegate::NO_GET_COOKIES); |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| req.Start(); |
| @@ -2737,8 +2747,8 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy) { |
| EXPECT_TRUE(d.data_received().find("Cookie: CookieToNotSend=1") |
| == std::string::npos); |
| - EXPECT_EQ(1, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(1, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2754,14 +2764,16 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Try to set-up another cookie and update the previous cookie. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| - d.set_cookie_options(TestDelegate::NO_SET_COOKIE); |
| + default_network_delegate_.set_cookie_options( |
| + TestNetworkDelegate::NO_SET_COOKIE); |
| URLRequest req(test_server.GetURL( |
| "set-cookie?CookieToNotSave=1&CookieToNotUpdate=1"), &d); |
| req.set_context(default_context_); |
| @@ -2769,13 +2781,14 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy) { |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(2, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(2, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify the cookies weren't saved or updated. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2787,8 +2800,8 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotUpdate=2") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2804,9 +2817,9 @@ TEST_F(URLRequestTest, DoNotSaveEmptyCookies) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| - EXPECT_EQ(0, d.set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.set_cookie_count()); |
| } |
| } |
| @@ -2822,12 +2835,13 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy_Async) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie is set. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2837,14 +2851,16 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy_Async) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotSend=1") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify that the cookie isn't sent. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| - d.set_cookie_options(TestDelegate::NO_GET_COOKIES); |
| + default_network_delegate_.set_cookie_options( |
| + TestNetworkDelegate::NO_GET_COOKIES); |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| req.Start(); |
| @@ -2853,8 +2869,8 @@ TEST_F(URLRequestTest, DoNotSendCookies_ViaPolicy_Async) { |
| EXPECT_TRUE(d.data_received().find("Cookie: CookieToNotSend=1") |
| == std::string::npos); |
| - EXPECT_EQ(1, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(1, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2870,14 +2886,16 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy_Async) { |
| req.Start(); |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Try to set-up another cookie and update the previous cookie. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| - d.set_cookie_options(TestDelegate::NO_SET_COOKIE); |
| + default_network_delegate_.set_cookie_options( |
| + TestNetworkDelegate::NO_SET_COOKIE); |
| URLRequest req(test_server.GetURL( |
| "set-cookie?CookieToNotSave=1&CookieToNotUpdate=1"), &d); |
| req.set_context(default_context_); |
| @@ -2885,12 +2903,13 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy_Async) { |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(2, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(2, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Verify the cookies weren't saved or updated. |
| { |
| + default_network_delegate_.ResetCookieCounters(); |
| TestDelegate d; |
| TestURLRequest req(test_server.GetURL("echoheader?Cookie"), &d); |
| req.set_context(default_context_); |
| @@ -2902,8 +2921,8 @@ TEST_F(URLRequestTest, DoNotSaveCookies_ViaPolicy_Async) { |
| EXPECT_TRUE(d.data_received().find("CookieToNotUpdate=2") |
| != std::string::npos); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| } |
| @@ -2921,7 +2940,8 @@ TEST_F(URLRequestTest, CookiePolicy_ForceSession) { |
| // Set up a cookie. |
| { |
| TestDelegate d; |
| - d.set_cookie_options(TestDelegate::FORCE_SESSION); |
| + default_network_delegate_.set_cookie_options( |
| + TestNetworkDelegate::FORCE_SESSION); |
| URLRequest req(test_server.GetURL( |
| "set-cookie?A=1;expires=\"Fri, 05 Feb 2010 23:42:01 GMT\""), &d); |
| req.set_context(default_context_); |
| @@ -2929,8 +2949,8 @@ TEST_F(URLRequestTest, CookiePolicy_ForceSession) { |
| MessageLoop::current()->Run(); |
| - EXPECT_EQ(0, d.blocked_get_cookies_count()); |
| - EXPECT_EQ(0, d.blocked_set_cookie_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_get_cookies_count()); |
| + EXPECT_EQ(0, default_network_delegate_.blocked_set_cookie_count()); |
| } |
| // Now, check the cookie store. |