| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/browsing_data/mock_browsing_data_media_license_helper.h" |
| |
| #include <algorithm> |
| |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| MockBrowsingDataMediaLicenseHelper::MockBrowsingDataMediaLicenseHelper( |
| Profile* profile) {} |
| |
| MockBrowsingDataMediaLicenseHelper::~MockBrowsingDataMediaLicenseHelper() {} |
| |
| void MockBrowsingDataMediaLicenseHelper::StartFetching( |
| const FetchCallback& callback) { |
| ASSERT_FALSE(callback.is_null()); |
| ASSERT_TRUE(callback_.is_null()); |
| callback_ = callback; |
| } |
| |
| void MockBrowsingDataMediaLicenseHelper::DeleteMediaLicenseOrigin( |
| const GURL& origin) { |
| auto entry = std::find_if(media_licenses_.begin(), media_licenses_.end(), |
| [origin](const MediaLicenseInfo& entry) { |
| return entry.origin == origin; |
| }); |
| ASSERT_TRUE(entry != media_licenses_.end()); |
| media_licenses_.erase(entry); |
| } |
| |
| void MockBrowsingDataMediaLicenseHelper::AddMediaLicenseSamples() { |
| const GURL kOrigin1("https://media1/"); |
| const GURL kOrigin2("https://media2/"); |
| const base::Time ten_days_ago = |
| base::Time::Now() - base::TimeDelta::FromDays(10); |
| const base::Time twenty_days_ago = |
| base::Time::Now() - base::TimeDelta::FromDays(20); |
| |
| media_licenses_.push_back(MediaLicenseInfo(kOrigin1, 1000, ten_days_ago)); |
| media_licenses_.push_back(MediaLicenseInfo(kOrigin2, 50, twenty_days_ago)); |
| } |
| |
| void MockBrowsingDataMediaLicenseHelper::Notify() { |
| callback_.Run(media_licenses_); |
| callback_ = FetchCallback(); |
| } |
| |
| bool MockBrowsingDataMediaLicenseHelper::AllDeleted() { |
| return media_licenses_.empty(); |
| } |