diff options
| -rw-r--r-- | vio/vio/tests/test_fake_heat_view.py | 20 |
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) |