Clean up error handling to make things slightly friendlier on an initial install.

[cfb.git] / test / net / jaekl / cfb / util / EnvMock.java

diff --git a/test/net/jaekl/cfb/util/EnvMock.java b/test/net/jaekl/cfb/util/EnvMock.java
new file mode 100644 (file)
index 0000000..cabf42f
--- /dev/null
+++ b/test/net/jaekl/cfb/util/EnvMock.java
@@ -0,0 +1,29 @@
+package net.jaekl.cfb.util;
+
+import java.util.HashMap;
+
+public class EnvMock extends Env {
+       private HashMap<String, String> m_map;
+       
+       EnvMock() {
+               m_map = new HashMap<String, String>();
+       }
+       
+       @Override
+       String getEnv(String variableName) {
+               if (m_map.containsKey(variableName)) {
+                       return m_map.get(variableName);
+               }
+               return super.getEnv(variableName);
+       }
+       
+       public static EnvMock mock_putInstance() {
+               EnvMock mock = new EnvMock();
+               m_inst = mock;
+               return mock;
+       }
+       
+       public void mock_putEnv(String variableName, String value) {
+               m_map.put(variableName, value);
+       }
+}