| Index: net/dns/host_resolver_mojo_unittest.cc
|
| diff --git a/net/dns/host_resolver_mojo_unittest.cc b/net/dns/host_resolver_mojo_unittest.cc
|
| index 0c791a101f3695d4225bf1edb8bfffeaa0b3d051..e398123ed29012ac5ddda5b50bad7c4e886f1957 100644
|
| --- a/net/dns/host_resolver_mojo_unittest.cc
|
| +++ b/net/dns/host_resolver_mojo_unittest.cc
|
| @@ -14,7 +14,6 @@
|
| #include "net/base/net_errors.h"
|
| #include "net/base/request_priority.h"
|
| #include "net/base/test_completion_callback.h"
|
| -#include "net/dns/mojo_host_type_converters.h"
|
| #include "net/test/event_waiter.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| @@ -55,33 +54,32 @@ struct HostResolverAction {
|
| RETAIN,
|
| };
|
|
|
| - static std::unique_ptr<HostResolverAction> ReturnError(Error error) {
|
| - std::unique_ptr<HostResolverAction> result(new HostResolverAction);
|
| - result->error = error;
|
| + static HostResolverAction ReturnError(Error error) {
|
| + HostResolverAction result;
|
| + result.error = error;
|
| return result;
|
| }
|
|
|
| - static std::unique_ptr<HostResolverAction> ReturnResult(
|
| - const AddressList& address_list) {
|
| - std::unique_ptr<HostResolverAction> result(new HostResolverAction);
|
| - result->addresses = interfaces::AddressList::From(address_list);
|
| + static HostResolverAction ReturnResult(const AddressList& address_list) {
|
| + HostResolverAction result;
|
| + result.addresses = address_list;
|
| return result;
|
| }
|
|
|
| - static std::unique_ptr<HostResolverAction> DropRequest() {
|
| - std::unique_ptr<HostResolverAction> result(new HostResolverAction);
|
| - result->action = DROP;
|
| + static HostResolverAction DropRequest() {
|
| + HostResolverAction result;
|
| + result.action = DROP;
|
| return result;
|
| }
|
|
|
| - static std::unique_ptr<HostResolverAction> RetainRequest() {
|
| - std::unique_ptr<HostResolverAction> result(new HostResolverAction);
|
| - result->action = RETAIN;
|
| + static HostResolverAction RetainRequest() {
|
| + HostResolverAction result;
|
| + result.action = RETAIN;
|
| return result;
|
| }
|
|
|
| Action action = COMPLETE;
|
| - interfaces::AddressListPtr addresses;
|
| + AddressList addresses;
|
| Error error = OK;
|
| };
|
|
|
| @@ -91,19 +89,19 @@ class MockMojoHostResolver : public HostResolverMojo::Impl {
|
| const base::Closure& request_connection_error_callback);
|
| ~MockMojoHostResolver() override;
|
|
|
| - void AddAction(std::unique_ptr<HostResolverAction> action);
|
| + void AddAction(HostResolverAction action);
|
|
|
| - const mojo::Array<interfaces::HostResolverRequestInfoPtr>& requests() {
|
| + const std::vector<HostResolver::RequestInfo>& requests() {
|
| return requests_received_;
|
| }
|
|
|
| - void ResolveDns(interfaces::HostResolverRequestInfoPtr request_info,
|
| + void ResolveDns(std::unique_ptr<HostResolver::RequestInfo> request_info,
|
| interfaces::HostResolverRequestClientPtr client) override;
|
|
|
| private:
|
| - std::vector<std::unique_ptr<HostResolverAction>> actions_;
|
| + std::vector<HostResolverAction> actions_;
|
| size_t results_returned_ = 0;
|
| - mojo::Array<interfaces::HostResolverRequestInfoPtr> requests_received_;
|
| + std::vector<HostResolver::RequestInfo> requests_received_;
|
| const base::Closure request_connection_error_callback_;
|
| std::vector<std::unique_ptr<MockMojoHostResolverRequest>> requests_;
|
| };
|
| @@ -117,20 +115,19 @@ MockMojoHostResolver::~MockMojoHostResolver() {
|
| EXPECT_EQ(results_returned_, actions_.size());
|
| }
|
|
|
| -void MockMojoHostResolver::AddAction(
|
| - std::unique_ptr<HostResolverAction> action) {
|
| +void MockMojoHostResolver::AddAction(HostResolverAction action) {
|
| actions_.push_back(std::move(action));
|
| }
|
|
|
| void MockMojoHostResolver::ResolveDns(
|
| - interfaces::HostResolverRequestInfoPtr request_info,
|
| + std::unique_ptr<HostResolver::RequestInfo> request_info,
|
| interfaces::HostResolverRequestClientPtr client) {
|
| - requests_received_.push_back(std::move(request_info));
|
| + requests_received_.push_back(std::move(*request_info));
|
| ASSERT_LE(results_returned_, actions_.size());
|
| - switch (actions_[results_returned_]->action) {
|
| + switch (actions_[results_returned_].action) {
|
| case HostResolverAction::COMPLETE:
|
| - client->ReportResult(actions_[results_returned_]->error,
|
| - std::move(actions_[results_returned_]->addresses));
|
| + client->ReportResult(actions_[results_returned_].error,
|
| + std::move(actions_[results_returned_].addresses));
|
| break;
|
| case HostResolverAction::RETAIN:
|
| requests_.push_back(base::WrapUnique(new MockMojoHostResolverRequest(
|
| @@ -191,11 +188,11 @@ TEST_F(HostResolverMojoTest, Basic) {
|
| EXPECT_EQ(address_list[1], result[1]);
|
|
|
| ASSERT_EQ(1u, mock_resolver_->requests().size());
|
| - interfaces::HostResolverRequestInfo& request = *mock_resolver_->requests()[0];
|
| - EXPECT_EQ("example.com", request.host.To<std::string>());
|
| - EXPECT_EQ(12345, request.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::UNSPECIFIED, request.address_family);
|
| - EXPECT_FALSE(request.is_my_ip_address);
|
| + const HostResolver::RequestInfo& request = mock_resolver_->requests()[0];
|
| + EXPECT_EQ("example.com", request.hostname());
|
| + EXPECT_EQ(12345, request.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_UNSPECIFIED, request.address_family());
|
| + EXPECT_FALSE(request.is_my_ip_address());
|
| }
|
|
|
| TEST_F(HostResolverMojoTest, ResolveCachedResult) {
|
| @@ -268,18 +265,16 @@ TEST_F(HostResolverMojoTest, Multiple) {
|
| ASSERT_EQ(0u, result2.size());
|
|
|
| ASSERT_EQ(2u, mock_resolver_->requests().size());
|
| - interfaces::HostResolverRequestInfo& request1 =
|
| - *mock_resolver_->requests()[0];
|
| - EXPECT_EQ("example.com", request1.host.To<std::string>());
|
| - EXPECT_EQ(12345, request1.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::IPV4, request1.address_family);
|
| - EXPECT_TRUE(request1.is_my_ip_address);
|
| - interfaces::HostResolverRequestInfo& request2 =
|
| - *mock_resolver_->requests()[1];
|
| - EXPECT_EQ("example.org", request2.host.To<std::string>());
|
| - EXPECT_EQ(80, request2.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::IPV6, request2.address_family);
|
| - EXPECT_FALSE(request2.is_my_ip_address);
|
| + const HostResolver::RequestInfo& request1 = mock_resolver_->requests()[0];
|
| + EXPECT_EQ("example.com", request1.hostname());
|
| + EXPECT_EQ(12345, request1.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_IPV4, request1.address_family());
|
| + EXPECT_TRUE(request1.is_my_ip_address());
|
| + const HostResolver::RequestInfo& request2 = mock_resolver_->requests()[1];
|
| + EXPECT_EQ("example.org", request2.hostname());
|
| + EXPECT_EQ(80, request2.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_IPV6, request2.address_family());
|
| + EXPECT_FALSE(request2.is_my_ip_address());
|
| }
|
|
|
| TEST_F(HostResolverMojoTest, Error) {
|
| @@ -293,11 +288,11 @@ TEST_F(HostResolverMojoTest, Error) {
|
| EXPECT_TRUE(result.empty());
|
|
|
| ASSERT_EQ(1u, mock_resolver_->requests().size());
|
| - interfaces::HostResolverRequestInfo& request = *mock_resolver_->requests()[0];
|
| - EXPECT_EQ("example.com", request.host.To<std::string>());
|
| - EXPECT_EQ(8080, request.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::IPV4, request.address_family);
|
| - EXPECT_FALSE(request.is_my_ip_address);
|
| + const HostResolver::RequestInfo& request = mock_resolver_->requests()[0];
|
| + EXPECT_EQ("example.com", request.hostname());
|
| + EXPECT_EQ(8080, request.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_IPV4, request.address_family());
|
| + EXPECT_FALSE(request.is_my_ip_address());
|
| }
|
|
|
| TEST_F(HostResolverMojoTest, EmptyResult) {
|
| @@ -325,11 +320,11 @@ TEST_F(HostResolverMojoTest, Cancel) {
|
| EXPECT_TRUE(result.empty());
|
|
|
| ASSERT_EQ(1u, mock_resolver_->requests().size());
|
| - interfaces::HostResolverRequestInfo& request = *mock_resolver_->requests()[0];
|
| - EXPECT_EQ("example.com", request.host.To<std::string>());
|
| - EXPECT_EQ(80, request.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::IPV6, request.address_family);
|
| - EXPECT_FALSE(request.is_my_ip_address);
|
| + const HostResolver::RequestInfo& request = mock_resolver_->requests()[0];
|
| + EXPECT_EQ("example.com", request.hostname());
|
| + EXPECT_EQ(80, request.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_IPV6, request.address_family());
|
| + EXPECT_FALSE(request.is_my_ip_address());
|
| }
|
|
|
| TEST_F(HostResolverMojoTest, ImplDropsClientConnection) {
|
| @@ -341,11 +336,11 @@ TEST_F(HostResolverMojoTest, ImplDropsClientConnection) {
|
| EXPECT_TRUE(result.empty());
|
|
|
| ASSERT_EQ(1u, mock_resolver_->requests().size());
|
| - interfaces::HostResolverRequestInfo& request = *mock_resolver_->requests()[0];
|
| - EXPECT_EQ("example.com", request.host.To<std::string>());
|
| - EXPECT_EQ(1, request.port);
|
| - EXPECT_EQ(interfaces::AddressFamily::UNSPECIFIED, request.address_family);
|
| - EXPECT_FALSE(request.is_my_ip_address);
|
| + const HostResolver::RequestInfo& request = mock_resolver_->requests()[0];
|
| + EXPECT_EQ("example.com", request.hostname());
|
| + EXPECT_EQ(1, request.port());
|
| + EXPECT_EQ(ADDRESS_FAMILY_UNSPECIFIED, request.address_family());
|
| + EXPECT_FALSE(request.is_my_ip_address());
|
| }
|
|
|
| TEST_F(HostResolverMojoTest, ResolveFromCache_Miss) {
|
|
|