Tests for status

cruatta · cruatta · commit 3a1dbfe354af · 2015-06-08T14:11:32.000-07:00
diff --git a/tests/test_std.py b/tests/test_std.py
@@ -758,3 +758,32 @@ def test_acl_implicit_token_use(self, acl_consul):
         acls = c.acl.list()
         assert set([x['ID'] for x in acls]) == \
             set(['anonymous', master_token])
+
+    def test_status_leader(self, consul_port):
+        c = consul.Consul(port=consul_port)
+
+        agent_self = c.agent.self()
+        port = agent_self['Config']['Ports']['Server']
+        addr = agent_self['Config']['AdvertiseAddr']
+
+        addr_port = "{0}:{1}".format(addr, port)
+        leader = c.status.leader()
+
+        assert leader == addr_port, \
+            "Leader value was {0}, expected value " \
+            "was {1}".format(leader, addr_port)
+
+    def test_status_peers(self, consul_port):
+
+        c = consul.Consul(port=consul_port)
+
+        agent_self = c.agent.self()
+        port = agent_self['Config']['Ports']['Server']
+        addr = agent_self['Config']['AdvertiseAddr']
+
+        addr_port = "{0}:{1}".format(addr, port)
+        peers = c.status.peers()
+
+        assert addr_port in peers, \
+            "Expected value '{0}' " \
+            "in peer list but it was not present".format(addr_port)
