blob: ab81ce6b27df76981cd441f48b61743766e70f66 [file] [log] [blame]
// Copyright 2018 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 "ios/web/public/network_context_owner.h"
#include "base/run_loop.h"
#include "ios/web/public/test/test_web_thread_bundle.h"
#include "net/url_request/url_request_context.h"
#include "net/url_request/url_request_context_getter.h"
#include "net/url_request/url_request_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
namespace web {
class NetworkContextOwnerTest : public PlatformTest {
protected:
NetworkContextOwnerTest()
: saw_connection_error_(false),
context_getter_(base::MakeRefCounted<net::TestURLRequestContextGetter>(
WebThread::GetTaskRunnerForThread(WebThread::IO))) {}
~NetworkContextOwnerTest() override {
// Tests should cleanup after themselves.
EXPECT_EQ(network_context_owner_.get(), nullptr);
}
void WatchForErrors() {
ASSERT_TRUE(network_context_.is_bound());
network_context_.set_connection_error_handler(base::BindOnce(
&NetworkContextOwnerTest::SawError, base::Unretained(this)));
}
void SawError() { saw_connection_error_ = true; }
bool saw_connection_error_;
TestWebThreadBundle test_web_thread_bundle_;
scoped_refptr<net::TestURLRequestContextGetter> context_getter_;
network::mojom::NetworkContextPtr network_context_;
std::unique_ptr<NetworkContextOwner> network_context_owner_;
};
// Test that NetworkContextOwner actually creates a NetworkContext owner and
// connects a pipe to it, and destroys its end of the pipe when it's gone.
TEST_F(NetworkContextOwnerTest, Basic) {
EXPECT_FALSE(network_context_.is_bound());
network_context_owner_ = std::make_unique<NetworkContextOwner>(
context_getter_.get(), &network_context_);
EXPECT_TRUE(network_context_.is_bound());
WatchForErrors();
base::RunLoop().RunUntilIdle();
EXPECT_FALSE(saw_connection_error_);
web::WebThread::DeleteSoon(web::WebThread::IO, FROM_HERE,
network_context_owner_.release());
base::RunLoop().RunUntilIdle();
EXPECT_TRUE(saw_connection_error_); // other end gone
}
// Test to make sure that explicit shutdown of URLRequestContextGetter destroys
// the NetworkContext object as expected.
TEST_F(NetworkContextOwnerTest, ShutdownHandling) {
EXPECT_FALSE(network_context_.is_bound());
network_context_owner_ = std::make_unique<NetworkContextOwner>(
context_getter_.get(), &network_context_);
EXPECT_TRUE(network_context_.is_bound());
WatchForErrors();
base::RunLoop().RunUntilIdle();
EXPECT_FALSE(saw_connection_error_);
web::WebThread::PostTask(
web::WebThread::IO, FROM_HERE,
base::BindOnce(
&net::TestURLRequestContextGetter::NotifyContextShuttingDown,
context_getter_));
base::RunLoop().RunUntilIdle();
EXPECT_TRUE(saw_connection_error_); // other end gone post-shutdown.
web::WebThread::DeleteSoon(web::WebThread::IO, FROM_HERE,
network_context_owner_.release());
}
} // namespace web