blob: 01563f52f9ec1c28f06299a5765821f2c3333b47 [file] [log] [blame]
// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "services/network/dhcp_pac_file_fetcher_mojo.h"
#include "net/base/test_completion_callback.h"
#include "net/proxy_resolution/mock_pac_file_fetcher.h"
#include "net/test/gtest_util.h"
#include "net/test/test_with_task_environment.h"
#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
#include "net/url_request/url_request_test_util.h"
#include "services/network/dhcp_pac_file_fetcher_mojo.h"
#include "services/network/mock_mojo_dhcp_wpad_url_client.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
using net::test::IsError;
using net::test::IsOk;
} // namespace
namespace network {
class DhcpPacFileFetcherMojoTest : public testing::Test {
public:
DhcpPacFileFetcherMojoTest() = default;
~DhcpPacFileFetcherMojoTest() override {}
protected:
void CreateFetcher(const std::string& pac_url) {
net::TestURLRequestContext context;
dhcp_pac_file_fetcher_mojo_ = std::make_unique<DhcpPacFileFetcherMojo>(
&context,
network::mojom::DhcpWpadUrlClientPtr(
network::MockMojoDhcpWpadUrlClient::CreateWithSelfOwnedReceiver(
pac_url)));
mock_pac_file_fetcher_ = new net::MockPacFileFetcher();
dhcp_pac_file_fetcher_mojo_->SetPacFileFetcherForTesting(
base::WrapUnique(mock_pac_file_fetcher_));
}
std::unique_ptr<DhcpPacFileFetcherMojo> dhcp_pac_file_fetcher_mojo_;
net::MockPacFileFetcher* mock_pac_file_fetcher_;
private:
base::test::TaskEnvironment task_environment_;
};
// Test that the PAC URL set by the client is used.
TEST_F(DhcpPacFileFetcherMojoTest, UsePacSctipt) {
GURL pac_url("http://wpad.test.com/wpad.dat");
CreateFetcher(pac_url.spec());
net::TestCompletionCallback callback;
base::string16 pac_text;
dhcp_pac_file_fetcher_mojo_->Fetch(&pac_text, callback.callback(),
net::NetLogWithSource(),
TRAFFIC_ANNOTATION_FOR_TESTS);
mock_pac_file_fetcher_->WaitUntilFetch();
EXPECT_EQ(pac_url, mock_pac_file_fetcher_->pending_request_url());
mock_pac_file_fetcher_->NotifyFetchCompletion(net::OK, "script");
EXPECT_THAT(callback.WaitForResult(), IsOk());
}
// Test that error is returned when PAC URL is missing.
TEST_F(DhcpPacFileFetcherMojoTest, PacScriptMissing) {
CreateFetcher(std::string());
net::TestCompletionCallback callback;
base::string16 pac_text;
dhcp_pac_file_fetcher_mojo_->Fetch(&pac_text, callback.callback(),
net::NetLogWithSource(),
TRAFFIC_ANNOTATION_FOR_TESTS);
EXPECT_THAT(callback.WaitForResult(), net::ERR_PAC_NOT_IN_DHCP);
}
} // namespace network