| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_event
_store.h" | 5 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_event
_store.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
| 11 #include "base/test/test_simple_task_runner.h" | 11 #include "base/test/test_simple_task_runner.h" |
| 12 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_param
s_test_utils.h" | 12 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_param
s_test_utils.h" |
| 13 #include "net/log/capturing_net_log.h" | |
| 14 #include "net/log/net_log.h" | 13 #include "net/log/net_log.h" |
| 14 #include "net/log/test_net_log.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 16 |
| 17 namespace data_reduction_proxy { | 17 namespace data_reduction_proxy { |
| 18 | 18 |
| 19 class DataReductionProxyEventStoreTest : public testing::Test { | 19 class DataReductionProxyEventStoreTest : public testing::Test { |
| 20 public: | 20 public: |
| 21 DataReductionProxyEventStoreTest() | 21 DataReductionProxyEventStoreTest() |
| 22 : task_runner_(scoped_refptr<base::TestSimpleTaskRunner>( | 22 : task_runner_(scoped_refptr<base::TestSimpleTaskRunner>( |
| 23 new base::TestSimpleTaskRunner())), | 23 new base::TestSimpleTaskRunner())), |
| 24 net_log_(new net::CapturingNetLog()) { | 24 net_log_(new net::TestNetLog()) { |
| 25 bound_net_log_ = net::BoundNetLog::Make( | 25 bound_net_log_ = net::BoundNetLog::Make( |
| 26 net_log_.get(), net::NetLog::SOURCE_DATA_REDUCTION_PROXY); | 26 net_log_.get(), net::NetLog::SOURCE_DATA_REDUCTION_PROXY); |
| 27 } | 27 } |
| 28 | 28 |
| 29 void SetUp() override { | 29 void SetUp() override { |
| 30 proxy_.reset(new DataReductionProxyEventStore(task_runner_)); | 30 proxy_.reset(new DataReductionProxyEventStore(task_runner_)); |
| 31 } | 31 } |
| 32 | 32 |
| 33 net::CapturingNetLog::CapturedEntry GetSingleEntry() const { | 33 net::TestNetLog::CapturedEntry GetSingleEntry() const { |
| 34 net::CapturingNetLog::CapturedEntryList entries; | 34 net::TestNetLog::CapturedEntryList entries; |
| 35 net_log_->GetEntries(&entries); | 35 net_log_->GetEntries(&entries); |
| 36 EXPECT_EQ(1u, entries.size()); | 36 EXPECT_EQ(1u, entries.size()); |
| 37 return entries[0]; | 37 return entries[0]; |
| 38 } | 38 } |
| 39 | 39 |
| 40 DataReductionProxyEventStore* proxy() { | 40 DataReductionProxyEventStore* proxy() { |
| 41 return proxy_.get(); | 41 return proxy_.get(); |
| 42 } | 42 } |
| 43 | 43 |
| 44 base::TestSimpleTaskRunner* task_runner() { | 44 base::TestSimpleTaskRunner* task_runner() { |
| 45 return task_runner_.get(); | 45 return task_runner_.get(); |
| 46 } | 46 } |
| 47 | 47 |
| 48 net::CapturingNetLog* net_log() { | 48 net::TestNetLog* net_log() { return net_log_.get(); } |
| 49 return net_log_.get(); | |
| 50 } | |
| 51 | 49 |
| 52 const net::BoundNetLog& bound_net_log() { | 50 const net::BoundNetLog& bound_net_log() { |
| 53 return bound_net_log_; | 51 return bound_net_log_; |
| 54 } | 52 } |
| 55 | 53 |
| 56 private: | 54 private: |
| 57 scoped_refptr<base::TestSimpleTaskRunner> task_runner_; | 55 scoped_refptr<base::TestSimpleTaskRunner> task_runner_; |
| 58 scoped_ptr<net::CapturingNetLog> net_log_; | 56 scoped_ptr<net::TestNetLog> net_log_; |
| 59 scoped_ptr<DataReductionProxyEventStore> proxy_; | 57 scoped_ptr<DataReductionProxyEventStore> proxy_; |
| 60 net::BoundNetLog bound_net_log_; | 58 net::BoundNetLog bound_net_log_; |
| 61 }; | 59 }; |
| 62 | 60 |
| 63 TEST_F(DataReductionProxyEventStoreTest, TestAddProxyEnabledEvent) { | 61 TEST_F(DataReductionProxyEventStoreTest, TestAddProxyEnabledEvent) { |
| 64 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 62 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 65 proxy()->AddProxyEnabledEvent( | 63 proxy()->AddProxyEnabledEvent( |
| 66 net_log(), false, false, | 64 net_log(), false, false, |
| 67 TestDataReductionProxyParams::DefaultOrigin(), | 65 TestDataReductionProxyParams::DefaultOrigin(), |
| 68 TestDataReductionProxyParams::DefaultFallbackOrigin(), | 66 TestDataReductionProxyParams::DefaultFallbackOrigin(), |
| 69 TestDataReductionProxyParams::DefaultSSLOrigin()); | 67 TestDataReductionProxyParams::DefaultSSLOrigin()); |
| 70 task_runner()->RunPendingTasks(); | 68 task_runner()->RunPendingTasks(); |
| 71 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 69 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 72 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 70 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 73 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_ENABLED, | 71 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_ENABLED, |
| 74 entry.type); | 72 entry.type); |
| 75 } | 73 } |
| 76 | 74 |
| 77 TEST_F(DataReductionProxyEventStoreTest, TestAddProxyDisabledEvent) { | 75 TEST_F(DataReductionProxyEventStoreTest, TestAddProxyDisabledEvent) { |
| 78 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 76 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 79 proxy()->AddProxyDisabledEvent(net_log()); | 77 proxy()->AddProxyDisabledEvent(net_log()); |
| 80 task_runner()->RunPendingTasks(); | 78 task_runner()->RunPendingTasks(); |
| 81 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 79 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 82 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 80 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 83 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_ENABLED, | 81 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_ENABLED, |
| 84 entry.type); | 82 entry.type); |
| 85 } | 83 } |
| 86 | 84 |
| 87 TEST_F(DataReductionProxyEventStoreTest, TestAddBypassActionEvent) { | 85 TEST_F(DataReductionProxyEventStoreTest, TestAddBypassActionEvent) { |
| 88 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 86 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 89 EXPECT_EQ(nullptr, proxy()->last_bypass_event_.get()); | 87 EXPECT_EQ(nullptr, proxy()->last_bypass_event_.get()); |
| 90 proxy()->AddBypassActionEvent(bound_net_log(), "bypass", GURL(), | 88 proxy()->AddBypassActionEvent(bound_net_log(), "bypass", GURL(), |
| 91 base::TimeDelta::FromMinutes(1)); | 89 base::TimeDelta::FromMinutes(1)); |
| 92 task_runner()->RunPendingTasks(); | 90 task_runner()->RunPendingTasks(); |
| 93 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 91 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 94 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 92 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 95 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_BYPASS_REQUESTED, | 93 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_BYPASS_REQUESTED, |
| 96 entry.type); | 94 entry.type); |
| 97 EXPECT_NE(nullptr, proxy()->last_bypass_event_.get()); | 95 EXPECT_NE(nullptr, proxy()->last_bypass_event_.get()); |
| 98 } | 96 } |
| 99 | 97 |
| 100 TEST_F(DataReductionProxyEventStoreTest, TestAddBypassTypeEvent) { | 98 TEST_F(DataReductionProxyEventStoreTest, TestAddBypassTypeEvent) { |
| 101 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 99 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 102 EXPECT_EQ(nullptr, proxy()->last_bypass_event_.get()); | 100 EXPECT_EQ(nullptr, proxy()->last_bypass_event_.get()); |
| 103 proxy()->AddBypassTypeEvent(bound_net_log(), BYPASS_EVENT_TYPE_LONG, GURL(), | 101 proxy()->AddBypassTypeEvent(bound_net_log(), BYPASS_EVENT_TYPE_LONG, GURL(), |
| 104 base::TimeDelta::FromMinutes(1)); | 102 base::TimeDelta::FromMinutes(1)); |
| 105 task_runner()->RunPendingTasks(); | 103 task_runner()->RunPendingTasks(); |
| 106 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 104 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 107 EXPECT_EQ(1u, net_log()->GetSize()); | 105 EXPECT_EQ(1u, net_log()->GetSize()); |
| 108 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 106 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 109 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_BYPASS_REQUESTED, | 107 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_BYPASS_REQUESTED, |
| 110 entry.type); | 108 entry.type); |
| 111 EXPECT_NE(nullptr, proxy()->last_bypass_event_.get()); | 109 EXPECT_NE(nullptr, proxy()->last_bypass_event_.get()); |
| 112 } | 110 } |
| 113 | 111 |
| 114 TEST_F(DataReductionProxyEventStoreTest, TestBeginSecureProxyCheck) { | 112 TEST_F(DataReductionProxyEventStoreTest, TestBeginSecureProxyCheck) { |
| 115 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 113 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 116 EXPECT_EQ(CHECK_UNKNOWN, proxy()->secure_proxy_check_state_); | 114 EXPECT_EQ(CHECK_UNKNOWN, proxy()->secure_proxy_check_state_); |
| 117 proxy()->BeginSecureProxyCheck(bound_net_log(), GURL()); | 115 proxy()->BeginSecureProxyCheck(bound_net_log(), GURL()); |
| 118 task_runner()->RunPendingTasks(); | 116 task_runner()->RunPendingTasks(); |
| 119 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 117 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 120 EXPECT_EQ(1u, net_log()->GetSize()); | 118 EXPECT_EQ(1u, net_log()->GetSize()); |
| 121 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 119 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 122 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_CANARY_REQUEST, | 120 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_CANARY_REQUEST, |
| 123 entry.type); | 121 entry.type); |
| 124 EXPECT_EQ(CHECK_PENDING, proxy()->secure_proxy_check_state_); | 122 EXPECT_EQ(CHECK_PENDING, proxy()->secure_proxy_check_state_); |
| 125 } | 123 } |
| 126 | 124 |
| 127 TEST_F(DataReductionProxyEventStoreTest, TestEndSecureProxyCheck) { | 125 TEST_F(DataReductionProxyEventStoreTest, TestEndSecureProxyCheck) { |
| 128 EXPECT_EQ(0u, proxy()->stored_events_.size()); | 126 EXPECT_EQ(0u, proxy()->stored_events_.size()); |
| 129 EXPECT_EQ(CHECK_UNKNOWN, proxy()->secure_proxy_check_state_); | 127 EXPECT_EQ(CHECK_UNKNOWN, proxy()->secure_proxy_check_state_); |
| 130 proxy()->EndSecureProxyCheck(bound_net_log(), 0); | 128 proxy()->EndSecureProxyCheck(bound_net_log(), 0); |
| 131 task_runner()->RunPendingTasks(); | 129 task_runner()->RunPendingTasks(); |
| 132 EXPECT_EQ(1u, proxy()->stored_events_.size()); | 130 EXPECT_EQ(1u, proxy()->stored_events_.size()); |
| 133 EXPECT_EQ(1u, net_log()->GetSize()); | 131 EXPECT_EQ(1u, net_log()->GetSize()); |
| 134 net::CapturingNetLog::CapturedEntry entry = GetSingleEntry(); | 132 net::TestNetLog::CapturedEntry entry = GetSingleEntry(); |
| 135 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_CANARY_REQUEST, | 133 EXPECT_EQ(net::NetLog::TYPE_DATA_REDUCTION_PROXY_CANARY_REQUEST, |
| 136 entry.type); | 134 entry.type); |
| 137 EXPECT_EQ(CHECK_SUCCESS, proxy()->secure_proxy_check_state_); | 135 EXPECT_EQ(CHECK_SUCCESS, proxy()->secure_proxy_check_state_); |
| 138 } | 136 } |
| 139 | 137 |
| 140 } // namespace data_reduction_proxy | 138 } // namespace data_reduction_proxy |
| OLD | NEW |