From c693cabf87e451c835fffaab14299c477a7c5796 Mon Sep 17 00:00:00 2001 From: wangqun Date: Tue, 1 Mar 2016 07:41:01 +0000 Subject: [PATCH] Add flavor_id unit test for post baymodel Change-Id: I0215a89cb98421fb2845b75d5ec6308c87f99591 --- .../unit/api/controllers/v1/test_baymodel.py | 28 +++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/magnum/tests/unit/api/controllers/v1/test_baymodel.py b/magnum/tests/unit/api/controllers/v1/test_baymodel.py index a5aadb0c06..0378fbcfdf 100644 --- a/magnum/tests/unit/api/controllers/v1/test_baymodel.py +++ b/magnum/tests/unit/api/controllers/v1/test_baymodel.py @@ -736,6 +736,34 @@ class TestPost(api_base.FunctionalTest): response = self.post_json('/baymodels', bdict, expect_errors=True) self.assertEqual(404, response.status_int) + @mock.patch('magnum.api.attr_validator.validate_image') + @mock.patch('magnum.api.attr_validator.validate_os_resources') + def test_create_baymodel_with_flavor(self, + mock_valid_os_res, + mock_image_data): + mock_valid_os_res.return_value = None + mock_image_data.return_value = {'name': 'mock_name', + 'os_distro': 'fedora-atomic'} + bdict = apiutils.baymodel_post_data() + response = self.post_json('/baymodels', bdict) + self.assertEqual(201, response.status_int) + self.assertEqual(bdict['flavor_id'], + response.json['flavor_id']) + self.assertEqual(bdict['master_flavor_id'], + response.json['master_flavor_id']) + + @mock.patch('magnum.api.attr_validator.validate_image') + @mock.patch('magnum.api.attr_validator.validate_os_resources') + def test_create_baymodel_with_no_exist_flavor(self, + mock_valid_os_res, + mock_image_data): + mock_valid_os_res.side_effect = exception.FlavorNotFound("flavor_id") + mock_image_data.return_value = {'name': 'mock_name', + 'os_distro': 'fedora-atomic'} + bdict = apiutils.baymodel_post_data() + response = self.post_json('/baymodels', bdict, expect_errors=True) + self.assertEqual(400, response.status_int) + class TestDelete(api_base.FunctionalTest):