1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
// (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
#include <gtest/gtest.h>
#include <memory>
// TODO(T90238193)
// @lint-ignore-every CLANGTIDY facebook-hte-RelativeInclude
#include "include/libkineto.h"
#include "src/Logger.h"
#include "LoggerCollector.h"
using namespace KINETO_NAMESPACE;
#if !USE_GOOGLE_LOG
constexpr char InfoTestStr[] = "Checking LOG(INFO)";
constexpr char WarningTestStr[] = "Checking LOG(WARNING)";
constexpr char ErrorTestStr[] = "Checking LOG(ERROR)";
TEST(LoggerObserverTest, SingleCollectorObserver) {
// Add a LoggerObserverCollector to collect all logs during the trace.
std::unique_ptr<LoggerCollector> lCollector = std::make_unique<LoggerCollector>();
Logger::addLoggerObserver(lCollector.get());
LOG(INFO) << InfoTestStr;
LOG(WARNING) << WarningTestStr;
LOG(ERROR) << ErrorTestStr;
auto LoggerMD = lCollector->extractCollectorMetadata();
EXPECT_TRUE(LoggerMD[LoggerOutputType::INFO][0].find(InfoTestStr) != std::string::npos);
EXPECT_TRUE(LoggerMD[LoggerOutputType::WARNING][0].find(WarningTestStr) != std::string::npos);
EXPECT_TRUE(LoggerMD[LoggerOutputType::ERROR][0].find(ErrorTestStr) != std::string::npos);
Logger::removeLoggerObserver(lCollector.get());
}
#define NUM_OF_MESSAGES_FOR_EACH_TYPE 10
#define NUM_OF_WRITE_THREADS 200
// Writes NUM_OF_MESSAGES_FOR_EACH_TYPE messages for each INFO, WARNING, and ERROR.
// NOLINTNEXTLINE(clang-diagnostic-unused-parameter)
void* writeSeveralMessages(void* ptr) {
for(int i=0; i<NUM_OF_MESSAGES_FOR_EACH_TYPE; i++) {
LOG(INFO) << InfoTestStr;
LOG(WARNING) << WarningTestStr;
LOG(ERROR) << ErrorTestStr;
}
return nullptr;
}
TEST(LoggerObserverTest, FourCollectorObserver) {
// There shouldn't be too many CUPTIActivityProfilers active at the same time.
std::unique_ptr<LoggerCollector> lc1 = std::make_unique<LoggerCollector>();
std::unique_ptr<LoggerCollector> lc2 = std::make_unique<LoggerCollector>();
std::unique_ptr<LoggerCollector> lc3 = std::make_unique<LoggerCollector>();
std::unique_ptr<LoggerCollector> lc4 = std::make_unique<LoggerCollector>();
Logger::addLoggerObserver(lc1.get());
Logger::addLoggerObserver(lc2.get());
Logger::addLoggerObserver(lc3.get());
Logger::addLoggerObserver(lc4.get());
// Launch NUM_OF_WRITE_THREADS threads writing several messages.
pthread_t ListOfThreads[NUM_OF_WRITE_THREADS];
for (int i=0; i<NUM_OF_WRITE_THREADS; i++) {
::pthread_create(&ListOfThreads[i], nullptr, writeSeveralMessages, nullptr);
}
// Wait for all threads to finish.
for (int i=0; i<NUM_OF_WRITE_THREADS; i++) {
::pthread_join(ListOfThreads[i], nullptr);
}
auto lc1MD = lc1->extractCollectorMetadata();
int InfoCount = 0, WarnCount = 0, ErrorCount = 0;
for (auto& md : lc1MD) {
InfoCount += md.first == LoggerOutputType::INFO ? md.second.size() : 0;
WarnCount += md.first == LoggerOutputType::WARNING ? md.second.size() : 0;
ErrorCount += md.first == LoggerOutputType::ERROR ? md.second.size() : 0;
}
EXPECT_EQ(InfoCount, NUM_OF_WRITE_THREADS * NUM_OF_MESSAGES_FOR_EACH_TYPE);
EXPECT_EQ(WarnCount, NUM_OF_WRITE_THREADS * NUM_OF_MESSAGES_FOR_EACH_TYPE);
EXPECT_EQ(ErrorCount, NUM_OF_WRITE_THREADS * NUM_OF_MESSAGES_FOR_EACH_TYPE);
Logger::removeLoggerObserver(lc1.get());
Logger::removeLoggerObserver(lc2.get());
Logger::removeLoggerObserver(lc3.get());
Logger::removeLoggerObserver(lc4.get());
}
#endif // !USE_GOOGLE_LOG
int main(int argc, char **argv) {
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
|
