1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70


ash / wm / lock_state_controller_test_api.h [blame]

// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef ASH_WM_LOCK_STATE_CONTROLLER_TEST_API_H_
#define ASH_WM_LOCK_STATE_CONTROLLER_TEST_API_H_

#include "ash/wm/lock_state_controller.h"
#include "base/memory/raw_ptr.h"

namespace ui {
class Layer;
}  // namespace ui

namespace ash {

// Helper class used by tests to access LockStateController's internal state.
class LockStateControllerTestApi {
 public:
  explicit LockStateControllerTestApi(LockStateController* controller);

  LockStateControllerTestApi(const LockStateControllerTestApi&) = delete;
  LockStateControllerTestApi& operator=(const LockStateControllerTestApi&) =
      delete;

  ~LockStateControllerTestApi();

  void set_shutdown_controller(ShutdownController* shutdown_controller) {
    controller_->shutdown_controller_ = shutdown_controller;
  }

  bool shutdown_timer_is_running() const {
    return controller_->cancelable_shutdown_timer_.IsRunning();
  }
  bool real_shutdown_timer_is_running() const {
    return controller_->session_state_change_timer_.IsRunning();
  }
  bool is_animating_lock() const { return controller_->animating_lock_; }

  void trigger_shutdown_timeout() {
    controller_->cancelable_shutdown_timer_.FireNow();
  }
  void trigger_real_shutdown_timeout() {
    controller_->session_state_change_timer_.FireNow();
  }

  void set_informed_restore_image_callback(base::OnceClosure callback) {
    controller_->informed_restore_image_callback_for_test_ =
        std::move(callback);
  }

  void disable_screenshot_timeout_for_test(bool value) {
    controller_->disable_screenshot_timeout_for_test_ = value;
  }

  void trigger_take_screenshot_timeout() const {
    controller_->take_screenshot_fail_timer_.FireNow();
  }

  const ui::Layer* mirror_wallpaper_layer() const {
    return controller_->mirror_wallpaper_layer_.get();
  }

 private:
  raw_ptr<LockStateController> controller_;  // not owned
};

}  // namespace ash

#endif  // ASH_WM_LOCK_STATE_CONTROLLER_TEST_API_H_