Screensaver at login screen.

chrome/browser/chromeos/kiosk_mode/kiosk_mode_screensaver_unittest.cc

+// Copyright (c) 2012 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 "chrome/browser/chromeos/kiosk_mode/kiosk_mode_screensaver.h"
+
+#include "base/bind.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop.h"
+#include "base/synchronization/waitable_event.h"
+#include "chrome/browser/chromeos/dbus/dbus_thread_manager.h"
+#include "chrome/browser/chromeos/login/user_manager.h"
+#include "chrome/common/chrome_notification_types.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/notification_registrar.h"
+#include "content/public/browser/notification_service.h"
+#include "content/test/test_browser_thread.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using content::BrowserThread;
+
+namespace chromeos {
+
+class KioskModeScreensaverTest : public testing::Test {
+ public:
+  KioskModeScreensaverTest() : ui_thread_(BrowserThread::UI, &message_loop_),

[sky, 2012/02/24 05:03:53]

: on next line

[rkc, 2012/02/25 02:14:02]

Done.

+                               setup_done_(
+                                   new base::WaitableEvent(true, false)) {
+  }
+
+  virtual void SetUp() OVERRIDE {
+    DBusThreadManager::Initialize();
+
+    BrowserThread::PostTask(BrowserThread::UI,

[sky, 2012/02/24 05:03:53]

Document why this needs to be delayed.

[rkc, 2012/02/25 02:14:02]

It doesn't. I'm new to using test threads :)
Fixed.
Done.

+        FROM_HERE,
+        base::Bind(
+            &KioskModeScreensaverTest::CreateScreensaverObserver,
+            base::Unretained(this)));
+
+    while (!setup_done_->IsSignaled())
+      message_loop_.RunAllPending();
+  }
+
+  virtual void TearDown() OVERRIDE {
+    DBusThreadManager::Shutdown();
+  }
+
+  void CreateScreensaverObserver() {
+    screensaver_observer_.reset(new KioskModeScreensaverObserver());
+    setup_done_->Signal();
+  }
+
+  bool LoginUserObserverRegistered() {
+    return screensaver_observer_.get()->
+        registrar_.IsRegistered(screensaver_observer_.get(),
+            chrome::NOTIFICATION_LOGIN_USER_CHANGED,
+            content::NotificationService::AllSources());
+  }
+
+  bool PowerManagerObserverRegistered() {
+    chromeos::PowerManagerClient* power_manager =
+                chromeos::DBusThreadManager::Get()->GetPowerManagerClient();
+    return power_manager->HasObserver(screensaver_observer_.get());
+  }
+
+  MessageLoopForUI message_loop_;
+  content::TestBrowserThread ui_thread_;
+
+  base::WaitableEvent* setup_done_;
+
+  scoped_ptr<KioskModeScreensaverObserver> screensaver_observer_;
+  content::NotificationRegistrar registrar_;
+};
+
+TEST_F(KioskModeScreensaverTest, CheckObservers) {
+  EXPECT_TRUE(LoginUserObserverRegistered());
+  EXPECT_TRUE(PowerManagerObserverRegistered());
+}
+
+TEST_F(KioskModeScreensaverTest, CheckObserversAfterUserLogin) {
+  content::NotificationService::current()->Notify(
+      chrome::NOTIFICATION_LOGIN_USER_CHANGED,
+      content::Source<UserManager>(UserManager::Get()),
+      // Ideally this should be the user logged in, but since we won't really be
+      // checking for the current logged in user in our observer anyway, giving
+      // NoDetails here is fine.
+      content::NotificationService::NoDetails());
+
+  EXPECT_FALSE(PowerManagerObserverRegistered());
+}
+
+}  // namespace chromeos