pools: add method to rebalance a pool

tests/test_pools.py

@@ -0,0 +1,30 @@
+from tsuruclient import client
+
+import json
+import unittest
+import httpretty
+
+
+class PoolsTestCase(unittest.TestCase):
+    def setUp(self):
+        httpretty.enable()
+
+    def tearDown(self):
+        httpretty.disable()
+        httpretty.reset()
+
+    def test_rebalance(self):
+        url = "http://target/docker/containers/rebalance"
+        httpretty.register_uri(
+            httpretty.POST,
+            url,
+            body=json.dumps("{}"),
+            status=200
+        )
+
+        cl = client.Client("http://target", "abc123")
+        cl.pools.rebalance("dev")
+
+        result = json.loads(httpretty.last_request().body)
+        expected = {"metadataFilter": {"pool": "dev"}}
+        self.assertDictEqual(expected, result)

tsuruclient/client.py

@@ -1,8 +1,9 @@
-from tsuruclient import apps, nodes, templates
+from tsuruclient import apps, nodes, templates, pools
 
 
 class Client(object):
     def __init__(self, target, token):
         self.apps = apps.Manager(target, token)
         self.nodes = nodes.Manager(target, token)
         self.templates = templates.Manager(target, token)
+        self.pools = pools.Manager(target, token)

tsuruclient/pools.py

@@ -0,0 +1,22 @@
+import json
+import requests
+
+from base import Manager as Base
+
+
+class Manager(Base):
+    """
+    Manage pool resources.
+    """
+
+    def rebalance(self, pool):
+        """
+        Rebalance a pool.
+        """
+        data = {"metadataFilter": {"pool": pool}}
+        response = requests.post(
+            "{}/docker/containers/rebalance".format(self.target),
+            data=json.dumps(data),
+            headers=self.headers
+        )
+        return response.json()