Add isDisabled to ConsulFactory

Closes #122
kiwiproject · Aug 8, 2023 · 6ade436 · 6ade436
1 parent 23168b3
commit 6ade436
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/main/java/org/kiwiproject/dropwizard/consul/ConsulFactory.java b/src/main/java/org/kiwiproject/dropwizard/consul/ConsulFactory.java
@@ -72,6 +72,11 @@ public void setEnabled(boolean enabled) {
         this.enabled = enabled;
     }
 
+    @JsonIgnore
+    public boolean isDisabled() {
+        return !isEnabled();
+    }
+
     @JsonProperty
     public HostAndPort getEndpoint() {
         return endpoint;

diff --git a/src/test/java/org/kiwiproject/dropwizard/consul/ConsulFactoryTest.java b/src/test/java/org/kiwiproject/dropwizard/consul/ConsulFactoryTest.java
@@ -4,6 +4,9 @@
 import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
 
 import io.dropwizard.util.Duration;
+
+import static org.junit.jupiter.api.Assertions.assertAll;
+
 import org.junit.jupiter.api.Test;
 
 import java.util.List;
@@ -61,6 +64,27 @@ void testSetServiceName() {
         assertThat(consulFactory.getServiceName()).isEqualTo(serviceName);
     }
 
+    @Test
+    void shouldBeEnabledByDefault() {
+        var consulFactory = new ConsulFactory();
+
+        assertAll(
+                () -> assertThat(consulFactory.isEnabled()).isTrue(),
+                () -> assertThat(consulFactory.isDisabled()).isFalse()
+        );
+    }
+
+    @Test
+    void shouldAllowDisabling() {
+        var consulFactory = new ConsulFactory();
+        consulFactory.setEnabled(false);
+
+        assertAll(
+                () -> assertThat(consulFactory.isEnabled()).isFalse(),
+                () -> assertThat(consulFactory.isDisabled()).isTrue()
+        );
+    }
+
     private ConsulFactory createFullyPopulatedConsulFactory() {
         var consulFactory = new ConsulFactory();
         consulFactory.setServiceName("serviceName");