diff options
-rw-r--r-- | zake/fake_client.py | 7 | ||||
-rw-r--r-- | zake/tests/test_client.py | 7 |
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=[]) |