Add test case for client

libra/client/libraapi.py

@@ -31,12 +31,16 @@ class LibraAPI(object):
         resp, body = self._get('/loadbalaners')
         columns = ['Name', 'ID', 'Protocol', 'Port', 'Algorithm', 'Status',
                    'Created', 'Updated']
-        self._render(columns, body, 'loadBalancers')
+        self._render(columns, body['loadBalancers'])
 
-    def _render(self, columns, data, row_item):
+    def _render(self, columns, data):
         table = prettytable.PrettyTable(columns)
-        for item in data[row_item]:
-            table.add_row(item)
+        for item in data:
+            row = []
+            for column in columns:
+                rdata = item[column.lower()]
+                row.append(rdata)
+            table.add_row(row)
         print table
 
     def _get(self, url, **kwargs):

tests/test_lbaas_client.py

@@ -0,0 +1,58 @@
+import unittest
+import json
+import mock
+import httplib2
+import sys
+from StringIO import StringIO
+from libra.client.libraapi import LibraAPI
+
+fake_response = httplib2.Response({"status": '200'})
+fake_body = json.dumps({
+    "loadBalancers":[
+        {
+            "name":"lb-site1",
+            "id":"71",
+            "protocol":"HTTP",
+            "port":"80",
+            "algorithm":"LEAST_CONNECTIONS",
+            "status":"ACTIVE",
+            "created":"2010-11-30T03:23:42Z",
+            "updated":"2010-11-30T03:23:44Z"
+        },
+        {
+            "name":"lb-site2",
+            "id":"166",
+            "protocol":"TCP",
+            "port":"9123",
+            "algorithm":"ROUND_ROBIN",
+            "status":"ACTIVE",
+            "created":"2010-11-30T03:23:42Z",
+            "updated":"2010-11-30T03:23:44Z"
+        }
+    ]
+})
+
+mock_request = mock.Mock(return_value=(fake_response, fake_body))
+
+
+class TestLBaaSClientLibraAPI(unittest.TestCase):
+    def setUp(self):
+        self.api = LibraAPI('username', 'password', 'tenant', 'auth_test', 'region')
+        self.api.nova.management_url = "http://example.com"
+        self.api.nova.auth_token = "token"
+
+    def tearDown(self):
+        pass
+
+    def testListLb(self):
+        with mock.patch.object(httplib2.Http, "request", mock_request):
+            with mock.patch('time.time', mock.Mock(return_value=1234)):
+                orig = sys.stdout
+                try:
+                    out = StringIO()
+                    sys.stdout = out
+                    self.api.list_lb()
+                    output = out.getvalue().strip()
+                    self.assertRegexpMatches(output, 'LEAST_CONNECTIONS')
+                finally:
+                    sys.stdout = orig