diff options
Diffstat (limited to 'yardstick/tests/unit/benchmark/scenarios/lib/test_create_image.py')
-rw-r--r-- | yardstick/tests/unit/benchmark/scenarios/lib/test_create_image.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/yardstick/tests/unit/benchmark/scenarios/lib/test_create_image.py b/yardstick/tests/unit/benchmark/scenarios/lib/test_create_image.py new file mode 100644 index 000000000..b26957979 --- /dev/null +++ b/yardstick/tests/unit/benchmark/scenarios/lib/test_create_image.py @@ -0,0 +1,43 @@ +############################################################################## +# Copyright (c) 2017 Huawei Technologies Co.,Ltd and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## +import unittest +import mock + +from yardstick.benchmark.scenarios.lib import create_image +from yardstick.common import openstack_utils + +# NOTE(elfoley): There should be more tests here. +class CreateImageTestCase(unittest.TestCase): + + @mock.patch.object(openstack_utils, 'create_image') + @mock.patch.object(openstack_utils, 'get_glance_client') + def test_create_image(self, mock_get_glance_client, mock_create_image): + options = { + 'image_name': 'yardstick_test_image_01', + 'disk_format': 'qcow2', + 'container_format': 'bare', + 'min_disk': '1', + 'min_ram': '512', + 'protected': 'False', + 'tags': '["yardstick automatic test image"]', + 'file_path': '/home/opnfv/images/cirros-0.3.5-x86_64-disk.img' + } + args = {"options": options} + obj = create_image.CreateImage(args, {}) + obj.run({}) + mock_create_image.assert_called_once() + mock_get_glance_client.assert_called_once() + + +def main(): + unittest.main() + + +if __name__ == '__main__': + main() |