components/browsing_data/content/mock_browsing_data_quota_helper.cc

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

#include "components/browsing_data/content/mock_browsing_data_quota_helper.h"

#include "content/public/browser/browser_thread.h"
#include "testing/gtest/include/gtest/gtest.h"

using content::BrowserThread;

MockBrowsingDataQuotaHelper::MockBrowsingDataQuotaHelper() = default;

MockBrowsingDataQuotaHelper::~MockBrowsingDataQuotaHelper() = default;

void MockBrowsingDataQuotaHelper::StartFetching(FetchResultCallback callback) {
  ASSERT_FALSE(callback.is_null());
  ASSERT_TRUE(callback_.is_null());
  callback_ = std::move(callback);
}

void MockBrowsingDataQuotaHelper::DeleteHostData(
    const std::string& host,
    blink::mojom::StorageType type) {}

void MockBrowsingDataQuotaHelper::AddHost(const blink::StorageKey& storage_key,
                                          int64_t temporary_usage,
                                          int64_t syncable_usage) {
  response_.push_back(QuotaInfo(storage_key, temporary_usage, syncable_usage));
}

void MockBrowsingDataQuotaHelper::AddQuotaSamples() {
  AddHost(blink::StorageKey::CreateFromStringForTesting("http://quotahost1"), 1,
          1);
  AddHost(blink::StorageKey::CreateFromStringForTesting("https://quotahost2"),
          10, 10);
}

void MockBrowsingDataQuotaHelper::Notify() {
  std::move(callback_).Run(response_);
  response_.clear();
}