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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ash/arc/idle_manager/arc_cpu_throttle_observer.h"
#include "base/memory/raw_ptr.h"
#include "base/test/bind.h"
#include "chrome/browser/ash/arc/idle_manager/arc_throttle_test_observer.h"
#include "chrome/browser/ash/arc/instance_throttle/arc_instance_throttle.h"
#include "chrome/browser/ash/arc/session/arc_session_manager.h"
#include "chrome/browser/ash/arc/test/test_arc_session_manager.h"
#include "chrome/test/base/testing_profile.h"
#include "chromeos/ash/components/dbus/concierge/concierge_client.h"
#include "chromeos/ash/experiences/arc/arc_prefs.h"
#include "chromeos/ash/experiences/arc/metrics/stability_metrics_manager.h"
#include "chromeos/ash/experiences/arc/power/arc_power_bridge.h"
#include "chromeos/ash/experiences/arc/session/arc_service_manager.h"
#include "chromeos/ash/experiences/arc/test/fake_arc_session.h"
#include "chromeos/dbus/power/power_manager_client.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/display/test/test_screen.h"
namespace arc {
class ArcCpuThrottleObserverTest : public testing::Test {
public:
ArcCpuThrottleObserverTest() = default;
ArcCpuThrottleObserverTest(const ArcCpuThrottleObserverTest&) = delete;
ArcCpuThrottleObserverTest& operator=(const ArcCpuThrottleObserverTest&) =
delete;
~ArcCpuThrottleObserverTest() override = default;
void SetUp() override {
chromeos::PowerManagerClient::InitializeFake();
ash::ConciergeClient::InitializeFake(/*fake_cicerone_client=*/nullptr);
service_manager_ = std::make_unique<ArcServiceManager>();
session_manager_ =
CreateTestArcSessionManager(std::make_unique<ArcSessionRunner>(
base::BindRepeating(FakeArcSession::Create)));
testing_profile_ = std::make_unique<TestingProfile>();
StabilityMetricsManager::Initialize(&local_state_);
prefs::RegisterLocalStatePrefs(local_state_.registry());
prefs::RegisterProfilePrefs(local_state_.registry());
arc_metrics_service_ = ArcMetricsService::GetForBrowserContextForTesting(
testing_profile_.get());
arc_metrics_service_->SetHistogramNamerCallback(base::BindLambdaForTesting(
[](const std::string& s) -> std::string { return s; }));
test_instance_throttle_ =
ArcInstanceThrottle::GetForBrowserContextForTesting(
testing_profile_.get());
DCHECK(test_instance_throttle_);
}
void TearDown() override {
arc::StabilityMetricsManager::Shutdown();
testing_profile_.reset();
session_manager_.reset();
service_manager_.reset();
chromeos::PowerManagerClient::Shutdown();
}
protected:
ArcCpuThrottleObserver* observer() { return &cpu_throttle_observer_; }
ArcInstanceThrottle* throttle() { return test_instance_throttle_; }
TestingProfile* profile() { return testing_profile_.get(); }
private:
content::BrowserTaskEnvironment task_environment_;
display::test::TestScreen test_screen_{/*create_display=*/true,
/*register_screen=*/true};
TestingPrefServiceSimple local_state_;
raw_ptr<ArcMetricsService, DanglingUntriaged> arc_metrics_service_ = nullptr;
ArcCpuThrottleObserver cpu_throttle_observer_;
std::unique_ptr<ArcServiceManager> service_manager_;
std::unique_ptr<ArcSessionManager> session_manager_;
std::unique_ptr<TestingProfile> testing_profile_;
raw_ptr<ArcInstanceThrottle, DanglingUntriaged> test_instance_throttle_;
};
TEST_F(ArcCpuThrottleObserverTest, TestConstructDestruct) {}
TEST_F(ArcCpuThrottleObserverTest, TestStatusChanges) {
unittest::ThrottleTestObserver test_observer;
EXPECT_EQ(0, test_observer.count());
EXPECT_FALSE(throttle()->HasServiceObserverForTesting(observer()));
// base::Unretained below: safe because all involved objects share scope.
observer()->StartObserving(
profile(), base::BindRepeating(&unittest::ThrottleTestObserver::Monitor,
base::Unretained(&test_observer)));
EXPECT_TRUE(throttle()->HasServiceObserverForTesting(observer()));
EXPECT_EQ(1, test_observer.count());
EXPECT_EQ(0, test_observer.active_count());
EXPECT_EQ(0, test_observer.enforced_count());
EXPECT_FALSE(observer()->active());
EXPECT_FALSE(observer()->enforced());
observer()->OnThrottle(false); // Not throttled.
EXPECT_EQ(2, test_observer.count());
EXPECT_EQ(1, test_observer.active_count());
EXPECT_EQ(0, test_observer.enforced_count());
EXPECT_TRUE(observer()->active());
EXPECT_FALSE(observer()->enforced());
observer()->OnThrottle(true); // Yes, throttled.
EXPECT_EQ(3, test_observer.count());
EXPECT_EQ(1, test_observer.active_count());
EXPECT_EQ(0, test_observer.enforced_count());
EXPECT_FALSE(observer()->active());
EXPECT_FALSE(observer()->enforced());
EXPECT_TRUE(throttle()->HasServiceObserverForTesting(observer()));
observer()->StopObserving();
EXPECT_FALSE(throttle()->HasServiceObserverForTesting(observer()));
}
} // namespace arc
|