| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/base/net_log_unittest.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/memory/scoped_vector.h" | |
| 9 #include "base/synchronization/waitable_event.h" | |
| 10 #include "base/threading/simple_thread.h" | |
| 11 #include "base/values.h" | |
| 12 #include "net/base/net_errors.h" | |
| 13 | |
| 14 namespace net { | |
| 15 | |
| 16 namespace { | |
| 17 | |
| 18 const int kThreads = 10; | |
| 19 const int kEvents = 100; | |
| 20 | |
| 21 base::Value* NetLogLevelCallback(NetLog::LogLevel log_level) { | |
| 22 base::DictionaryValue* dict = new base::DictionaryValue(); | |
| 23 dict->SetInteger("log_level", log_level); | |
| 24 return dict; | |
| 25 } | |
| 26 | |
| 27 TEST(NetLogTest, Basic) { | |
| 28 CapturingNetLog net_log; | |
| 29 CapturingNetLog::CapturedEntryList entries; | |
| 30 net_log.GetEntries(&entries); | |
| 31 EXPECT_EQ(0u, entries.size()); | |
| 32 | |
| 33 net_log.AddGlobalEntry(NetLog::TYPE_CANCELLED); | |
| 34 | |
| 35 net_log.GetEntries(&entries); | |
| 36 ASSERT_EQ(1u, entries.size()); | |
| 37 EXPECT_EQ(NetLog::TYPE_CANCELLED, entries[0].type); | |
| 38 EXPECT_EQ(NetLog::SOURCE_NONE, entries[0].source.type); | |
| 39 EXPECT_NE(NetLog::Source::kInvalidId, entries[0].source.id); | |
| 40 EXPECT_EQ(NetLog::PHASE_NONE, entries[0].phase); | |
| 41 EXPECT_GE(base::TimeTicks::Now(), entries[0].time); | |
| 42 EXPECT_FALSE(entries[0].params); | |
| 43 } | |
| 44 | |
| 45 // Check that the correct LogLevel is sent to NetLog Value callbacks. | |
| 46 TEST(NetLogTest, LogLevels) { | |
| 47 CapturingNetLog net_log; | |
| 48 for (int log_level = NetLog::LOG_ALL; log_level < NetLog::LOG_NONE; | |
| 49 ++log_level) { | |
| 50 net_log.SetLogLevel(static_cast<NetLog::LogLevel>(log_level)); | |
| 51 EXPECT_EQ(log_level, net_log.GetLogLevel()); | |
| 52 | |
| 53 net_log.AddGlobalEntry(NetLog::TYPE_SOCKET_ALIVE, | |
| 54 base::Bind(NetLogLevelCallback)); | |
| 55 | |
| 56 CapturingNetLog::CapturedEntryList entries; | |
| 57 net_log.GetEntries(&entries); | |
| 58 | |
| 59 ASSERT_EQ(1u, entries.size()); | |
| 60 EXPECT_EQ(NetLog::TYPE_SOCKET_ALIVE, entries[0].type); | |
| 61 EXPECT_EQ(NetLog::SOURCE_NONE, entries[0].source.type); | |
| 62 EXPECT_NE(NetLog::Source::kInvalidId, entries[0].source.id); | |
| 63 EXPECT_EQ(NetLog::PHASE_NONE, entries[0].phase); | |
| 64 EXPECT_GE(base::TimeTicks::Now(), entries[0].time); | |
| 65 | |
| 66 int logged_log_level; | |
| 67 ASSERT_TRUE(entries[0].GetIntegerValue("log_level", &logged_log_level)); | |
| 68 EXPECT_EQ(log_level, logged_log_level); | |
| 69 | |
| 70 net_log.Clear(); | |
| 71 } | |
| 72 } | |
| 73 | |
| 74 class CountingObserver : public NetLog::ThreadSafeObserver { | |
| 75 public: | |
| 76 CountingObserver() : count_(0) {} | |
| 77 | |
| 78 ~CountingObserver() override { | |
| 79 if (net_log()) | |
| 80 net_log()->RemoveThreadSafeObserver(this); | |
| 81 } | |
| 82 | |
| 83 void OnAddEntry(const NetLog::Entry& entry) override { ++count_; } | |
| 84 | |
| 85 int count() const { return count_; } | |
| 86 | |
| 87 private: | |
| 88 int count_; | |
| 89 }; | |
| 90 | |
| 91 class LoggingObserver : public NetLog::ThreadSafeObserver { | |
| 92 public: | |
| 93 LoggingObserver() {} | |
| 94 | |
| 95 ~LoggingObserver() override { | |
| 96 if (net_log()) | |
| 97 net_log()->RemoveThreadSafeObserver(this); | |
| 98 } | |
| 99 | |
| 100 void OnAddEntry(const NetLog::Entry& entry) override { | |
| 101 base::Value* value = entry.ToValue(); | |
| 102 base::DictionaryValue* dict = NULL; | |
| 103 ASSERT_TRUE(value->GetAsDictionary(&dict)); | |
| 104 values_.push_back(dict); | |
| 105 } | |
| 106 | |
| 107 size_t GetNumValues() const { return values_.size(); } | |
| 108 base::DictionaryValue* GetValue(size_t index) const { return values_[index]; } | |
| 109 | |
| 110 private: | |
| 111 ScopedVector<base::DictionaryValue> values_; | |
| 112 }; | |
| 113 | |
| 114 base::Value* LogLevelToValue(NetLog::LogLevel log_level) { | |
| 115 return new base::FundamentalValue(log_level); | |
| 116 } | |
| 117 | |
| 118 void AddEvent(NetLog* net_log) { | |
| 119 net_log->AddGlobalEntry(NetLog::TYPE_CANCELLED, base::Bind(LogLevelToValue)); | |
| 120 } | |
| 121 | |
| 122 // A thread that waits until an event has been signalled before calling | |
| 123 // RunTestThread. | |
| 124 class NetLogTestThread : public base::SimpleThread { | |
| 125 public: | |
| 126 NetLogTestThread() | |
| 127 : base::SimpleThread("NetLogTest"), | |
| 128 net_log_(NULL), | |
| 129 start_event_(NULL) { | |
| 130 } | |
| 131 | |
| 132 // We'll wait for |start_event| to be triggered before calling a subclass's | |
| 133 // subclass's RunTestThread() function. | |
| 134 void Init(NetLog* net_log, base::WaitableEvent* start_event) { | |
| 135 start_event_ = start_event; | |
| 136 net_log_ = net_log; | |
| 137 } | |
| 138 | |
| 139 void Run() override { | |
| 140 start_event_->Wait(); | |
| 141 RunTestThread(); | |
| 142 } | |
| 143 | |
| 144 // Subclasses must override this with the code they want to run on their | |
| 145 // thread. | |
| 146 virtual void RunTestThread() = 0; | |
| 147 | |
| 148 protected: | |
| 149 NetLog* net_log_; | |
| 150 | |
| 151 private: | |
| 152 // Only triggered once all threads have been created, to make it less likely | |
| 153 // each thread completes before the next one starts. | |
| 154 base::WaitableEvent* start_event_; | |
| 155 | |
| 156 DISALLOW_COPY_AND_ASSIGN(NetLogTestThread); | |
| 157 }; | |
| 158 | |
| 159 // A thread that adds a bunch of events to the NetLog. | |
| 160 class AddEventsTestThread : public NetLogTestThread { | |
| 161 public: | |
| 162 AddEventsTestThread() {} | |
| 163 ~AddEventsTestThread() override {} | |
| 164 | |
| 165 private: | |
| 166 void RunTestThread() override { | |
| 167 for (int i = 0; i < kEvents; ++i) | |
| 168 AddEvent(net_log_); | |
| 169 } | |
| 170 | |
| 171 DISALLOW_COPY_AND_ASSIGN(AddEventsTestThread); | |
| 172 }; | |
| 173 | |
| 174 // A thread that adds and removes an observer from the NetLog repeatedly. | |
| 175 class AddRemoveObserverTestThread : public NetLogTestThread { | |
| 176 public: | |
| 177 AddRemoveObserverTestThread() {} | |
| 178 | |
| 179 ~AddRemoveObserverTestThread() override { EXPECT_TRUE(!observer_.net_log()); } | |
| 180 | |
| 181 private: | |
| 182 void RunTestThread() override { | |
| 183 for (int i = 0; i < kEvents; ++i) { | |
| 184 ASSERT_FALSE(observer_.net_log()); | |
| 185 | |
| 186 net_log_->AddThreadSafeObserver(&observer_, NetLog::LOG_ALL_BUT_BYTES); | |
| 187 ASSERT_EQ(net_log_, observer_.net_log()); | |
| 188 ASSERT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer_.log_level()); | |
| 189 ASSERT_LE(net_log_->GetLogLevel(), NetLog::LOG_ALL_BUT_BYTES); | |
| 190 | |
| 191 net_log_->SetObserverLogLevel(&observer_, NetLog::LOG_ALL); | |
| 192 ASSERT_EQ(net_log_, observer_.net_log()); | |
| 193 ASSERT_EQ(NetLog::LOG_ALL, observer_.log_level()); | |
| 194 ASSERT_LE(net_log_->GetLogLevel(), NetLog::LOG_ALL); | |
| 195 | |
| 196 net_log_->RemoveThreadSafeObserver(&observer_); | |
| 197 ASSERT_TRUE(!observer_.net_log()); | |
| 198 } | |
| 199 } | |
| 200 | |
| 201 CountingObserver observer_; | |
| 202 | |
| 203 DISALLOW_COPY_AND_ASSIGN(AddRemoveObserverTestThread); | |
| 204 }; | |
| 205 | |
| 206 // Creates |kThreads| threads of type |ThreadType| and then runs them all | |
| 207 // to completion. | |
| 208 template<class ThreadType> | |
| 209 void RunTestThreads(NetLog* net_log) { | |
| 210 ThreadType threads[kThreads]; | |
| 211 base::WaitableEvent start_event(true, false); | |
| 212 | |
| 213 for (size_t i = 0; i < arraysize(threads); ++i) { | |
| 214 threads[i].Init(net_log, &start_event); | |
| 215 threads[i].Start(); | |
| 216 } | |
| 217 | |
| 218 start_event.Signal(); | |
| 219 | |
| 220 for (size_t i = 0; i < arraysize(threads); ++i) | |
| 221 threads[i].Join(); | |
| 222 } | |
| 223 | |
| 224 // Makes sure that events on multiple threads are dispatched to all observers. | |
| 225 TEST(NetLogTest, NetLogEventThreads) { | |
| 226 NetLog net_log; | |
| 227 | |
| 228 // Attach some observers. Since they're created after |net_log|, they'll | |
| 229 // safely detach themselves on destruction. | |
| 230 CountingObserver observers[3]; | |
| 231 for (size_t i = 0; i < arraysize(observers); ++i) | |
| 232 net_log.AddThreadSafeObserver(&observers[i], NetLog::LOG_ALL); | |
| 233 | |
| 234 // Run a bunch of threads to completion, each of which will emit events to | |
| 235 // |net_log|. | |
| 236 RunTestThreads<AddEventsTestThread>(&net_log); | |
| 237 | |
| 238 // Check that each observer saw the emitted events. | |
| 239 const int kTotalEvents = kThreads * kEvents; | |
| 240 for (size_t i = 0; i < arraysize(observers); ++i) | |
| 241 EXPECT_EQ(kTotalEvents, observers[i].count()); | |
| 242 } | |
| 243 | |
| 244 // Test adding and removing a single observer. | |
| 245 TEST(NetLogTest, NetLogAddRemoveObserver) { | |
| 246 NetLog net_log; | |
| 247 CountingObserver observer; | |
| 248 | |
| 249 AddEvent(&net_log); | |
| 250 EXPECT_EQ(0, observer.count()); | |
| 251 EXPECT_EQ(NULL, observer.net_log()); | |
| 252 EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel()); | |
| 253 | |
| 254 // Add the observer and add an event. | |
| 255 net_log.AddThreadSafeObserver(&observer, NetLog::LOG_ALL_BUT_BYTES); | |
| 256 EXPECT_EQ(&net_log, observer.net_log()); | |
| 257 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer.log_level()); | |
| 258 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel()); | |
| 259 | |
| 260 AddEvent(&net_log); | |
| 261 EXPECT_EQ(1, observer.count()); | |
| 262 | |
| 263 // Change the observer's logging level and add an event. | |
| 264 net_log.SetObserverLogLevel(&observer, NetLog::LOG_ALL); | |
| 265 EXPECT_EQ(&net_log, observer.net_log()); | |
| 266 EXPECT_EQ(NetLog::LOG_ALL, observer.log_level()); | |
| 267 EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel()); | |
| 268 | |
| 269 AddEvent(&net_log); | |
| 270 EXPECT_EQ(2, observer.count()); | |
| 271 | |
| 272 // Remove observer and add an event. | |
| 273 net_log.RemoveThreadSafeObserver(&observer); | |
| 274 EXPECT_EQ(NULL, observer.net_log()); | |
| 275 EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel()); | |
| 276 | |
| 277 AddEvent(&net_log); | |
| 278 EXPECT_EQ(2, observer.count()); | |
| 279 | |
| 280 // Add the observer a final time, and add an event. | |
| 281 net_log.AddThreadSafeObserver(&observer, NetLog::LOG_ALL); | |
| 282 EXPECT_EQ(&net_log, observer.net_log()); | |
| 283 EXPECT_EQ(NetLog::LOG_ALL, observer.log_level()); | |
| 284 EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel()); | |
| 285 | |
| 286 AddEvent(&net_log); | |
| 287 EXPECT_EQ(3, observer.count()); | |
| 288 } | |
| 289 | |
| 290 // Test adding and removing two observers at different log levels. | |
| 291 TEST(NetLogTest, NetLogTwoObservers) { | |
| 292 NetLog net_log; | |
| 293 LoggingObserver observer[2]; | |
| 294 | |
| 295 // Add first observer. | |
| 296 net_log.AddThreadSafeObserver(&observer[0], NetLog::LOG_ALL_BUT_BYTES); | |
| 297 EXPECT_EQ(&net_log, observer[0].net_log()); | |
| 298 EXPECT_EQ(NULL, observer[1].net_log()); | |
| 299 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level()); | |
| 300 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel()); | |
| 301 | |
| 302 // Add second observer observer. | |
| 303 net_log.AddThreadSafeObserver(&observer[1], NetLog::LOG_ALL); | |
| 304 EXPECT_EQ(&net_log, observer[0].net_log()); | |
| 305 EXPECT_EQ(&net_log, observer[1].net_log()); | |
| 306 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level()); | |
| 307 EXPECT_EQ(NetLog::LOG_ALL, observer[1].log_level()); | |
| 308 EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel()); | |
| 309 | |
| 310 // Add event and make sure both observers receive it at their respective log | |
| 311 // levels. | |
| 312 int param; | |
| 313 AddEvent(&net_log); | |
| 314 ASSERT_EQ(1U, observer[0].GetNumValues()); | |
| 315 ASSERT_TRUE(observer[0].GetValue(0)->GetInteger("params", ¶m)); | |
| 316 EXPECT_EQ(observer[0].log_level(), param); | |
| 317 ASSERT_EQ(1U, observer[1].GetNumValues()); | |
| 318 ASSERT_TRUE(observer[1].GetValue(0)->GetInteger("params", ¶m)); | |
| 319 EXPECT_EQ(observer[1].log_level(), param); | |
| 320 | |
| 321 // Remove second observer. | |
| 322 net_log.RemoveThreadSafeObserver(&observer[1]); | |
| 323 EXPECT_EQ(&net_log, observer[0].net_log()); | |
| 324 EXPECT_EQ(NULL, observer[1].net_log()); | |
| 325 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level()); | |
| 326 EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel()); | |
| 327 | |
| 328 // Add event and make sure only second observer gets it. | |
| 329 AddEvent(&net_log); | |
| 330 EXPECT_EQ(2U, observer[0].GetNumValues()); | |
| 331 EXPECT_EQ(1U, observer[1].GetNumValues()); | |
| 332 | |
| 333 // Remove first observer. | |
| 334 net_log.RemoveThreadSafeObserver(&observer[0]); | |
| 335 EXPECT_EQ(NULL, observer[0].net_log()); | |
| 336 EXPECT_EQ(NULL, observer[1].net_log()); | |
| 337 EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel()); | |
| 338 | |
| 339 // Add event and make sure neither observer gets it. | |
| 340 AddEvent(&net_log); | |
| 341 EXPECT_EQ(2U, observer[0].GetNumValues()); | |
| 342 EXPECT_EQ(1U, observer[1].GetNumValues()); | |
| 343 } | |
| 344 | |
| 345 // Makes sure that adding and removing observers simultaneously on different | |
| 346 // threads works. | |
| 347 TEST(NetLogTest, NetLogAddRemoveObserverThreads) { | |
| 348 NetLog net_log; | |
| 349 | |
| 350 // Run a bunch of threads to completion, each of which will repeatedly add | |
| 351 // and remove an observer, and set its logging level. | |
| 352 RunTestThreads<AddRemoveObserverTestThread>(&net_log); | |
| 353 } | |
| 354 | |
| 355 } // namespace | |
| 356 | |
| 357 } // namespace net | |
| OLD | NEW |