services/ui/view_manager/tests/view_associate_table_test.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 "mojo/services/ui/views/interfaces/view_manager.mojom.h"
#include "mojo/services/ui/views/interfaces/views.mojom.h"
#include "mojo/ui/associates/mock_view_inspector.h"
#include "services/ui/view_manager/tests/mock_view_associate.h"
#include "services/ui/view_manager/tests/view_manager_test_base.h"
#include "services/ui/view_manager/view_associate_table.h"

namespace view_manager {
namespace test {

class ViewAssociateTableTest : public ViewManagerTestBase {
 public:
  ViewAssociateTableTest() {}
  ~ViewAssociateTableTest() override {}

  void SetUp() override { ViewManagerTestBase::SetUp(); }

 private:
  DISALLOW_COPY_AND_ASSIGN(ViewAssociateTableTest);
};

TEST_F(ViewAssociateTableTest, RegisterViewAssociateThenCloseIt) {
  // Create a mock view registry
  mojo::ui::MockViewInspector mock_view_inspector;

  // Create a view associate table
  ViewAssociateTable view_associate_table;

  {
    // Create and bind a MockViewAssociate
    mojo::ui::ViewAssociatePtr associate;
    MockViewAssociate mock_view_associate;
    mojo::Binding<mojo::ui::ViewAssociate> view_associate_binding(
        &mock_view_associate, mojo::GetProxy(&associate));

    // call ViewAssociateTable::RegisterViewAssociate
    EXPECT_EQ((size_t)0, view_associate_table.associate_count());

    mojo::ui::ViewAssociateOwnerPtr view_associate_owner;
    view_associate_table.RegisterViewAssociate(
        &mock_view_inspector, associate.Pass(),
        mojo::GetProxy(&view_associate_owner), "test_view_associate");
    KICK_MESSAGE_LOOP_WHILE(view_associate_table.associate_count() != 1);
    EXPECT_EQ((size_t)1, view_associate_table.associate_count());
  }

  // ViewAssociate has been destroyed (since it's out of scope now)
  // Make sure it's been removed
  KICK_MESSAGE_LOOP_WHILE(view_associate_table.associate_count() != 0);
  EXPECT_EQ((size_t)0, view_associate_table.associate_count());
}

}  // namespace test
}  // namespace view_manager