| // Copyright 2014 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 "net/log/net_log_util.h" |
| |
| #include <set> |
| #include <vector> |
| |
| #include "base/files/file_path.h" |
| #include "base/test/task_environment.h" |
| #include "base/values.h" |
| #include "net/base/net_errors.h" |
| #include "net/base/test_completion_callback.h" |
| #include "net/http/http_cache.h" |
| #include "net/http/http_transaction.h" |
| #include "net/log/net_log_source.h" |
| #include "net/log/net_log_with_source.h" |
| #include "net/log/test_net_log.h" |
| #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
| #include "net/url_request/url_request_test_util.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace net { |
| |
| namespace { |
| |
| // Make sure GetNetConstants doesn't crash. |
| TEST(NetLogUtil, GetNetConstants) { |
| base::Value constants(GetNetConstants()); |
| } |
| |
| // Make sure GetNetInfo doesn't crash when called on contexts with and without |
| // caches, and they have the same number of elements. |
| TEST(NetLogUtil, GetNetInfo) { |
| base::test::TaskEnvironment task_environment; |
| |
| TestURLRequestContext context; |
| HttpCache* http_cache = context.http_transaction_factory()->GetCache(); |
| |
| // Get NetInfo when there's no cache backend (It's only created on first use). |
| EXPECT_FALSE(http_cache->GetCurrentBackend()); |
| base::Value net_info_without_cache( |
| GetNetInfo(&context, NET_INFO_ALL_SOURCES)); |
| EXPECT_FALSE(http_cache->GetCurrentBackend()); |
| EXPECT_GT(net_info_without_cache.DictSize(), 0u); |
| |
| // Fore creation of a cache backend, and get NetInfo again. |
| disk_cache::Backend* backend = nullptr; |
| EXPECT_EQ(OK, context.http_transaction_factory()->GetCache()->GetBackend( |
| &backend, TestCompletionCallback().callback())); |
| EXPECT_TRUE(http_cache->GetCurrentBackend()); |
| base::Value net_info_with_cache = GetNetInfo(&context, NET_INFO_ALL_SOURCES); |
| EXPECT_GT(net_info_with_cache.DictSize(), 0u); |
| |
| EXPECT_EQ(net_info_without_cache.DictSize(), net_info_with_cache.DictSize()); |
| } |
| |
| // Make sure CreateNetLogEntriesForActiveObjects works for requests from a |
| // single URLRequestContext. |
| TEST(NetLogUtil, CreateNetLogEntriesForActiveObjectsOneContext) { |
| base::test::TaskEnvironment task_environment; |
| |
| // Using same context for each iteration makes sure deleted requests don't |
| // appear in the list, or result in crashes. |
| TestURLRequestContext context(true); |
| TestNetLog net_log; |
| context.set_net_log(&net_log); |
| context.Init(); |
| TestDelegate delegate; |
| for (size_t num_requests = 0; num_requests < 5; ++num_requests) { |
| std::vector<std::unique_ptr<URLRequest>> requests; |
| for (size_t i = 0; i < num_requests; ++i) { |
| requests.push_back(context.CreateRequest(GURL("about:life"), |
| DEFAULT_PRIORITY, &delegate, |
| TRAFFIC_ANNOTATION_FOR_TESTS)); |
| } |
| std::set<URLRequestContext*> contexts; |
| contexts.insert(&context); |
| RecordingTestNetLog test_net_log; |
| CreateNetLogEntriesForActiveObjects(contexts, test_net_log.GetObserver()); |
| auto entry_list = test_net_log.GetEntries(); |
| ASSERT_EQ(num_requests, entry_list.size()); |
| |
| for (size_t i = 0; i < num_requests; ++i) { |
| EXPECT_EQ(entry_list[i].source.id, requests[i]->net_log().source().id); |
| } |
| } |
| } |
| |
| // Make sure CreateNetLogEntriesForActiveObjects works with multiple |
| // URLRequestContexts. |
| TEST(NetLogUtil, CreateNetLogEntriesForActiveObjectsMultipleContexts) { |
| base::test::TaskEnvironment task_environment; |
| |
| TestDelegate delegate; |
| for (size_t num_requests = 0; num_requests < 5; ++num_requests) { |
| TestNetLog net_log; |
| std::vector<std::unique_ptr<TestURLRequestContext>> contexts; |
| std::vector<std::unique_ptr<URLRequest>> requests; |
| std::set<URLRequestContext*> context_set; |
| for (size_t i = 0; i < num_requests; ++i) { |
| contexts.push_back(std::make_unique<TestURLRequestContext>(true)); |
| contexts[i]->set_net_log(&net_log); |
| contexts[i]->Init(); |
| context_set.insert(contexts[i].get()); |
| requests.push_back( |
| contexts[i]->CreateRequest(GURL("about:hats"), DEFAULT_PRIORITY, |
| &delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
| } |
| RecordingTestNetLog test_net_log; |
| CreateNetLogEntriesForActiveObjects(context_set, |
| test_net_log.GetObserver()); |
| auto entry_list = test_net_log.GetEntries(); |
| ASSERT_EQ(num_requests, entry_list.size()); |
| |
| for (size_t i = 0; i < num_requests; ++i) { |
| EXPECT_EQ(entry_list[i].source.id, requests[i]->net_log().source().id); |
| } |
| } |
| } |
| |
| } // namespace |
| |
| } // namespace net |