1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.hadoop.hbase.metrics;
20
21 import org.apache.hadoop.hbase.testclassification.SmallTests;
22 import org.apache.hadoop.hbase.testclassification.MetricsTests;
23 import org.apache.hadoop.metrics2.lib.MutableCounterLong;
24 import org.apache.hadoop.metrics2.lib.MutableFastCounter;
25 import org.apache.hadoop.metrics2.lib.MutableGaugeLong;
26 import org.junit.BeforeClass;
27 import org.junit.Test;
28 import org.junit.experimental.categories.Category;
29
30 import static org.junit.Assert.assertEquals;
31 import static org.junit.Assert.assertNull;
32
33
34
35
36 @Category({MetricsTests.class, SmallTests.class})
37 public class TestBaseSourceImpl {
38
39 private static BaseSourceImpl bmsi;
40
41 @BeforeClass
42 public static void setUp() throws Exception {
43 bmsi = new BaseSourceImpl("TestName", "test description", "testcontext", "TestContext");
44 }
45
46 @Test
47 public void testSetGauge() throws Exception {
48 bmsi.setGauge("testset", 100);
49 assertEquals(100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testset")).value());
50 bmsi.setGauge("testset", 300);
51 assertEquals(300, ((MutableGaugeLong) bmsi.metricsRegistry.get("testset")).value());
52
53 }
54
55 @Test
56 public void testIncGauge() throws Exception {
57 bmsi.incGauge("testincgauge", 100);
58 assertEquals(100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testincgauge")).value());
59 bmsi.incGauge("testincgauge", 100);
60 assertEquals(200, ((MutableGaugeLong) bmsi.metricsRegistry.get("testincgauge")).value());
61
62 }
63
64 @Test
65 public void testDecGauge() throws Exception {
66 bmsi.decGauge("testdec", 100);
67 assertEquals(-100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testdec")).value());
68 bmsi.decGauge("testdec", 100);
69 assertEquals(-200, ((MutableGaugeLong) bmsi.metricsRegistry.get("testdec")).value());
70
71 }
72
73 @Test
74 public void testIncCounters() throws Exception {
75 bmsi.incCounters("testinccounter", 100);
76 assertEquals(100, ((MutableFastCounter) bmsi.metricsRegistry.get("testinccounter")).value());
77 bmsi.incCounters("testinccounter", 100);
78 assertEquals(200, ((MutableFastCounter) bmsi.metricsRegistry.get("testinccounter")).value());
79
80 }
81
82 @Test
83 public void testRemoveMetric() throws Exception {
84 bmsi.setGauge("testrmgauge", 100);
85 bmsi.removeMetric("testrmgauge");
86 assertNull(bmsi.metricsRegistry.get("testrmgauge"));
87 }
88
89 }