EndSessionWatcherWindow implementation.

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