chromium/components/omnibox/browser/autocomplete_scoring_model_handler_unittest.cc

// Copyright 2023 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/omnibox/browser/autocomplete_scoring_model_handler.h"

#include "base/base_paths.h"
#include "base/path_service.h"
#include "base/task/sequenced_task_runner.h"
#include "base/test/task_environment.h"
#include "components/omnibox/browser/autocomplete_scoring_model_executor.h"
#include "components/optimization_guide/core/test_model_info_builder.h"
#include "components/optimization_guide/core/test_optimization_guide_model_provider.h"
#include "components/optimization_guide/proto/autocomplete_scoring_model_metadata.pb.h"
#include "components/optimization_guide/proto/models.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/metrics_proto/omnibox_scoring_signals.pb.h"

ScoringSignals;
AutocompleteScoringModelMetadata;
ScoringSignalSpec;
ScoringSignalTransformation;
ScoringSignalType;

namespace {

ScoringSignalSpec CreateScoringSignalSpec(
    ScoringSignalType type,
    std::optional<ScoringSignalTransformation> transformation = std::nullopt,
    std::optional<float> min_val = std::nullopt,
    std::optional<float> max_val = std::nullopt,
    std::optional<float> missing_val = std::nullopt) { … }

}  // namespace

class TestAutocompleteScoringModelExecutor
    : public AutocompleteScoringModelExecutor { … };

class AutocompleteScoringModelHandlerTest : public testing::Test { … };

TEST_F(AutocompleteScoringModelHandlerTest,
       ExtractInputFromScoringSignalsTest) { … }

TEST_F(AutocompleteScoringModelHandlerTest, GetBatchModelInputTest) { … }