Index: net/proxy/multi_threaded_proxy_resolver_unittest.cc |
diff --git a/net/proxy/multi_threaded_proxy_resolver_unittest.cc b/net/proxy/multi_threaded_proxy_resolver_unittest.cc |
index 83e11ae4be7f4aa19c8b9f43a5ad91cefc842faf..3231b88a92f9c2c91ab7372e157198451d57db76 100644 |
--- a/net/proxy/multi_threaded_proxy_resolver_unittest.cc |
+++ b/net/proxy/multi_threaded_proxy_resolver_unittest.cc |
@@ -175,9 +175,8 @@ class BlockableProxyResolverFactory : public LegacyProxyResolverFactory { |
TEST(MultiThreadedProxyResolverTest, SingleThread_Basic) { |
const size_t kNumThreads = 1u; |
- scoped_ptr<MockProxyResolver> mock(new MockProxyResolver); |
- MultiThreadedProxyResolver resolver( |
- new ForwardingProxyResolverFactory(mock.get()), kNumThreads); |
+ BlockableProxyResolverFactory* factory = new BlockableProxyResolverFactory; |
+ MultiThreadedProxyResolver resolver(factory, kNumThreads); |
int rv; |
@@ -191,8 +190,9 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_Basic) { |
set_script_callback.callback()); |
EXPECT_EQ(ERR_IO_PENDING, rv); |
EXPECT_EQ(OK, set_script_callback.WaitForResult()); |
+ ASSERT_EQ(1u, factory->resolvers().size()); |
EXPECT_EQ(ASCIIToUTF16("pac script bytes"), |
- mock->last_script_data()->utf16()); |
+ factory->resolvers()[0]->last_script_data()->utf16()); |
// Start request 0. |
TestCompletionCallback callback0; |
@@ -258,9 +258,8 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_Basic) { |
TEST(MultiThreadedProxyResolverTest, |
SingleThread_UpdatesNetLogWithThreadWait) { |
const size_t kNumThreads = 1u; |
- scoped_ptr<BlockableProxyResolver> mock(new BlockableProxyResolver); |
- MultiThreadedProxyResolver resolver( |
- new ForwardingProxyResolverFactory(mock.get()), kNumThreads); |
+ BlockableProxyResolverFactory* factory = new BlockableProxyResolverFactory; |
+ MultiThreadedProxyResolver resolver(factory, kNumThreads); |
int rv; |
@@ -270,6 +269,9 @@ TEST(MultiThreadedProxyResolverTest, |
init_callback.callback()); |
EXPECT_EQ(OK, init_callback.WaitForResult()); |
+ ASSERT_EQ(1u, factory->resolvers().size()); |
+ BlockableProxyResolver* mock = factory->resolvers()[0]; |
+ |
// Block the proxy resolver, so no request can complete. |
mock->Block(); |
@@ -351,10 +353,8 @@ TEST(MultiThreadedProxyResolverTest, |
// is pending. |
TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequest) { |
const size_t kNumThreads = 1u; |
- scoped_ptr<BlockableProxyResolver> mock(new BlockableProxyResolver); |
- MultiThreadedProxyResolver resolver( |
- new ForwardingProxyResolverFactory(mock.get()), |
- kNumThreads); |
+ BlockableProxyResolverFactory* factory = new BlockableProxyResolverFactory; |
+ MultiThreadedProxyResolver resolver(factory, kNumThreads); |
int rv; |
@@ -364,6 +364,9 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequest) { |
init_callback.callback()); |
EXPECT_EQ(OK, init_callback.WaitForResult()); |
+ ASSERT_EQ(1u, factory->resolvers().size()); |
+ BlockableProxyResolver* mock = factory->resolvers()[0]; |
+ |
// Block the proxy resolver, so no request can complete. |
mock->Block(); |
@@ -428,10 +431,9 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequest) { |
// outstanding cancels them (and doesn't leak anything). |
TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequestByDeleting) { |
const size_t kNumThreads = 1u; |
- scoped_ptr<BlockableProxyResolver> mock(new BlockableProxyResolver); |
+ BlockableProxyResolverFactory* factory = new BlockableProxyResolverFactory; |
scoped_ptr<MultiThreadedProxyResolver> resolver( |
- new MultiThreadedProxyResolver( |
- new ForwardingProxyResolverFactory(mock.get()), kNumThreads)); |
+ new MultiThreadedProxyResolver(factory, kNumThreads)); |
int rv; |
@@ -441,6 +443,9 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequestByDeleting) { |
init_callback.callback()); |
EXPECT_EQ(OK, init_callback.WaitForResult()); |
+ ASSERT_EQ(1u, factory->resolvers().size()); |
+ BlockableProxyResolver* mock = factory->resolvers()[0]; |
+ |
// Block the proxy resolver, so no request can complete. |
mock->Block(); |
@@ -491,9 +496,8 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelRequestByDeleting) { |
// Cancel an outstanding call to SetPacScriptByData(). |
TEST(MultiThreadedProxyResolverTest, SingleThread_CancelSetPacScript) { |
const size_t kNumThreads = 1u; |
- scoped_ptr<BlockableProxyResolver> mock(new BlockableProxyResolver); |
- MultiThreadedProxyResolver resolver( |
- new ForwardingProxyResolverFactory(mock.get()), kNumThreads); |
+ BlockableProxyResolverFactory* factory = new BlockableProxyResolverFactory; |
+ MultiThreadedProxyResolver resolver(factory, kNumThreads); |
int rv; |
@@ -502,6 +506,8 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelSetPacScript) { |
set_pac_script_callback.callback()); |
EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_EQ(1u, factory->resolvers().size()); |
+ |
// Cancel the SetPacScriptByData request. |
resolver.CancelSetPacScript(); |
@@ -515,7 +521,9 @@ TEST(MultiThreadedProxyResolverTest, SingleThread_CancelSetPacScript) { |
rv = set_pac_script_callback2.WaitForResult(); |
EXPECT_EQ(0, rv); |
- EXPECT_EQ(ASCIIToUTF16("data2"), mock->last_script_data()->utf16()); |
+ ASSERT_EQ(2u, factory->resolvers().size()); |
+ EXPECT_EQ(ASCIIToUTF16("data2"), |
+ factory->resolvers()[1]->last_script_data()->utf16()); |
// The first SetPacScript callback should never have been completed. |
EXPECT_FALSE(set_pac_script_callback.have_result()); |