Merge "Add fake create heat service test"

1 files changed, 20 insertions, 0 deletions

diff --git a/vio/vio/tests/test_fake_heat_view.py b/vio/vio/tests/test_fake_heat_view.py
index bca8c68..01a7568 100644
--- a/vio/vio/tests/test_fake_heat_view.py
+++ b/vio/vio/tests/test_fake_heat_view.py
@@ -12,6 +12,7 @@
 
 import mock
 import unittest
+import json
 
 from vio.swagger.views.fakeplugin.heat import views
 from vio.swagger.views.fakeplugin.fakeData import fakeResponse
@@ -40,3 +41,22 @@ class TestFakeHeatResources(unittest.TestCase):
         }
         resp = self.view.get(req, "1234abcd", STACK_ID)
         self.assertEqual(200, resp.status_code)
+
+
+class TestFakeHeatServicePreview(unittest.TestCase):
+
+    def setUp(self):
+        self.view = views.FakeHeatServicePreview()
+
+    @mock.patch.object(fakeResponse, "upload_image")
+    def test_create_service_preview(self, mock_createStackPreview):
+        req = mock.Mock()
+        req.META = {
+            "HTTP_X_AUTH_TOKEN": Token
+        }
+        req.body = json.dumps({"stack_name": STACK_NAME})
+        mock_createStackPreview.return_value = {
+            "stack": "stack1"
+        }
+        resp = self.view.post(req, "1234abcd")
+        self.assertEqual(201, resp.status_code)