Make ToolbarActionsModel own ComponentToolbarActionsFactory

chrome/browser/ui/toolbar/component_toolbar_actions_factory_unittest.cc

+// Copyright 2017 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 "chrome/browser/extensions/extension_action_test_util.h"
+#include "chrome/browser/ui/toolbar/component_toolbar_actions_factory.h"
+#include "chrome/browser/ui/toolbar/toolbar_actions_model.h"
+#include "chrome/test/base/browser_with_test_window_test.h"
+
+class ComponentToolbarActionsFactoryUnitTest
+    : public BrowserWithTestWindowTest {
+ public:
+  ComponentToolbarActionsFactoryUnitTest() {}
+  ~ComponentToolbarActionsFactoryUnitTest() override {}
+
+  void SetUp() override {
+    BrowserWithTestWindowTest::SetUp();
+    actions_factory_ =
+        extensions::extension_action_test_util::
+            CreateToolbarModelForProfileWithoutWaitingForReady(profile())
+                ->component_actions_factory();
+  }
+
+ protected:
+  ComponentToolbarActionsFactory* actions_factory_;

[msw, 2017/01/10 23:23:59]

nit: = nullptr

[takumif, 2017/02/17 03:22:51]

Done.

+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(ComponentToolbarActionsFactoryUnitTest);
+};
+
+TEST_F(ComponentToolbarActionsFactoryUnitTest, GetInitialIds) {
+  std::string id1("id1");
+  std::string id2("id2");
+  std::string id3("id3");
+
+  actions_factory_->OnAddComponentActionBeforeInit(id1);
+  actions_factory_->OnAddComponentActionBeforeInit(id2);
+  actions_factory_->OnAddComponentActionBeforeInit(id3);
+  actions_factory_->OnRemoveComponentActionBeforeInit(id2);
+
+  std::set<std::string> initial_ids =
+      actions_factory_->GetInitialComponentIds();
+  EXPECT_TRUE(base::ContainsKey(initial_ids, id1));
+  EXPECT_FALSE(base::ContainsKey(initial_ids, id2));
+  EXPECT_TRUE(base::ContainsKey(initial_ids, id3));
+}