Merge pull request #56 from imgix/health_node_endpoint

cablehead · cablehead · commit cb0ac732481e · 2015-06-18T18:57:39.000-07:00
add health/node/&lt;node&gt; endpoint
diff --git a/consul/base.py b/consul/base.py
@@ -1143,6 +1143,27 @@ def callback(response):
                 callback,
                 '/v1/health/state/%s' % name, params=params)
 
+        def node(self, node, index=None):
+            """
+            Returns a tuple of (*index*, *checks*)
+
+            *index* is the current Consul index, suitable for making subsequent
+            calls to wait for changes since this query was last run.
+
+            *nodes* are the nodes providing the given service.
+            """
+            params = {}
+            if index:
+                params['index'] = index
+
+            def callback(response):
+                data = json.loads(response.body)
+                return response.headers['X-Consul-Index'], data
+
+            return self.agent.http.get(
+                callback,
+                '/v1/health/node/%s' % node, params=params)
+
     class Session(object):
         def __init__(self, agent):
             self.agent = agent
diff --git a/tests/test_std.py b/tests/test_std.py
@@ -560,6 +560,13 @@ def test_health_state(self, consul_port):
         index, nodes = c.health.state('any')
         assert [node['ServiceID'] for node in nodes] == ['']
 
+    def test_health_node(self, consul_port):
+        c = consul.Consul(port=consul_port)
+        # grab local node name
+        node = c.agent.self()['Config']['NodeName']
+        index, checks = c.health.node(node)
+        assert node in [check["Node"] for check in checks]
+
     def test_session(self, consul_port):
         c = consul.Consul(port=consul_port)
 
