Add basic envi command support

2 files changed, 14 insertions, 0 deletions

diff --git a/zake/fake_client.py b/zake/fake_client.py
index 6c4787f..d5554d0 100644
--- a/zake/fake_client.py
+++ b/zake/fake_client.py
@@ -131,6 +131,13 @@ class FakeClient(object):
                               'Mode: standalone'])
         if cmd == "kill":
             self.stop()
+        if cmd == 'envi':
+            server_version = ".".join([str(s) for s in self._server_version])
+            lines = [
+                "Environment:",
+                "zookeeper.version=%s" % server_version,
+            ]
+            return "\n".join(lines)
         return ''
 
     def verify(self):
diff --git a/zake/tests/test_client.py b/zake/tests/test_client.py
index 334cc05..44e0acd 100644
--- a/zake/tests/test_client.py
+++ b/zake/tests/test_client.py
@@ -87,6 +87,13 @@ class TestClient(test.Test):
             version = ".".join([str(s) for s in fake_client.SERVER_VERSION])
             self.assertIn(version, stats)
 
+    def test_command_envi(self):
+        with start_close(self.client) as c:
+            envi = c.command('envi')
+            self.assertIn("zookeeper.version", envi)
+            version = ".".join([str(s) for s in fake_client.SERVER_VERSION])
+            self.assertIn(version, envi)
+
     def test_command_empty_version(self):
         self.assertRaises(ValueError, fake_client.FakeClient,
                           server_version=[])