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
|
// 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 "chromeos/ash/components/timer_factory/fake_timer_factory.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "chromeos/ash/components/timer_factory/fake_one_shot_timer.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace ash::timer_factory {
FakeTimerFactory::FakeTimerFactory() = default;
FakeTimerFactory::~FakeTimerFactory() = default;
std::unique_ptr<ash::timer_factory::TimerFactory>
FakeTimerFactory::Factory::CreateInstance() {
EXPECT_FALSE(instance_);
auto instance = std::make_unique<FakeTimerFactory>();
instance_ = instance.get();
return instance;
}
std::unique_ptr<base::OneShotTimer> FakeTimerFactory::CreateOneShotTimer() {
++num_instances_created_;
auto fake_one_shot_timer = std::make_unique<FakeOneShotTimer>(
base::BindOnce(&FakeTimerFactory::OnOneShotTimerDeleted,
weak_ptr_factory_.GetWeakPtr()));
id_for_last_created_one_shot_timer_ = fake_one_shot_timer->id();
id_to_active_one_shot_timer_map_.insert_or_assign(fake_one_shot_timer->id(),
fake_one_shot_timer.get());
return fake_one_shot_timer;
}
void FakeTimerFactory::FireAll() {
// Make a copy because firing a timer will usually destroy it. This calls
// OnOneShotTimerDeleted and removes it from |id_to_timer_map_|.
auto id_to_timer_map_copy = id_to_active_one_shot_timer_map_;
for (auto [id, timer] : id_to_timer_map_copy) {
timer->Fire();
}
}
void FakeTimerFactory::OnOneShotTimerDeleted(
const base::UnguessableToken& deleted_timer_id) {
id_to_active_one_shot_timer_map_.erase(deleted_timer_id);
}
} // namespace ash::timer_factory
|
