chromium/components/page_content_annotations/core/test_page_content_annotator.cc

// Copyright 2021 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/page_content_annotations/core/test_page_content_annotator.h"

#include "base/task/sequenced_task_runner.h"

namespace page_content_annotations {

TestPageContentAnnotator::~TestPageContentAnnotator() = default;
TestPageContentAnnotator::TestPageContentAnnotator() = default;

void TestPageContentAnnotator::Annotate(BatchAnnotationCallback callback,
                                        const std::vector<std::string>& inputs,
                                        AnnotationType annotation_type) { … }

void TestPageContentAnnotator::SetAlwaysHang(bool hang) { … }

std::optional<optimization_guide::ModelInfo>
TestPageContentAnnotator::GetModelInfoForType(
    AnnotationType annotation_type) const { … }

void TestPageContentAnnotator::UseVisibilityScores(
    const std::optional<optimization_guide::ModelInfo>& model_info,
    const base::flat_map<std::string, double>& visibility_scores_for_input) { … }

bool TestPageContentAnnotator::ModelRequestedForType(
    AnnotationType type) const { … }

void TestPageContentAnnotator::RequestAndNotifyWhenModelAvailable(
    AnnotationType type,
    base::OnceCallback<void(bool)> callback) { … }

}  // namespace page_content_annotations