Add missing test to WeakPtr

base/memory/weak_ptr_unittest.cc

 // Copyright (c) 2011 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 "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "base/message_loop.h"
 #include "base/threading/thread.h"
 
@@ skipping 62 matching lines @@
     WeakPtrFactory<int> factory(&data);
     a = factory.GetWeakPtr();
     b = factory.GetWeakPtr();
     EXPECT_EQ(&data, a.get());
     EXPECT_EQ(&data, b.get());
   }
   EXPECT_TRUE(a.get() == NULL);
   EXPECT_TRUE(b.get() == NULL);
 }
 
+TEST(WeakPtrTest, MultipleStaged) {
+  WeakPtr<int> a;
+  {
+    int data;
+    WeakPtrFactory<int> factory(&data);
+    a = factory.GetWeakPtr();
+    {
+      WeakPtr<int> b = factory.GetWeakPtr();

[willchan no longer on Chromium, 2011/05/24 22:29:24]

Maybe you should also add EXPECTs for b? Up to you.

+    }
+    EXPECT_TRUE(a.get() != NULL);
+  }
+  EXPECT_TRUE(a.get() == NULL);
+}
+
 TEST(WeakPtrTest, UpCast) {
   Derived data;
   WeakPtrFactory<Derived> factory(&data);
   WeakPtr<Base> ptr = factory.GetWeakPtr();
   ptr = factory.GetWeakPtr();
   EXPECT_EQ(ptr.get(), &data);
 }
 
 TEST(WeakPtrTest, SupportsWeakPtr) {
   Producer f;
@@ skipping 35 matching lines @@
   // Test that it is OK to create a class that has a WeakPtr member on one
   // thread, but use it on another.  This tests that we do not trip runtime
   // checks that ensure that a weak reference is not used by multiple threads.
   scoped_ptr<Consumer> consumer(OffThreadObjectCreator<Consumer>::NewObject());
   Producer producer;
   consumer->producer = producer.AsWeakPtr();
   EXPECT_EQ(&producer, consumer->producer.get());
 }
 
 }  // namespace base