chromium/chrome/renderer/companion/visual_query/visual_query_classifier_agent_browsertest.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/renderer/companion/visual_query/visual_query_classifier_agent.h"

#include <memory>
#include <optional>

#include "base/base_paths.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/callback.h"
#include "base/memory/discardable_memory_allocator.h"
#include "base/memory/raw_ptr.h"
#include "base/path_service.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/test_discardable_memory_allocator.h"
#include "base/test/test_timeouts.h"
#include "chrome/common/companion/visual_query.mojom.h"
#include "chrome/common/companion/visual_query/features.h"
#include "chrome/test/base/chrome_render_view_test.h"
#include "mojo/public/cpp/base/proto_wrapper.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "mojo/public/cpp/bindings/receiver_set.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/platform/browser_interface_broker_proxy.h"

_;
AtLeast;
Return;

namespace companion::visual_query {

namespace {

base::File LoadModelFile(const base::FilePath& model_file_path) { … }

base::FilePath model_file_path() { … }

base::FilePath img_file_path() { … }

}  // namespace

class TestVisualResultHandler : mojom::VisualSuggestionsResultHandler { … };

class FakeModelProvider : mojom::VisualSuggestionsModelProvider { … };

class VisualQueryClassifierAgentTest : public ChromeRenderViewTest { … };

TEST_F(VisualQueryClassifierAgentTest,
       StartClassification_SingleImageNonShoppy) { … }

TEST_F(VisualQueryClassifierAgentTest,
       StartClassification_SingleImageNonShoppy_AgentEnabled) { … }

TEST_F(VisualQueryClassifierAgentTest, StartClassification_NoImages) { … }

TEST_F(VisualQueryClassifierAgentTest, StartClassification_InvalidModel) { … }
}  // namespace companion::visual_query