| Index: net/log/net_log_unittest.cc
|
| diff --git a/net/log/net_log_unittest.cc b/net/log/net_log_unittest.cc
|
| index 73e4e3027188533a729870d41118cf2396fa8584..9fad152c732c7c52d305e16bdb0b417f73663d7b 100644
|
| --- a/net/log/net_log_unittest.cc
|
| +++ b/net/log/net_log_unittest.cc
|
| @@ -18,9 +18,13 @@ namespace {
|
| const int kThreads = 10;
|
| const int kEvents = 100;
|
|
|
| -base::Value* NetLogLevelCallback(NetLog::LogLevel log_level) {
|
| +base::Value* CaptureModeToValue(NetLogCaptureMode capture_mode) {
|
| + return new base::FundamentalValue(capture_mode.ToInternalValueForTesting());
|
| +}
|
| +
|
| +base::Value* NetCaptureModeCallback(NetLogCaptureMode capture_mode) {
|
| base::DictionaryValue* dict = new base::DictionaryValue();
|
| - dict->SetInteger("log_level", log_level);
|
| + dict->Set("capture_mode", CaptureModeToValue(capture_mode));
|
| return dict;
|
| }
|
|
|
| @@ -42,16 +46,22 @@ TEST(NetLogTest, Basic) {
|
| EXPECT_FALSE(entries[0].params);
|
| }
|
|
|
| -// Check that the correct LogLevel is sent to NetLog Value callbacks.
|
| -TEST(NetLogTest, LogLevels) {
|
| +// Check that the correct CaptureMode is sent to NetLog Value callbacks.
|
| +TEST(NetLogTest, CaptureModes) {
|
| + NetLogCaptureMode kModes[] = {
|
| + NetLogCaptureMode::Default(),
|
| + NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + NetLogCaptureMode::IncludeSocketBytes(),
|
| + };
|
| +
|
| TestNetLog net_log;
|
| - for (int log_level = NetLog::LOG_ALL; log_level < NetLog::LOG_NONE;
|
| - ++log_level) {
|
| - net_log.SetLogLevel(static_cast<NetLog::LogLevel>(log_level));
|
| - EXPECT_EQ(log_level, net_log.GetLogLevel());
|
| +
|
| + for (NetLogCaptureMode mode : kModes) {
|
| + net_log.SetCaptureMode(mode);
|
| + EXPECT_EQ(mode, net_log.GetCaptureMode());
|
|
|
| net_log.AddGlobalEntry(NetLog::TYPE_SOCKET_ALIVE,
|
| - base::Bind(NetLogLevelCallback));
|
| + base::Bind(NetCaptureModeCallback));
|
|
|
| TestNetLog::CapturedEntryList entries;
|
| net_log.GetEntries(&entries);
|
| @@ -63,9 +73,10 @@ TEST(NetLogTest, LogLevels) {
|
| EXPECT_EQ(NetLog::PHASE_NONE, entries[0].phase);
|
| EXPECT_GE(base::TimeTicks::Now(), entries[0].time);
|
|
|
| - int logged_log_level;
|
| - ASSERT_TRUE(entries[0].GetIntegerValue("log_level", &logged_log_level));
|
| - EXPECT_EQ(log_level, logged_log_level);
|
| + int logged_capture_mode;
|
| + ASSERT_TRUE(
|
| + entries[0].GetIntegerValue("capture_mode", &logged_capture_mode));
|
| + EXPECT_EQ(mode.ToInternalValueForTesting(), logged_capture_mode);
|
|
|
| net_log.Clear();
|
| }
|
| @@ -111,12 +122,9 @@ class LoggingObserver : public NetLog::ThreadSafeObserver {
|
| ScopedVector<base::DictionaryValue> values_;
|
| };
|
|
|
| -base::Value* LogLevelToValue(NetLog::LogLevel log_level) {
|
| - return new base::FundamentalValue(log_level);
|
| -}
|
| -
|
| void AddEvent(NetLog* net_log) {
|
| - net_log->AddGlobalEntry(NetLog::TYPE_CANCELLED, base::Bind(LogLevelToValue));
|
| + net_log->AddGlobalEntry(NetLog::TYPE_CANCELLED,
|
| + base::Bind(CaptureModeToValue));
|
| }
|
|
|
| // A thread that waits until an event has been signalled before calling
|
| @@ -180,15 +188,17 @@ class AddRemoveObserverTestThread : public NetLogTestThread {
|
| for (int i = 0; i < kEvents; ++i) {
|
| ASSERT_FALSE(observer_.net_log());
|
|
|
| - net_log_->DeprecatedAddObserver(&observer_, NetLog::LOG_ALL_BUT_BYTES);
|
| + net_log_->DeprecatedAddObserver(
|
| + &observer_, NetLogCaptureMode::IncludeCookiesAndCredentials());
|
| ASSERT_EQ(net_log_, observer_.net_log());
|
| - ASSERT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer_.log_level());
|
| - ASSERT_LE(net_log_->GetLogLevel(), NetLog::LOG_ALL_BUT_BYTES);
|
| + ASSERT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + observer_.capture_mode());
|
|
|
| - net_log_->SetObserverLogLevel(&observer_, NetLog::LOG_ALL);
|
| + net_log_->SetObserverCaptureMode(&observer_,
|
| + NetLogCaptureMode::IncludeSocketBytes());
|
| ASSERT_EQ(net_log_, observer_.net_log());
|
| - ASSERT_EQ(NetLog::LOG_ALL, observer_.log_level());
|
| - ASSERT_LE(net_log_->GetLogLevel(), NetLog::LOG_ALL);
|
| + ASSERT_EQ(NetLogCaptureMode::IncludeSocketBytes(),
|
| + observer_.capture_mode());
|
|
|
| net_log_->DeprecatedRemoveObserver(&observer_);
|
| ASSERT_TRUE(!observer_.net_log());
|
| @@ -225,8 +235,10 @@ TEST(NetLogTest, NetLogEventThreads) {
|
| // Attach some observers. Since they're created after |net_log|, they'll
|
| // safely detach themselves on destruction.
|
| CountingObserver observers[3];
|
| - for (size_t i = 0; i < arraysize(observers); ++i)
|
| - net_log.DeprecatedAddObserver(&observers[i], NetLog::LOG_ALL);
|
| + for (size_t i = 0; i < arraysize(observers); ++i) {
|
| + net_log.DeprecatedAddObserver(&observers[i],
|
| + NetLogCaptureMode::IncludeSocketBytes());
|
| + }
|
|
|
| // Run a bunch of threads to completion, each of which will emit events to
|
| // |net_log|.
|
| @@ -246,22 +258,26 @@ TEST(NetLogTest, NetLogAddRemoveObserver) {
|
| AddEvent(&net_log);
|
| EXPECT_EQ(0, observer.count());
|
| EXPECT_EQ(NULL, observer.net_log());
|
| - EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel());
|
| + EXPECT_FALSE(net_log.GetCaptureMode().enabled());
|
|
|
| // Add the observer and add an event.
|
| - net_log.DeprecatedAddObserver(&observer, NetLog::LOG_ALL_BUT_BYTES);
|
| + net_log.DeprecatedAddObserver(
|
| + &observer, NetLogCaptureMode::IncludeCookiesAndCredentials());
|
| EXPECT_EQ(&net_log, observer.net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer.log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + observer.capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + net_log.GetCaptureMode());
|
|
|
| AddEvent(&net_log);
|
| EXPECT_EQ(1, observer.count());
|
|
|
| // Change the observer's logging level and add an event.
|
| - net_log.SetObserverLogLevel(&observer, NetLog::LOG_ALL);
|
| + net_log.SetObserverCaptureMode(&observer,
|
| + NetLogCaptureMode::IncludeSocketBytes());
|
| EXPECT_EQ(&net_log, observer.net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL, observer.log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(), observer.capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(), net_log.GetCaptureMode());
|
|
|
| AddEvent(&net_log);
|
| EXPECT_EQ(2, observer.count());
|
| @@ -269,16 +285,17 @@ TEST(NetLogTest, NetLogAddRemoveObserver) {
|
| // Remove observer and add an event.
|
| net_log.DeprecatedRemoveObserver(&observer);
|
| EXPECT_EQ(NULL, observer.net_log());
|
| - EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel());
|
| + EXPECT_FALSE(net_log.GetCaptureMode().enabled());
|
|
|
| AddEvent(&net_log);
|
| EXPECT_EQ(2, observer.count());
|
|
|
| // Add the observer a final time, and add an event.
|
| - net_log.DeprecatedAddObserver(&observer, NetLog::LOG_ALL);
|
| + net_log.DeprecatedAddObserver(&observer,
|
| + NetLogCaptureMode::IncludeSocketBytes());
|
| EXPECT_EQ(&net_log, observer.net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL, observer.log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(), observer.capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(), net_log.GetCaptureMode());
|
|
|
| AddEvent(&net_log);
|
| EXPECT_EQ(3, observer.count());
|
| @@ -290,19 +307,25 @@ TEST(NetLogTest, NetLogTwoObservers) {
|
| LoggingObserver observer[2];
|
|
|
| // Add first observer.
|
| - net_log.DeprecatedAddObserver(&observer[0], NetLog::LOG_ALL_BUT_BYTES);
|
| + net_log.DeprecatedAddObserver(
|
| + &observer[0], NetLogCaptureMode::IncludeCookiesAndCredentials());
|
| EXPECT_EQ(&net_log, observer[0].net_log());
|
| EXPECT_EQ(NULL, observer[1].net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + observer[0].capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + net_log.GetCaptureMode());
|
|
|
| // Add second observer observer.
|
| - net_log.DeprecatedAddObserver(&observer[1], NetLog::LOG_ALL);
|
| + net_log.DeprecatedAddObserver(&observer[1],
|
| + NetLogCaptureMode::IncludeSocketBytes());
|
| EXPECT_EQ(&net_log, observer[0].net_log());
|
| EXPECT_EQ(&net_log, observer[1].net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL, observer[1].log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + observer[0].capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(),
|
| + observer[1].capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeSocketBytes(), net_log.GetCaptureMode());
|
|
|
| // Add event and make sure both observers receive it at their respective log
|
| // levels.
|
| @@ -310,17 +333,19 @@ TEST(NetLogTest, NetLogTwoObservers) {
|
| AddEvent(&net_log);
|
| ASSERT_EQ(1U, observer[0].GetNumValues());
|
| ASSERT_TRUE(observer[0].GetValue(0)->GetInteger("params", ¶m));
|
| - EXPECT_EQ(observer[0].log_level(), param);
|
| + EXPECT_EQ(observer[0].capture_mode().ToInternalValueForTesting(), param);
|
| ASSERT_EQ(1U, observer[1].GetNumValues());
|
| ASSERT_TRUE(observer[1].GetValue(0)->GetInteger("params", ¶m));
|
| - EXPECT_EQ(observer[1].log_level(), param);
|
| + EXPECT_EQ(observer[1].capture_mode().ToInternalValueForTesting(), param);
|
|
|
| // Remove second observer.
|
| net_log.DeprecatedRemoveObserver(&observer[1]);
|
| EXPECT_EQ(&net_log, observer[0].net_log());
|
| EXPECT_EQ(NULL, observer[1].net_log());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, observer[0].log_level());
|
| - EXPECT_EQ(NetLog::LOG_ALL_BUT_BYTES, net_log.GetLogLevel());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + observer[0].capture_mode());
|
| + EXPECT_EQ(NetLogCaptureMode::IncludeCookiesAndCredentials(),
|
| + net_log.GetCaptureMode());
|
|
|
| // Add event and make sure only second observer gets it.
|
| AddEvent(&net_log);
|
| @@ -331,7 +356,7 @@ TEST(NetLogTest, NetLogTwoObservers) {
|
| net_log.DeprecatedRemoveObserver(&observer[0]);
|
| EXPECT_EQ(NULL, observer[0].net_log());
|
| EXPECT_EQ(NULL, observer[1].net_log());
|
| - EXPECT_EQ(NetLog::LOG_NONE, net_log.GetLogLevel());
|
| + EXPECT_FALSE(net_log.GetCaptureMode().enabled());
|
|
|
| // Add event and make sure neither observer gets it.
|
| AddEvent(&net_log);
|
|
|