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

#include <memory>

#include "base/containers/flat_set.h"
#include "base/debug/stack_trace.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/bind.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/test_timeouts.h"
#include "base/time/time.h"
#include "chrome/browser/companion/core/companion_metrics_logger.h"
#include "chrome/common/companion/visual_query.mojom.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.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 "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/mock_render_process_host.h"
#include "content/public/test/test_renderer_host.h"
#include "content/public/test/web_contents_tester.h"
#include "services/metrics/public/cpp/ukm_source.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace companion::visual_query {

namespace {

constexpr char kValidUrl[] = …;

base::FilePath model_file_path() { … }

base::FilePath invalid_file_path() { … }

const SkBitmap create_bitmap(int width, int height, int r, int g, int b) { … }

}  // namespace

class VisualQueryClassifierHostTest : public ChromeRenderViewHostTestHarness { … };

TEST_F(VisualQueryClassifierHostTest, StartClassification) { … }

TEST_F(VisualQueryClassifierHostTest, StartClassification_NoModelSet) { … }

TEST_F(VisualQueryClassifierHostTest, StartClassification_WithInvalidModel) { … }

TEST_F(VisualQueryClassifierHostTest, StartClassification_WithCancellation) { … }

TEST_F(VisualQueryClassifierHostTest, HandleClassification) { … }

}  // namespace companion::visual_query