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.MutableGaugeLong;
25 import org.junit.BeforeClass;
26 import org.junit.Test;
27 import org.junit.experimental.categories.Category;
28
29 import static org.junit.Assert.assertEquals;
30 import static org.junit.Assert.assertNull;
31
32
33
34
35 @Category({MetricsTests.class, SmallTests.class})
36 public class TestBaseSourceImpl {
37
38 private static BaseSourceImpl bmsi;
39
40 @BeforeClass
41 public static void setUp() throws Exception {
42 bmsi = new BaseSourceImpl("TestName", "test description", "testcontext", "TestContext");
43 }
44
45 @Test
46 public void testSetGauge() throws Exception {
47 bmsi.setGauge("testset", 100);
48 assertEquals(100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testset")).value());
49 bmsi.setGauge("testset", 300);
50 assertEquals(300, ((MutableGaugeLong) bmsi.metricsRegistry.get("testset")).value());
51
52 }
53
54 @Test
55 public void testIncGauge() throws Exception {
56 bmsi.incGauge("testincgauge", 100);
57 assertEquals(100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testincgauge")).value());
58 bmsi.incGauge("testincgauge", 100);
59 assertEquals(200, ((MutableGaugeLong) bmsi.metricsRegistry.get("testincgauge")).value());
60
61 }
62
63 @Test
64 public void testDecGauge() throws Exception {
65 bmsi.decGauge("testdec", 100);
66 assertEquals(-100, ((MutableGaugeLong) bmsi.metricsRegistry.get("testdec")).value());
67 bmsi.decGauge("testdec", 100);
68 assertEquals(-200, ((MutableGaugeLong) bmsi.metricsRegistry.get("testdec")).value());
69
70 }
71
72 @Test
73 public void testIncCounters() throws Exception {
74 bmsi.incCounters("testinccounter", 100);
75 assertEquals(100, ((MutableCounterLong) bmsi.metricsRegistry.get("testinccounter")).value());
76 bmsi.incCounters("testinccounter", 100);
77 assertEquals(200, ((MutableCounterLong) bmsi.metricsRegistry.get("testinccounter")).value());
78
79 }
80
81 @Test
82 public void testRemoveMetric() throws Exception {
83 bmsi.setGauge("testrmgauge", 100);
84 bmsi.removeMetric("testrmgauge");
85 assertNull(bmsi.metricsRegistry.get("testrmgauge"));
86 }
87
88 }