CDM-20 better isEnabled wrapper

Author: mieslep

src/main/java/datastax/astra/migrate/cql/CqlHelper.java

@@ -457,4 +457,11 @@ private void abendIfSessionsNotSet() {
     public Feature getFeature(Featureset featureEnum) {
         return featureMap.get(featureEnum);
     }
+
+    public Boolean isFeatureEnabled(Featureset featureEnum) {
+        if (!featureMap.containsKey(featureEnum)) {
+            return false;
+        }
+        return featureMap.get(featureEnum).isEnabled();
+    }
 }

src/test/java/datastax/astra/migrate/CqlHelperTest.java

@@ -1,15 +1,15 @@
 package datastax.astra.migrate;
 
 import datastax.astra.migrate.cql.CqlHelper;
+import datastax.astra.migrate.cql.features.Featureset;
 import datastax.astra.migrate.properties.KnownProperties;
 import datastax.astra.migrate.properties.PropertyHelper;
 import org.apache.spark.SparkConf;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
-import static org.junit.jupiter.api.Assertions.assertAll;
-import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.*;
 
 public class CqlHelperTest {
 
@@ -54,4 +54,10 @@ public void smokeTest() {
         );
     }
 
+    @Test
+    public void featureHelper_disabledWhenNull() {
+        propertyHelper.initializeSparkConf(sparkConf);
+        cqlHelper.initialize();
+        assertFalse(cqlHelper.isFeatureEnabled(Featureset.TEST_UNIMPLEMENTED_FEATURE));
+    }
 }