Avi Drissman | 4a8573c | 2022-09-09 19:35:54 | [diff] [blame] | 1 | // Copyright 2019 The Chromium Authors |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 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/badging/test_badge_manager_delegate.h" |
| 6 | |
| 7 | #include "chrome/browser/badging/badge_manager.h" |
| 8 | #include "chrome/browser/badging/badge_manager_delegate.h" |
| 9 | #include "chrome/browser/profiles/profile.h" |
Song Fangzhen | ed83c14 | 2021-07-13 03:21:44 | [diff] [blame] | 10 | #include "chrome/browser/web_applications/web_app_registrar.h" |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 11 | |
| 12 | namespace badging { |
| 13 | |
Jay Harris | fb5bcba | 2019-10-01 06:40:43 | [diff] [blame] | 14 | TestBadgeManagerDelegate::TestBadgeManagerDelegate(Profile* profile, |
| 15 | BadgeManager* badge_manager) |
| 16 | : BadgeManagerDelegate(profile, badge_manager) {} |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 17 | |
| 18 | TestBadgeManagerDelegate::~TestBadgeManagerDelegate() = default; |
| 19 | |
| 20 | void TestBadgeManagerDelegate::SetOnBadgeChanged( |
| 21 | base::RepeatingCallback<void()> on_badge_changed) { |
| 22 | on_badge_changed_ = on_badge_changed; |
| 23 | } |
| 24 | |
Jay Harris | fb5bcba | 2019-10-01 06:40:43 | [diff] [blame] | 25 | void TestBadgeManagerDelegate::OnAppBadgeUpdated(const web_app::AppId& app_id) { |
| 26 | const auto& value = badge_manager()->GetBadgeValue(app_id); |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 27 | if (!value) |
Jay Harris | fb5bcba | 2019-10-01 06:40:43 | [diff] [blame] | 28 | cleared_badges_.push_back(app_id); |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 29 | else |
Jay Harris | fb5bcba | 2019-10-01 06:40:43 | [diff] [blame] | 30 | set_badges_.push_back(std::make_pair(app_id, value.value())); |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 31 | |
| 32 | if (on_badge_changed_) |
| 33 | on_badge_changed_.Run(); |
| 34 | } |
| 35 | |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 36 | void TestBadgeManagerDelegate::ResetBadges() { |
Jay Harris | fb5bcba | 2019-10-01 06:40:43 | [diff] [blame] | 37 | cleared_badges_.clear(); |
| 38 | set_badges_.clear(); |
Jay Harris | 1d535191 | 2019-09-02 07:47:14 | [diff] [blame] | 39 | } |
| 40 | |
| 41 | } // namespace badging |