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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.logserver.handlers.logmetrics.test;
import java.util.Map;
import com.yahoo.log.InvalidLogFormatException;
import com.yahoo.log.LogMessage;
import com.yahoo.logserver.handlers.logmetrics.LogMetricsHandler;
import com.yahoo.logserver.handlers.logmetrics.LogMetricsPlugin;
import com.yahoo.plugin.SystemPropertyConfig;
import org.junit.*;
import static org.junit.Assert.*;
/**
* @author hmusum
*/
public class LogMetricsTestCase {
// Some of the tests depend upon the number of messages for a
// host, log level etc. to succeed, so you may have update the
// tests if you change something in mStrings. config, debug and
// spam are filtered out and not handled.
private static final String[] mStrings = {
"1095159244.095\thostA\t1/2\tservice\tcomponent\tconfig\tpayload1",
"1095206399.000\thostA\t1/2\tservice\tcomponent\tinfo\tpayload2",
"1095206400.000\thostA\t1/2\tservice\tcomponent\tinfo\tpayload3",
"1095206401.000\thostA\t1/2\tservice\tcomponent\tinfo\tpayload4",
"1095206402.000\thostA\t1/2\tservice\tcomponent\twarning\tpayload5",
"1095206403.000\thostA\t1/2\tservice\tcomponent\terror\tpayload6",
"1095206404.000\thostB\t1/2\tservice\tcomponent\tinfo\tpayload7",
"1095206405.000\thostB\t1/2\tservice\tcomponent\tfatal\tpayload8",
"1095206406.000\thostB\t1/2\tservice\tcomponent\tdebug\tpayload9",
};
private static final LogMessage[] msg = new LogMessage[mStrings.length];
static {
try {
for (int i = 0; i < mStrings.length; i++) {
msg[i] = LogMessage.parseNativeFormat(mStrings[i]);
}
} catch (InvalidLogFormatException e) {
throw new RuntimeException(e);
}
}
/**
* Log some messages to the handler and verify that they are
* counted by the handler and that the count for each level is
* correct.
*/
@Test
public void testLevelCountTotal() throws java.io.IOException, InvalidLogFormatException {
LogMetricsHandler a = new LogMetricsHandler();
for (LogMessage aMsg : msg) {
a.handle(aMsg);
}
long count = a.getMetricsCount();
a.close();
// Not all messages are processes (debug and spam)
assertEquals(count, 7);
}
/**
* Log some messages to the handler and verify that they are
* counted by the handler and that the count for each level is
* correct (the count for each host is summed into one count for
* each level).
*/
@Test
public void testLevelCountAggregated() throws java.io.IOException, InvalidLogFormatException {
LogMetricsHandler a = new LogMetricsHandler();
for (LogMessage aMsg : msg) {
a.handle(aMsg);
}
Map<String, Long> levelCount = a.getMetricsPerLevel();
assertEquals(levelCount.entrySet().size(), 5);
for (Map.Entry<String, Long> entry : levelCount.entrySet()) {
String key = entry.getKey();
switch (key) {
case "config":
assertEquals(entry.getValue(), Long.valueOf(1));
break;
case "info":
assertEquals(entry.getValue(), Long.valueOf(4));
break;
case "warning":
assertEquals(entry.getValue(), Long.valueOf(1));
break;
case "severe":
assertEquals(entry.getValue(), Long.valueOf(0));
break;
case "error":
assertEquals(entry.getValue(), Long.valueOf(1));
break;
case "fatal":
assertEquals(entry.getValue(), Long.valueOf(1));
break;
case "debug":
assertEquals(entry.getValue(), Long.valueOf(0)); // always 0
break;
}
}
a.close();
}
@Test
public void testLogMetricsPlugin() {
LogMetricsPlugin lp = new LogMetricsPlugin();
try {
lp.shutdownPlugin();
fail("Shutdown before init didn't throw.");
} catch (Exception e) {
}
lp.initPlugin(new SystemPropertyConfig("test"));
try {
lp.initPlugin(new SystemPropertyConfig("test"));
fail("Multiple init didn't throw.");
} catch (Exception e) {
}
lp.shutdownPlugin();
}
}
|
