Skip to content

Commit 3e9d1ac

Browse files
committed
Fix metrics health check tests
1 parent 38bbd8c commit 3e9d1ac

3 files changed

Lines changed: 53 additions & 40 deletions

File tree

jooby-metrics/src/test/java/org/jooby/internal/metrics/HealthCheckRegistryInitializerTest.java

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,13 @@
44

55
import java.util.Map;
66

7-
import com.codahale.metrics.health.HealthCheck;
87
import org.jooby.test.MockUnit;
98
import org.junit.Test;
109
import org.junit.runner.RunWith;
1110
import org.powermock.core.classloader.annotations.PrepareForTest;
1211
import org.powermock.modules.junit4.PowerMockRunner;
1312

13+
import com.codahale.metrics.health.HealthCheck;
1414
import com.codahale.metrics.health.HealthCheckRegistry;
1515
import com.google.common.collect.ImmutableMap;
1616

@@ -21,15 +21,14 @@ public class HealthCheckRegistryInitializerTest {
2121
@Test
2222
public void register() throws Exception {
2323
new MockUnit(HealthCheckRegistry.class, HealthCheck.class)
24-
.expect(unit -> {
25-
HealthCheckRegistry registry = unit.get(HealthCheckRegistry.class);
26-
registry.register("h", unit.get(HealthCheck.class));
27-
expectLastCall();
28-
})
29-
.run(unit -> {
30-
Map<String, HealthCheck> checks = ImmutableMap.of("h", unit.get(HealthCheck.class));
31-
HealthCheckRegistryInitializer hcri = new HealthCheckRegistryInitializer(
32-
unit.get(HealthCheckRegistry.class), checks);
33-
});
24+
.expect(unit -> {
25+
HealthCheckRegistry registry = unit.get(HealthCheckRegistry.class);
26+
registry.register("h", unit.get(HealthCheck.class));
27+
expectLastCall();
28+
})
29+
.run(unit -> {
30+
Map<String, HealthCheck> checks = ImmutableMap.of("h", unit.get(HealthCheck.class));
31+
new HealthCheckRegistryInitializer(unit.get(HealthCheckRegistry.class), checks);
32+
});
3433
}
3534
}

jooby-metrics/src/test/java/org/jooby/metrics/MetricsTest.java

Lines changed: 36 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55

66
import java.util.function.BiConsumer;
77

8-
import com.codahale.metrics.health.HealthCheckRegistry;
98
import org.jooby.Env;
109
import org.jooby.Jooby;
1110
import org.jooby.Route;
@@ -25,6 +24,7 @@
2524
import com.codahale.metrics.MetricRegistry;
2625
import com.codahale.metrics.Reporter;
2726
import com.codahale.metrics.health.HealthCheck;
27+
import com.codahale.metrics.health.HealthCheckRegistry;
2828
import com.google.inject.Binder;
2929
import com.google.inject.binder.AnnotatedBindingBuilder;
3030
import com.google.inject.binder.LinkedBindingBuilder;
@@ -59,6 +59,7 @@ public class MetricsTest {
5959
expect(binder.bind(MetricRegistryInitializer.class)).andReturn(mrABB);
6060
};
6161

62+
@SuppressWarnings("unchecked")
6263
private Block bindHealthCheckRegistry = unit -> {
6364
HealthCheckRegistry registry = unit.get(HealthCheckRegistry.class);
6465

@@ -72,7 +73,7 @@ public class MetricsTest {
7273
@SuppressWarnings("unchecked")
7374
private Block bindHealthCheckRegistryInitializer = unit -> {
7475
AnnotatedBindingBuilder<HealthCheckRegistryInitializer> hcrABB = unit
75-
.mock(AnnotatedBindingBuilder.class);
76+
.mock(AnnotatedBindingBuilder.class);
7677
hcrABB.asEagerSingleton();
7778

7879
Binder binder = unit.get(Binder.class);
@@ -87,7 +88,7 @@ public class MetricsTest {
8788

8889
private Block newHealthCheckRegistry = unit -> {
8990
HealthCheckRegistry registry = unit.constructor(HealthCheckRegistry.class)
90-
.build();
91+
.build();
9192
unit.registerMock(HealthCheckRegistry.class, registry);
9293
};
9394

@@ -166,6 +167,11 @@ public void basicWithExtMetricRegistry() throws Exception {
166167
MetricRegistry registry = unit.mock(MetricRegistry.class);
167168
unit.registerMock(MetricRegistry.class, registry);
168169
})
170+
.expect(unit -> {
171+
HealthCheckRegistry registry = unit.constructor(HealthCheckRegistry.class)
172+
.build();
173+
unit.registerMock(HealthCheckRegistry.class, registry);
174+
})
169175
.expect(mapBinderStatic)
170176
.expect(mapbinder(Metric.class, (unit, binder) -> {
171177
}))
@@ -191,31 +197,31 @@ public void basicWithExtMetricRegistry() throws Exception {
191197
@Test
192198
public void basicWithExtHealthCheckRegistry() throws Exception {
193199
new MockUnit(Env.class, Config.class, Binder.class)
194-
.expect(newMetricRegistry)
195-
.expect(unit -> {
196-
HealthCheckRegistry registry = unit.mock(HealthCheckRegistry.class);
197-
unit.registerMock(HealthCheckRegistry.class, registry);
198-
})
199-
.expect(mapBinderStatic)
200-
.expect(mapbinder(Metric.class, (unit, binder) -> {
201-
}))
202-
.expect(mapbinder(HealthCheck.class, (unit, binder) -> {
203-
}))
204-
.expect(multibinderStatic)
205-
.expect(routes)
206-
.expect(setbinder(Reporter.class, (unit, binder) -> {
207-
208-
}))
209-
.expect(bindMetricRegistry)
210-
.expect(bindMetricRegistryInitializer)
211-
.expect(bindHealthCheckRegistry)
212-
.expect(bindHealthCheckRegistryInitializer)
213-
.expect(onStop)
214-
.run(unit -> {
215-
HealthCheckRegistry registry = unit.get(HealthCheckRegistry.class);
216-
new Metrics(registry)
217-
.configure(unit.get(Env.class), unit.get(Config.class), unit.get(Binder.class));
218-
});
200+
.expect(newMetricRegistry)
201+
.expect(unit -> {
202+
HealthCheckRegistry registry = unit.mock(HealthCheckRegistry.class);
203+
unit.registerMock(HealthCheckRegistry.class, registry);
204+
})
205+
.expect(mapBinderStatic)
206+
.expect(mapbinder(Metric.class, (unit, binder) -> {
207+
}))
208+
.expect(mapbinder(HealthCheck.class, (unit, binder) -> {
209+
}))
210+
.expect(multibinderStatic)
211+
.expect(routes)
212+
.expect(setbinder(Reporter.class, (unit, binder) -> {
213+
214+
}))
215+
.expect(bindMetricRegistry)
216+
.expect(bindMetricRegistryInitializer)
217+
.expect(bindHealthCheckRegistry)
218+
.expect(bindHealthCheckRegistryInitializer)
219+
.expect(onStop)
220+
.run(unit -> {
221+
HealthCheckRegistry registry = unit.get(HealthCheckRegistry.class);
222+
new Metrics(registry)
223+
.configure(unit.get(Env.class), unit.get(Config.class), unit.get(Binder.class));
224+
});
219225
}
220226

221227
@Test
@@ -305,7 +311,8 @@ public void threadDump() throws Exception {
305311
.expect(unit -> {
306312
Router routes = unit.get(Router.class);
307313

308-
route(unit, routes, "/sys/thread-dump", unit.constructor(ThreadDumpHandler.class).build());
314+
route(unit, routes, "/sys/thread-dump",
315+
unit.constructor(ThreadDumpHandler.class).build());
309316
})
310317
.expect(setbinder(Reporter.class, (unit, binder) -> {
311318

jooby-servlet/src/test/java/org/jooby/servlet/ServletContainerTest.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
package org.jooby.servlet;
22

3+
import static org.junit.Assert.assertFalse;
4+
35
import org.junit.Test;
46

57
public class ServletContainerTest {
@@ -18,4 +20,9 @@ public void stop() throws Exception {
1820
public void join() throws Exception {
1921
ServletContainer.NOOP.join();
2022
}
23+
24+
@Test
25+
public void excutor() throws Exception {
26+
assertFalse(ServletContainer.NOOP.executor().isPresent());
27+
}
2128
}

0 commit comments

Comments
 (0)