Improve coverage flow/controller/node #3

[appc.git] / appc-config / appc-flow-controller / provider / src / main / java / org / onap / appc / flow / controller / node / EnvVariables.java

diff --git a/appc-config/appc-flow-controller/provider/src/main/java/org/onap/appc/flow/controller/node/EnvVariables.java b/appc-config/appc-flow-controller/provider/src/main/java/org/onap/appc/flow/controller/node/EnvVariables.java
new file mode 100644 (file)
index 0000000..6cc3b75
--- /dev/null
+++ b/appc-config/appc-flow-controller/provider/src/main/java/org/onap/appc/flow/controller/node/EnvVariables.java
@@ -0,0 +1,26 @@
+package org.onap.appc.flow.controller.node;
+
+import java.util.function.Function;
+
+/**
+ * Wrapper for accessing environment variables
+ */
+class EnvVariables {
+
+  private Function<String, String> envSupplier;
+
+  EnvVariables() {
+    envSupplier = System::getenv;
+  }
+
+  /**
+   * Allows to override environment variables in tests, prefer to use default constructor
+   */
+  EnvVariables(Function<String, String> envSupplier) {
+    this.envSupplier = envSupplier;
+  }
+
+  String getenv(String variable) {
+    return envSupplier.apply(variable);
+  }
+}