Restrict number of BlimpContents to [0, 1]

blimp/client/core/contents/blimp_contents_manager_unittest.cc

 // Copyright 2016 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 "blimp/client/core/contents/blimp_contents_manager.h"
 
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "blimp/client/core/contents/blimp_contents_impl.h"
 #include "blimp/client/core/contents/tab_control_feature.h"
@@ skipping 64 matching lines @@
       blimp_contents_manager.GetBlimpContents(id);
   EXPECT_EQ(blimp_contents.get(), existing_contents);
 
   EXPECT_CALL(tab_control_feature, CloseTab(id)).Times(1);
   blimp_contents.reset();
 
   loop.RunUntilIdle();
   EXPECT_EQ(nullptr, blimp_contents_manager.GetBlimpContents(id));
 }
 
+TEST(BlimpContentsManagerUnittest, CreateTwoBlimpContentsDestroyAndCreate) {

[David Trainor- moved to gerrit, 2016/08/30 23:31:12]

Add a TODO and a bug to remove this test and the related code once we support
more than one BlimpContents.

[Menglin, 2016/08/31 00:13:02]

Done.

+  base::MessageLoop loop;
+  MockTabControlFeature tab_control_feature;
+  BlimpContentsManager blimp_contents_manager(nullptr, nullptr,
+                                              &tab_control_feature);
+
+  EXPECT_CALL(tab_control_feature, CreateTab(_)).Times(2);
+  std::unique_ptr<BlimpContentsImpl> blimp_contents =
+      blimp_contents_manager.CreateBlimpContents();
+  DCHECK(blimp_contents);

[David Trainor- moved to gerrit, 2016/08/30 23:31:12]

EXPECT_NE()?  Same below.

[Menglin, 2016/08/31 00:13:02]

Done.

+
+  std::unique_ptr<BlimpContentsImpl> second_blimp_contents =
+      blimp_contents_manager.CreateBlimpContents();
+  EXPECT_EQ(second_blimp_contents, nullptr);
+
+  blimp_contents.reset();
+  std::unique_ptr<BlimpContentsImpl> third_blimp_contents =
+      blimp_contents_manager.CreateBlimpContents();
+  DCHECK(third_blimp_contents);
+}
+
 }  // namespace
 }  // namespace client
 }  // namespace blimp