feat: add environmentEnabled property on feature defs (#295)

Author: sighphyre

pom.xml

@@ -59,7 +59,7 @@
         <dependency>
             <groupId>io.getunleash</groupId>
             <artifactId>yggdrasil-engine</artifactId>
-            <version>0.1.0</version>
+            <version>0.1.1</version>
         </dependency>
         <dependency>
             <groupId>com.google.code.gson</groupId>

src/main/java/io/getunleash/FakeUnleash.java

@@ -142,7 +142,10 @@ public Optional<FeatureDefinition> getFeatureToggleDefinition(String toggleName)
                     .map(
                             value ->
                                     new FeatureDefinition(
-                                            toggleName, Optional.of("experiment"), "default"));
+                                            toggleName,
+                                            Optional.of("experiment"),
+                                            "default",
+                                            true));
         }
 
         @Override

src/main/java/io/getunleash/FeatureDefinition.java

@@ -8,17 +8,21 @@ public class FeatureDefinition {
     private final String name;
     private final Optional<String> type;
     private final String project;
+    private final boolean environmentEnabled;
 
     public FeatureDefinition(FeatureDef source) {
         this.name = source.getName();
         this.type = source.getType();
         this.project = source.getProject();
+        this.environmentEnabled = source.isEnabled();
     }
 
-    public FeatureDefinition(String name, Optional<String> type, String project) {
+    public FeatureDefinition(
+            String name, Optional<String> type, String project, boolean environmentEnabled) {
         this.name = name;
         this.type = type;
         this.project = project;
+        this.environmentEnabled = environmentEnabled;
     }
 
     public String getName() {
@@ -32,4 +36,8 @@ public Optional<String> getType() {
     public String getProject() {
         return project;
     }
+
+    public boolean environmentEnabled() {
+        return environmentEnabled;
+    }
 }

src/main/java/io/getunleash/util/FeatureDefinitionAdapter.java

@@ -15,6 +15,7 @@ public void write(JsonWriter out, FeatureDefinition value) throws IOException {
         out.name("name").value(value.getName());
         out.name("project").value(value.getProject());
         out.name("type").value(value.getType().orElse(null));
+        out.name("enabled").value(value.environmentEnabled());
         out.endObject();
     }
 
@@ -23,6 +24,7 @@ public FeatureDefinition read(JsonReader in) throws IOException {
         String name = null;
         String project = null;
         Optional<String> type = Optional.empty();
+        boolean enabled = false;
 
         in.beginObject();
         while (in.hasNext()) {
@@ -36,6 +38,9 @@ public FeatureDefinition read(JsonReader in) throws IOException {
                 case "type":
                     type = Optional.of(in.nextString());
                     break;
+                case "enabled":
+                    enabled = in.nextBoolean();
+                    break;
                 default:
                     in.skipValue();
             }
@@ -46,6 +51,6 @@ public FeatureDefinition read(JsonReader in) throws IOException {
             throw new IOException("Missing required field 'name'");
         }
 
-        return new FeatureDefinition(name, type, project);
+        return new FeatureDefinition(name, type, project, enabled);
     }
 }

src/test/java/io/getunleash/repository/FeatureRepositoryTest.java

@@ -182,6 +182,29 @@ public void should_read_from_bootstrap_location_if_backup_was_empty() {
         assertEquals(5, knownToggles.size());
     }
 
+    @Test
+    public void should_return_enabled_property_on_known_toggles() {
+        when(backupHandler.read())
+                .thenReturn(Optional.of(loadMockFeatures("unleash-repo-v2.json")));
+
+        when(bootstrapHandler.read()).thenReturn(Optional.empty());
+
+        FeatureRepository featureRepository =
+                new FeatureRepositoryImpl(defaultConfig, backupHandler, new UnleashEngine());
+
+        List<FeatureDefinition> knownToggles =
+                featureRepository.listKnownToggles().collect(Collectors.toList());
+
+        assertEquals(5, knownToggles.size());
+        FeatureDefinition featureY =
+                knownToggles.stream().filter(f -> f.getName().equals("featureY")).findFirst().get();
+        assertThat(featureY.environmentEnabled()).isFalse();
+
+        FeatureDefinition featureX =
+                knownToggles.stream().filter(f -> f.getName().equals("featureX")).findFirst().get();
+        assertThat(featureX.environmentEnabled()).isTrue();
+    }
+
     @Test
     public void should_not_read_bootstrap_if_backup_was_found() {
 

src/test/java/io/getunleash/util/ClientFeaturesParserTest.java

@@ -66,4 +66,34 @@ public void test_deserialize_fails_if_name_is_not_set() {
             assertThat(e.getMessage()).contains("Missing required field 'name'");
         }
     }
+
+    @Test
+    public void test_enabled_property_returned_if_set() {
+        String basicFeatures =
+                "{\"features\":[{\"name\":\"featureX\",\"project\":\"default\",\"enabled\":true,\"strategies\":[{\"name\":\"default\"}]}]}";
+        List<FeatureDefinition> parsed = ClientFeaturesParser.parse(basicFeatures);
+
+        assertEquals(1, parsed.size());
+
+        FeatureDefinition feature = parsed.get(0);
+
+        assertEquals(feature.getName(), "featureX");
+        assertEquals(feature.getProject(), "default");
+        assertEquals(feature.environmentEnabled(), true);
+    }
+
+    @Test
+    public void test_enabled_property_defaults_to_false() {
+        String basicFeatures =
+                "{\"features\":[{\"name\":\"featureX\",\"project\":\"default\",\"strategies\":[{\"name\":\"default\"}]}]}";
+        List<FeatureDefinition> parsed = ClientFeaturesParser.parse(basicFeatures);
+
+        assertEquals(1, parsed.size());
+
+        FeatureDefinition feature = parsed.get(0);
+
+        assertEquals(feature.getName(), "featureX");
+        assertEquals(feature.getProject(), "default");
+        assertEquals(feature.environmentEnabled(), false);
+    }
 }