// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ui/views/view_tracker.h"

#include <memory>

#include "base/functional/bind.h"
#include "base/test/mock_callback.h"
#include "ui/views/test/views_test_base.h"
#include "ui/views/view.h"

namespace views {

using ViewTrackerTest = ViewsTestBase;

TEST_F(ViewTrackerTest, RemovedOnDelete) {
  ViewTracker tracker;
  {
    View view;
    tracker.SetView(&view);
    EXPECT_EQ(&view, tracker.view());
  }
  EXPECT_EQ(nullptr, tracker.view());
}

TEST_F(ViewTrackerTest, ObservedAtConstruction) {
  std::unique_ptr<ViewTracker> tracker;
  {
    View view;
    tracker = std::make_unique<ViewTracker>(&view);
    EXPECT_EQ(&view, tracker->view());
  }
  EXPECT_EQ(nullptr, tracker->view());
}

TEST_F(ViewTrackerTest, RunCallbackOnViewDeletion) {
  base::MockCallback<base::OnceClosure> is_deleting_callback;
  EXPECT_CALL(is_deleting_callback, Run);
  ViewTracker tracker;
  {
    std::unique_ptr<View> tracked_view = std::make_unique<View>();
    tracker.SetView(tracked_view.get());
    EXPECT_EQ(tracked_view.get(), tracker.view());
    tracker.SetIsDeletingCallback(is_deleting_callback.Get());
  }
}

TEST_F(ViewTrackerTest, TestTrackEntireViewHierarchy) {
  auto hierarchy_deleting_callback = [](View* view) {
    // Expected to be called before children are destroyed.
    EXPECT_EQ(view->children().size(), 1u);
  };
  ViewTracker tracker;
  {
    tracker.SetTrackEntireViewHierarchy(true);
    std::unique_ptr<View> tracked_view = std::make_unique<View>();
    tracked_view->AddChildView(std::make_unique<View>());
    tracker.SetView(tracked_view.get());
    tracker.SetIsDeletingCallback(base::BindOnce(
        hierarchy_deleting_callback, base::Unretained(tracked_view.get())));
  }
  auto view_deleting_callback = [](View* view) {
    // Expected to be called after children are destroyed.
    EXPECT_EQ(view->children().size(), 0u);
  };
  {
    tracker.SetTrackEntireViewHierarchy(false);
    std::unique_ptr<View> tracked_view = std::make_unique<View>();
    tracked_view->AddChildView(std::make_unique<View>());
    tracker.SetView(tracked_view.get());
    tracker.SetIsDeletingCallback(base::BindOnce(
        view_deleting_callback, base::Unretained(tracked_view.get())));
  }
}

}  // namespace views