blob: d5bd9d4a0053632e3d65694fc8441f668cfded67 [file] [log] [blame]
henrike@webrtc.orgf0488722014-05-13 18:00:261/*
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include "webrtc/base/fileutils.h"
12#include "webrtc/base/gunit.h"
13#include "webrtc/base/logging.h"
14#include "webrtc/base/pathutils.h"
15#include "webrtc/base/stream.h"
16#include "webrtc/base/thread.h"
17
18namespace rtc {
19
Tommi0eefb4d2015-05-23 07:54:0720template <typename Base>
21class LogSinkImpl
22 : public LogSink,
23 public Base {
24 public:
25 LogSinkImpl() {}
26
Tommi0eefb4d2015-05-23 07:54:0727 template<typename P>
Tommi00aac5a2015-05-25 09:25:5928 explicit LogSinkImpl(P* p) : Base(p) {}
Tommi0eefb4d2015-05-23 07:54:0729
30 private:
31 void OnLogMessage(const std::string& message) override {
32 static_cast<Base*>(this)->WriteAll(
33 message.data(), message.size(), nullptr, nullptr);
34 }
35};
36
henrike@webrtc.orgf0488722014-05-13 18:00:2637// Test basic logging operation. We should get the INFO log but not the VERBOSE.
38// We should restore the correct global state at the end.
39TEST(LogTest, SingleStream) {
40 int sev = LogMessage::GetLogToStream(NULL);
41
42 std::string str;
Tommi00aac5a2015-05-25 09:25:5943 LogSinkImpl<StringStream> stream(&str);
henrike@webrtc.orgf0488722014-05-13 18:00:2644 LogMessage::AddLogToStream(&stream, LS_INFO);
45 EXPECT_EQ(LS_INFO, LogMessage::GetLogToStream(&stream));
46
47 LOG(LS_INFO) << "INFO";
48 LOG(LS_VERBOSE) << "VERBOSE";
49 EXPECT_NE(std::string::npos, str.find("INFO"));
50 EXPECT_EQ(std::string::npos, str.find("VERBOSE"));
51
52 LogMessage::RemoveLogToStream(&stream);
Tommi0eefb4d2015-05-23 07:54:0753 EXPECT_EQ(LS_NONE, LogMessage::GetLogToStream(&stream));
henrike@webrtc.orgf0488722014-05-13 18:00:2654
55 EXPECT_EQ(sev, LogMessage::GetLogToStream(NULL));
56}
57
58// Test using multiple log streams. The INFO stream should get the INFO message,
59// the VERBOSE stream should get the INFO and the VERBOSE.
60// We should restore the correct global state at the end.
61TEST(LogTest, MultipleStreams) {
62 int sev = LogMessage::GetLogToStream(NULL);
63
64 std::string str1, str2;
Tommi00aac5a2015-05-25 09:25:5965 LogSinkImpl<StringStream> stream1(&str1), stream2(&str2);
henrike@webrtc.orgf0488722014-05-13 18:00:2666 LogMessage::AddLogToStream(&stream1, LS_INFO);
67 LogMessage::AddLogToStream(&stream2, LS_VERBOSE);
68 EXPECT_EQ(LS_INFO, LogMessage::GetLogToStream(&stream1));
69 EXPECT_EQ(LS_VERBOSE, LogMessage::GetLogToStream(&stream2));
70
71 LOG(LS_INFO) << "INFO";
72 LOG(LS_VERBOSE) << "VERBOSE";
73
74 EXPECT_NE(std::string::npos, str1.find("INFO"));
75 EXPECT_EQ(std::string::npos, str1.find("VERBOSE"));
76 EXPECT_NE(std::string::npos, str2.find("INFO"));
77 EXPECT_NE(std::string::npos, str2.find("VERBOSE"));
78
79 LogMessage::RemoveLogToStream(&stream2);
80 LogMessage::RemoveLogToStream(&stream1);
Tommi0eefb4d2015-05-23 07:54:0781 EXPECT_EQ(LS_NONE, LogMessage::GetLogToStream(&stream2));
82 EXPECT_EQ(LS_NONE, LogMessage::GetLogToStream(&stream1));
henrike@webrtc.orgf0488722014-05-13 18:00:2683
84 EXPECT_EQ(sev, LogMessage::GetLogToStream(NULL));
85}
86
87// Ensure we don't crash when adding/removing streams while threads are going.
88// We should restore the correct global state at the end.
89class LogThread : public Thread {
90 public:
91 virtual ~LogThread() {
92 Stop();
93 }
94
95 private:
96 void Run() {
97 // LS_SENSITIVE to avoid cluttering up any real logging going on
98 LOG(LS_SENSITIVE) << "LOG";
99 }
100};
101
henrike@webrtc.orgc732a3e2014-10-09 22:08:15102TEST(LogTest, MultipleThreads) {
henrike@webrtc.orgf0488722014-05-13 18:00:26103 int sev = LogMessage::GetLogToStream(NULL);
104
105 LogThread thread1, thread2, thread3;
106 thread1.Start();
107 thread2.Start();
108 thread3.Start();
109
Tommi0eefb4d2015-05-23 07:54:07110 LogSinkImpl<NullStream> stream1, stream2, stream3;
henrike@webrtc.orgf0488722014-05-13 18:00:26111 for (int i = 0; i < 1000; ++i) {
112 LogMessage::AddLogToStream(&stream1, LS_INFO);
113 LogMessage::AddLogToStream(&stream2, LS_VERBOSE);
114 LogMessage::AddLogToStream(&stream3, LS_SENSITIVE);
115 LogMessage::RemoveLogToStream(&stream1);
116 LogMessage::RemoveLogToStream(&stream2);
117 LogMessage::RemoveLogToStream(&stream3);
118 }
119
120 EXPECT_EQ(sev, LogMessage::GetLogToStream(NULL));
121}
122
123
124TEST(LogTest, WallClockStartTime) {
Peter Boström0c4e06b2015-10-07 10:23:21125 uint32_t time = LogMessage::WallClockStartTime();
henrike@webrtc.orgf0488722014-05-13 18:00:26126 // Expect the time to be in a sensible range, e.g. > 2012-01-01.
127 EXPECT_GT(time, 1325376000u);
128}
129
130// Test the time required to write 1000 80-character logs to an unbuffered file.
131TEST(LogTest, Perf) {
132 Pathname path;
133 EXPECT_TRUE(Filesystem::GetTemporaryFolder(path, true, NULL));
134 path.SetPathname(Filesystem::TempFilename(path, "ut"));
135
Tommi0eefb4d2015-05-23 07:54:07136 LogSinkImpl<FileStream> stream;
henrike@webrtc.orgf0488722014-05-13 18:00:26137 EXPECT_TRUE(stream.Open(path.pathname(), "wb", NULL));
138 stream.DisableBuffering();
139 LogMessage::AddLogToStream(&stream, LS_SENSITIVE);
140
Honghai Zhang82d78622016-05-06 18:29:15141 int64_t start = TimeMillis(), finish;
henrike@webrtc.orgf0488722014-05-13 18:00:26142 std::string message('X', 80);
143 for (int i = 0; i < 1000; ++i) {
144 LOG(LS_SENSITIVE) << message;
145 }
Honghai Zhang82d78622016-05-06 18:29:15146 finish = TimeMillis();
henrike@webrtc.orgf0488722014-05-13 18:00:26147
148 LogMessage::RemoveLogToStream(&stream);
149 stream.Close();
150 Filesystem::DeleteFile(path);
151
Honghai Zhang82d78622016-05-06 18:29:15152 LOG(LS_INFO) << "Average log time: " << TimeDiff(finish, start) << " ms";
henrike@webrtc.orgf0488722014-05-13 18:00:26153}
154
155} // namespace rtc