chromium/components/autofill/core/browser/ml_model/autofill_ml_prediction_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/autofill/core/browser/ml_model/autofill_ml_prediction_model_handler.h"

#include <optional>
#include <vector>

#include "base/base_paths.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/path_service.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "components/autofill/core/browser/autofill_form_test_utils.h"
#include "components/autofill/core/browser/field_types.h"
#include "components/autofill/core/browser/form_structure.h"
#include "components/autofill/core/browser/heuristic_source.h"
#include "components/autofill/core/common/autofill_features.h"
#include "components/autofill/core/common/autofill_test_utils.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/autofill_field_classification_model_metadata.pb.h"
#include "components/optimization_guide/proto/common_types.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace autofill {

namespace {

// The matcher expects two arguments of types std::unique_ptr<AutofillField>
// and FieldType respectively.
MATCHER(MlTypeEq, "") { … }

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

}  // namespace

TEST_F(AutofillMlPredictionModelHandlerTest, GetModelPredictionsForForm) { … }

// Tests that predictions with a confidence below the threshold are reported as
// UNKNOWN_TYPE.
TEST_F(AutofillMlPredictionModelHandlerTest,
       GetModelPredictionsForForm_Threshold) { … }

TEST_F(AutofillMlPredictionModelHandlerTest, GetModelPredictionsForForms) { … }

}  // namespace autofill