chrome/browser/ui/ash/screenshot_taker_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/ash/screenshot_taker.h"
   6
   7 #include "ash/shell.h"
   8 #include "ash/test/ash_test_base.h"
   9 #include "base/bind.h"
  10 #include "base/command_line.h"
  11 #include "base/file_util.h"
  12 #include "base/files/scoped_temp_dir.h"
  13 #include "base/message_loop/message_loop.h"
  14 #include "chrome/browser/browser_process.h"
  15 #include "chrome/browser/notifications/notification_ui_manager.h"
  16 #include "chrome/test/base/in_process_browser_test.h"
  17 #include "chrome/test/base/testing_browser_process.h"
  18 #include "chrome/test/base/testing_profile.h"
  19 #include "chrome/test/base/testing_profile_manager.h"
  20 #include "content/public/browser/browser_thread.h"
  21 #include "content/public/test/test_utils.h"
  22 #include "ui/aura/root_window.h"
  23 #include "ui/message_center/message_center_switches.h"
  24
  25 #if defined(OS_CHROMEOS)
  26 #include "chromeos/login/login_state.h"
  27 #endif
  28
  29 namespace ash {
  30 namespace test {
  31
  32 class ScreenshotTakerTest : public AshTestBase,
  33                             public ScreenshotTakerObserver {
  34  public:
  35   ScreenshotTakerTest()
  36       : running_(false),
  37         screenshot_complete_(false),
  38         screenshot_result_(ScreenshotTakerObserver::SCREENSHOT_SUCCESS) {
  39   }
  40
  41   virtual void SetUp() {
  42     AshTestBase::SetUp();
  43   }
  44
  45   virtual void TearDown() {
  46     RunAllPendingInMessageLoop();
  47     AshTestBase::TearDown();
  48   }
  49
  50   // Overridden from ScreenshotTakerObserver
  51   virtual void OnScreenshotCompleted(
  52       ScreenshotTakerObserver::Result screenshot_result,
  53       const base::FilePath& screenshot_path) OVERRIDE {
  54     screenshot_complete_ = true;
  55     screenshot_result_ = screenshot_result;
  56     screenshot_path_ = screenshot_path;
  57     if (!running_)
  58       return;
  59     message_loop_runner_->Quit();
  60     running_ = false;
  61   }
  62
  63  protected:
  64   // ScreenshotTakerTest is a friend of ScreenshotTaker and therefore
  65   // allowed to set the directory, basename and profile.
  66   void SetScreenshotDirectoryForTest(
  67       ScreenshotTaker* screenshot_taker,
  68       const base::FilePath& screenshot_directory) {
  69     screenshot_taker->SetScreenshotDirectoryForTest(screenshot_directory);
  70   }
  71   void SetScreenshotBasenameForTest(
  72       ScreenshotTaker* screenshot_taker,
  73       const std::string& screenshot_basename) {
  74     screenshot_taker->SetScreenshotBasenameForTest(screenshot_basename);
  75   }
  76   void SetScreenshotProfileForTest(
  77       ScreenshotTaker* screenshot_taker,
  78       Profile* profile) {
  79     screenshot_taker->SetScreenshotProfileForTest(profile);
  80   }
  81
  82   void Wait() {
  83     if (screenshot_complete_)
  84       return;
  85     running_ = true;
  86     message_loop_runner_ = new content::MessageLoopRunner;
  87     message_loop_runner_->Run();
  88     EXPECT_TRUE(screenshot_complete_);
  89   }
  90
  91   bool running_;
  92   bool screenshot_complete_;
  93   ScreenshotTakerObserver::Result screenshot_result_;
  94   base::FilePath screenshot_path_;
  95   scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
  96
  97   DISALLOW_COPY_AND_ASSIGN(ScreenshotTakerTest);
  98 };
  99
 100 TEST_F(ScreenshotTakerTest, TakeScreenshot) {
 101 #if defined(OS_CHROMEOS)
 102   // Note that within the test framework the LoginState object will always
 103   // claim that the user did log in.
 104   ASSERT_FALSE(chromeos::LoginState::IsInitialized());
 105   chromeos::LoginState::Initialize();
 106 #endif
 107   scoped_ptr<TestingProfileManager> profile_manager(
 108       new TestingProfileManager(TestingBrowserProcess::GetGlobal()));
 109   ASSERT_TRUE(profile_manager->SetUp());
 110   TestingProfile* profile =
 111       profile_manager->CreateTestingProfile("test_profile");
 112   ScreenshotTaker screenshot_taker;
 113   screenshot_taker.AddObserver(this);
 114   base::ScopedTempDir directory;
 115   ASSERT_TRUE(directory.CreateUniqueTempDir());
 116   SetScreenshotDirectoryForTest(&screenshot_taker, directory.path());
 117   SetScreenshotBasenameForTest(&screenshot_taker, "Screenshot");
 118   SetScreenshotProfileForTest(&screenshot_taker, profile);
 119
 120   EXPECT_TRUE(screenshot_taker.CanTakeScreenshot());
 121
 122   screenshot_taker.HandleTakePartialScreenshot(
 123       Shell::GetPrimaryRootWindow(), gfx::Rect(0, 0, 100, 100));
 124
 125   EXPECT_FALSE(screenshot_taker.CanTakeScreenshot());
 126
 127   Wait();
 128
 129 #if defined(OS_CHROMEOS)
 130   // Screenshot notifications on Windows not yet turned on.
 131   EXPECT_TRUE(g_browser_process->notification_ui_manager()->FindById(
 132       std::string("screenshot")) != NULL);
 133   g_browser_process->notification_ui_manager()->CancelAll();
 134 #endif
 135
 136   EXPECT_EQ(ScreenshotTakerObserver::SCREENSHOT_SUCCESS, screenshot_result_);
 137
 138   if (ScreenshotTakerObserver::SCREENSHOT_SUCCESS == screenshot_result_)
 139     EXPECT_TRUE(base::PathExists(screenshot_path_));
 140
 141 #if defined(OS_CHROMEOS)
 142   chromeos::LoginState::Shutdown();
 143 #endif
 144 }
 145
 146 }  // namespace test
 147 }  // namespace ash