Home | History | Annotate | Download | only in login 
      1 // Copyright 2014 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/chromeos/login/screenshot_testing_mixin.h"
      6 
      7 #include "chrome/browser/browser_process.h"
      8 #include "chrome/browser/chrome_notification_types.h"
      9 #include "content/public/browser/browser_thread.h"
     10 #include "content/public/browser/notification_service.h"
     11 #include "content/public/browser/notification_types.h"
     12 #include "ui/compositor/compositor_switches.h"
     13 
     14 namespace chromeos {
     15 
     16 ScreenshotTestingMixin::ScreenshotTestingMixin() {
     17 }
     18 
     19 ScreenshotTestingMixin::~ScreenshotTestingMixin() {
     20 }
     21 
     22 void ScreenshotTestingMixin::SetUpInProcessBrowserTestFixture() {
     23   enable_test_screenshots_ = screenshot_tester_.TryInitialize();
     24 }
     25 
     26 void ScreenshotTestingMixin::SetUpCommandLine(base::CommandLine* command_line) {
     27   if (enable_test_screenshots_) {
     28     command_line->AppendSwitch(switches::kEnablePixelOutputInTests);
     29     command_line->AppendSwitch(switches::kUIEnableImplSidePainting);
     30   }
     31 }
     32 
     33 void ScreenshotTestingMixin::RunScreenshotTesting(
     34     const std::string& test_name) {
     35   if (enable_test_screenshots_) {
     36     SynchronizeAnimationLoadWithCompositor();
     37     screenshot_tester_.Run(test_name);
     38   }
     39 }
     40 
     41 // Current implementation is a mockup.
     42 // It simply waits for 5 seconds, assuming that this time is enough for
     43 // animation to load completely.
     44 // TODO(elizavetai): Replace this temporary hack with getting a
     45 // valid notification from compositor.
     46 void ScreenshotTestingMixin::SynchronizeAnimationLoadWithCompositor() {
     47   base::RunLoop waiter;
     48   animation_waiter_quitter_ = waiter.QuitClosure();
     49   timer_.Start(FROM_HERE,
     50                base::TimeDelta::FromSeconds(5),
     51                this,
     52                &ScreenshotTestingMixin::HandleAnimationLoad);
     53   waiter.Run();
     54 }
     55 
     56 void ScreenshotTestingMixin::HandleAnimationLoad() {
     57   timer_.Stop();
     58   content::BrowserThread::PostTask(
     59       content::BrowserThread::UI, FROM_HERE, animation_waiter_quitter_);
     60 }
     61 
     62 }  // namespace chromeos
     63