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
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83

ash / metrics / unlock_throughput_recorder_unittest.cc [blame]

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

#include "ash/metrics/unlock_throughput_recorder.h"

#include <memory>
#include <vector>

#include "ash/login/ui/login_test_base.h"
#include "ash/session/test_session_controller_client.h"
#include "ash/shell.h"
#include "ash/test/animation_metrics_test_util.h"
#include "ash/wm/tablet_mode/tablet_mode_controller.h"
#include "base/test/metrics/histogram_tester.h"
#include "chromeos/ash/components/login/login_state/login_state.h"

namespace ash {
namespace {
constexpr char kAshUnlockAnimationSmoothnessTabletMode[] =
    "Ash.UnlockAnimation.Smoothness.TabletMode";
constexpr char kAshUnlockAnimationSmoothnessClamshellMode[] =
    "Ash.UnlockAnimation.Smoothness.ClamshellMode";
}  // namespace

class UnlockThroughputRecorderTest : public LoginTestBase,
                                     public testing::WithParamInterface<bool> {
 public:
  UnlockThroughputRecorderTest() = default;
  UnlockThroughputRecorderTest(const UnlockThroughputRecorderTest&) = delete;
  UnlockThroughputRecorderTest& operator=(const UnlockThroughputRecorderTest&) =
      delete;
  ~UnlockThroughputRecorderTest() override = default;

  // LoginTestBase:
  void SetUp() override {
    LoginTestBase::SetUp();
    histogram_tester_ = std::make_unique<base::HistogramTester>();
  }

 protected:
  void LoginOwner() {
    SimulateUserLogin(kDefaultUserEmail);
    LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE,
                                        LoginState::LOGGED_IN_USER_REGULAR);
  }

  void EnableTabletMode(bool enable) {
    Shell::Get()->tablet_mode_controller()->SetEnabledForTest(enable);
  }

  void LockScreenAndAnimate() {
    GetSessionControllerClient()->LockScreen();
    test::RunSimpleAnimation();
  }

  void UnlockScreenAndAnimate() {
    GetSessionControllerClient()->UnlockScreen();
    test::RunSimpleAnimation();
  }

  // Used to verify recorded data.
  std::unique_ptr<base::HistogramTester> histogram_tester_;
};

// Boolean parameter controls tablet mode.
INSTANTIATE_TEST_SUITE_P(All, UnlockThroughputRecorderTest, testing::Bool());

TEST_P(UnlockThroughputRecorderTest, ReportUnlock) {
  LoginOwner();

  EnableTabletMode(GetParam());

  LockScreenAndAnimate();
  UnlockScreenAndAnimate();

  test::MetricsWaiter(histogram_tester_.get(),
                      GetParam() ? kAshUnlockAnimationSmoothnessTabletMode
                                 : kAshUnlockAnimationSmoothnessClamshellMode)
      .Wait();
}

}  // namespace ash