46 lines
1.5 KiB
C++
46 lines
1.5 KiB
C++
// 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_server.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(), ProxyServer(),
|
|
/*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
|