chrome/browser/rlz/chrome_rlz_tracker_delegate_unittest.cc

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

#include "chrome/browser/rlz/chrome_rlz_tracker_delegate.h"

#include <memory>

#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

class ChromeRLZTrackerDelegateTest : public testing::Test {
 public:
  ChromeRLZTrackerDelegate* delegate() { return delegate_.get(); }

  MOCK_METHOD0(TestCallback, void(void));

 private:
  std::unique_ptr<ChromeRLZTrackerDelegate> delegate_{
      std::make_unique<ChromeRLZTrackerDelegate>()};
};

TEST_F(ChromeRLZTrackerDelegateTest, HomepageSearchCallback) {
  // The callback is not run if it is not set.
  EXPECT_CALL(*this, TestCallback()).Times(0);
  delegate()->RunHomepageSearchCallback();

  // Sets the callback and runs it.
  delegate()->SetHomepageSearchCallback(base::BindOnce(
      &ChromeRLZTrackerDelegateTest::TestCallback, base::Unretained(this)));
  EXPECT_CALL(*this, TestCallback());
  delegate()->RunHomepageSearchCallback();

  // The callback should be cleared and will not run again.
  EXPECT_CALL(*this, TestCallback()).Times(0);
  delegate()->RunHomepageSearchCallback();
}