1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "test_composebox_query_controller.h"
#include "base/strings/utf_string_conversions.h"
#include "components/lens/lens_features.h"
#include "google_apis/common/api_error_codes.h"
#include "testing/gtest/include/gtest/gtest.h"
using endpoint_fetcher::EndpointFetcher;
using endpoint_fetcher::EndpointFetcherCallback;
using endpoint_fetcher::EndpointResponse;
using endpoint_fetcher::HttpMethod;
FakeEndpointFetcher::FakeEndpointFetcher(EndpointResponse response)
: EndpointFetcher(
net::DefineNetworkTrafficAnnotation("compbosebox_mock_fetcher",
R"()")),
response_(response) {}
void FakeEndpointFetcher::PerformRequest(
EndpointFetcherCallback endpoint_fetcher_callback,
const char* key) {
if (!disable_responding_) {
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE,
base::BindOnce(std::move(endpoint_fetcher_callback),
std::make_unique<EndpointResponse>(response_)));
}
}
bool FakeVariationsClient::IsOffTheRecord() const {
return false;
}
variations::mojom::VariationsHeadersPtr
FakeVariationsClient::GetVariationsHeaders() const {
base::flat_map<variations::mojom::GoogleWebVisibility, std::string> headers =
{{variations::mojom::GoogleWebVisibility::FIRST_PARTY, "123xyz"}};
return variations::mojom::VariationsHeaders::New(headers);
}
TestComposeboxQueryController::TestComposeboxQueryController(
signin::IdentityManager* identity_manager,
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory,
version_info::Channel channel,
std::string locale,
TemplateURLService* template_url_service,
variations::VariationsClient* variations_client,
bool send_lns_surface)
: ComposeboxQueryController(identity_manager,
url_loader_factory,
channel,
locale,
template_url_service,
variations_client,
send_lns_surface) {}
TestComposeboxQueryController::~TestComposeboxQueryController() = default;
std::unique_ptr<EndpointFetcher>
TestComposeboxQueryController::CreateEndpointFetcher(
std::string request_string,
const GURL& fetch_url,
HttpMethod http_method,
base::TimeDelta timeout,
const std::vector<std::string>& request_headers,
const std::vector<std::string>& cors_exempt_headers,
UploadProgressCallback upload_progress_callback) {
std::string fake_server_response_string;
google_apis::ApiErrorCode fake_server_response_code =
google_apis::ApiErrorCode::HTTP_SUCCESS;
// Whether or not to disable the response.
bool disable_response = false;
bool is_cluster_info_request =
fetch_url == GURL(lens::features::GetLensOverlayClusterInfoEndpointUrl());
if (is_cluster_info_request) {
// Cluster info request.
num_cluster_info_fetch_requests_sent_++;
if (next_cluster_info_request_should_return_error_) {
fake_server_response_code =
google_apis::ApiErrorCode::HTTP_INTERNAL_SERVER_ERROR;
} else {
fake_server_response_string =
fake_cluster_info_response_.SerializeAsString();
}
} else {
num_file_upload_requests_sent_++;
last_sent_fetch_url_ = fetch_url;
if (next_file_upload_request_should_return_error_) {
fake_server_response_code =
google_apis::ApiErrorCode::HTTP_INTERNAL_SERVER_ERROR;
}
last_sent_file_upload_request_ = lens::LensOverlayServerRequest();
last_sent_file_upload_request_->ParseFromString(request_string);
}
last_sent_cors_exempt_headers_.clear();
for (const auto& header : cors_exempt_headers) {
last_sent_cors_exempt_headers_.push_back(header);
}
// Create the fake endpoint fetcher to return the fake response.
EndpointResponse fake_endpoint_response;
fake_endpoint_response.response = fake_server_response_string;
fake_endpoint_response.http_status_code = fake_server_response_code;
auto response = std::make_unique<FakeEndpointFetcher>(fake_endpoint_response);
response->disable_responding_ = disable_response;
return response;
}
void TestComposeboxQueryController::ResetRequestClusterInfoState(
int session_id) {
if (!enable_cluster_info_ttl_) {
return;
}
ComposeboxQueryController::ResetRequestClusterInfoState(session_id);
}
|
