Merge "Add test_list_stacks"

1 files changed, 10 insertions, 5 deletions

diff --git a/vio/vio/tests/test_fake_heat_view.py b/vio/vio/tests/test_fake_heat_view.py
index 3cdb41a..5e87a61 100644
--- a/vio/vio/tests/test_fake_heat_view.py
+++ b/vio/vio/tests/test_fake_heat_view.py
@@ -67,18 +67,23 @@ class TestFakeHeatService(unittest.TestCase):
     def setUp(self):
         self.view = views.FakeHeatService()
 
-    @mock.patch.object(fakeResponse, "getAllStacks")
-    def test_get_heat_stacks(self, mock_getAllStacks):
+    def test_get_heat_stack(self):
         req = mock.Mock()
         req.META = {
             "HTTP_X_AUTH_TOKEN": Token
         }
-        mock_getAllStacks.return_value = {
-            "stack": "1234abcd"
-        }
         resp = self.view.get(req, "1234abcd", STACK_NAME, STACK_ID)
         self.assertEqual(200, resp.status_code)
 
+    def test_list_stacks(self):
+        req = mock.Mock()
+        req.META = {
+            "HTTP_X_AUTH_TOKEN": Token
+        }
+        resp = self.view.get(req, "1234abcd")
+        self.assertEqual(200, resp.status_code)
+        self.assertEqual(1, len(resp.data['stacks']))
+
     @mock.patch.object(fakeResponse, "createStack")
     def test_create_service_preview(self, mock_createStack):
         req = mock.Mock()