chromium/components/optimization_guide/core/test_model_info_builder.h

// 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.

#ifndef COMPONENTS_OPTIMIZATION_GUIDE_CORE_TEST_MODEL_INFO_BUILDER_H_
#define COMPONENTS_OPTIMIZATION_GUIDE_CORE_TEST_MODEL_INFO_BUILDER_H_

#include <memory>
#include <optional>

#include "base/containers/flat_set.h"
#include "components/optimization_guide/core/model_info.h"
#include "components/optimization_guide/proto/models.pb.h"

namespace optimization_guide {

// A builder for ModelInfo's that should only be used in testing.
// In production code, these classes are assembled by internal OptGuide code but
// external callers may wish to create their own for testing. An unmodified
// builder will return a valid ModelInfo when built with no
// metadata and dummy values for the model file path (which won't exist) and
// version.
class TestModelInfoBuilder { … };

}  // namespace optimization_guide

#endif  // COMPONENTS_OPTIMIZATION_GUIDE_CORE_TEST_MODEL_INFO_BUILDER_H_