chromium/chrome/browser/companion/visual_query/visual_query_suggestions_service_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 "chrome/browser/companion/visual_query/visual_query_suggestions_service.h"

#include <optional>

#include "base/containers/flat_set.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/memory/scoped_refptr.h"
#include "base/path_service.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "base/test/task_environment.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/models.pb.h"
#include "mojo/public/cpp/base/proto_wrapper.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

VisualQuerySuggestionsService;

namespace {

base::FilePath model_file_path() { … }

void GetModelWithMetadataCallback(
    base::File model,
    std::optional<mojo_base::ProtoWrapper> config_proto) { … }
}  // namespace

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

TEST_F(VisualQuerySuggestionsServiceTest, OnModelUpdated) { … }

TEST_F(VisualQuerySuggestionsServiceTest,
       OnModelUpdated_BadOptimizationTarget) { … }

TEST_F(VisualQuerySuggestionsServiceTest, OnModelUpdated_InvalidModelFile) { … }

TEST_F(VisualQuerySuggestionsServiceTest, OnModelUpdated_ModelAlreadyLoaded) { … }

TEST_F(VisualQuerySuggestionsServiceTest, OnModelUpdated_NullModelUpdate) { … }