Merge r205158 (SocketStream uses a weak ptr to URLRequestContext)

net/socket_stream/socket_stream_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/socket_stream/socket_stream.h"
 
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 906 matching lines @@
   delegate->set_socket_stream(socket_stream);
   // The delegate pointer will become invalid during the test. Set it to NULL to
   // avoid holding a dangling pointer.
   delegate = NULL;
 
   socket_stream->Connect();
 
   EXPECT_EQ(OK, test_callback.WaitForResult());
 }
 
+// A test for the WeakPtr workaround patch which should be deleted
+// in the future.
+TEST_F(SocketStreamTest, ContextDestroyedBeforeDoBeforeConnect) {
+  TestCompletionCallback test_callback;
+
+  scoped_ptr<SocketStreamEventRecorder> delegate(
+      new SocketStreamEventRecorder(test_callback.callback()));
+
+  scoped_ptr<TestURLRequestContext> context(new TestURLRequestContext);
+  TestSocketStreamNetworkDelegate network_delegate;
+  network_delegate.SetBeforeConnectResult(OK);
+  context->set_network_delegate(&network_delegate);
+
+  scoped_refptr<SocketStream> socket_stream(
+      new SocketStream(GURL("ws://example.com/demo"), delegate.get()));
+
+  socket_stream->set_context(context.get());
+
+  socket_stream->Connect();
+  context.reset();
+
+  ASSERT_EQ(0U, delegate->GetSeenEvents().size());
+
+  test_callback.WaitForResult();
+
+  const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents();
+  ASSERT_EQ(1U, events.size());
+
+  EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[0].event_type);
+}
+
+// A test for the WeakPtr workaround patch which should be deleted
+// in the future.
+TEST_F(SocketStreamTest, SocketStreamResetBeforeDoBeforeConnect) {
+  TestCompletionCallback test_callback;
+
+  scoped_ptr<SocketStreamEventRecorder> delegate(
+      new SocketStreamEventRecorder(test_callback.callback()));
+
+  scoped_ptr<TestURLRequestContext> context(new TestURLRequestContext);
+  TestSocketStreamNetworkDelegate network_delegate;
+  network_delegate.SetBeforeConnectResult(OK);
+  context->set_network_delegate(&network_delegate);
+
+  scoped_refptr<SocketStream> socket_stream(
+      new SocketStream(GURL("ws://example.com/demo"), delegate.get()));
+
+  socket_stream->set_context(context.get());
+
+  socket_stream->Connect();
+  socket_stream = NULL;
+  context.reset();
+
+  ASSERT_EQ(0U, delegate->GetSeenEvents().size());
+  test_callback.WaitForResult();
+
+  const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents();
+  ASSERT_EQ(1U, events.size());
+
+  EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[0].event_type);
+}
+
 }  // namespace net