components/browser_watcher/endsession_watcher_window_win_unittest.cc

// Copyright (c) 2015 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 "components/browser_watcher/endsession_watcher_window_win.h"

#include <windows.h>

#include "base/bind.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace browser_watcher {

namespace {

void OnEndSession(bool* called, LPARAM* out_lparam, LPARAM lparam) {
  *called = true;
  *out_lparam = lparam;
}

}  // namespace browser_watcher

TEST(EndSessionWatcherWindowTest, NoCallbackOnDestruction) {
  LPARAM lparam = 0;
  bool was_called = false;

  {
    EndSessionWatcherWindow watcher_window(
        base::Bind(&OnEndSession, &was_called, &lparam));
  }

  EXPECT_FALSE(was_called);
  EXPECT_EQ(lparam, 0);
}

TEST(EndSessionWatcherWindowTest, IssuesCallbackOnMessage) {
  LPARAM lparam = 0;
  bool was_called = false;

  EndSessionWatcherWindow watcher_window(
      base::Bind(&OnEndSession, &was_called, &lparam));

  ::SendMessage(watcher_window.window(), WM_ENDSESSION, TRUE, 0xCAFEBABE);

  EXPECT_TRUE(was_called);
  EXPECT_EQ(lparam, 0xCAFEBABE);
}

}  // namespace browser_watcher