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
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/http/http_proxy_client_socket.h"
#include "build/build_config.h"
#include "net/base/address_list.h"
#include "net/base/host_port_pair.h"
#include "net/base/proxy_chain.h"
#include "net/socket/next_proto.h"
#include "net/socket/socket_tag.h"
#include "net/socket/socket_test_util.h"
#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
namespace {
TEST(HttpProxyClientSocketTest, Tag) {
StaticSocketDataProvider data;
auto tagging_sock = std::make_unique<MockTaggingStreamSocket>(
std::make_unique<MockTCPClientSocket>(AddressList(),
nullptr /* net_log */, &data));
auto* tagging_sock_ptr = tagging_sock.get();
// |socket| takes ownership of |tagging_sock|, but the test keeps a non-owning
// pointer to it.
HttpProxyClientSocket socket(
std::move(tagging_sock), /*user_agent=*/"", HostPortPair(), ProxyChain(),
/*proxy_chain_index=*/0,
/*http_auth_controller=*/nullptr,
/*proxy_delegate=*/nullptr, TRAFFIC_ANNOTATION_FOR_TESTS);
EXPECT_EQ(tagging_sock_ptr->tag(), SocketTag());
#if BUILDFLAG(IS_ANDROID)
SocketTag tag(0x12345678, 0x87654321);
socket.ApplySocketTag(tag);
EXPECT_EQ(tagging_sock_ptr->tag(), tag);
#endif // BUILDFLAG(IS_ANDROID)
}
} // namespace
} // namespace net
|
