2e0e35317d112e040d436b19bf28873a2665e0aa
[snaps.git] / snaps / test_suite_builder.py
1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
2 #                    and others.  All rights reserved.
3 #
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at:
7 #
8 #     http://www.apache.org/licenses/LICENSE-2.0
9 #
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15
16 import logging
17 import unittest
18
19 from snaps.domain.test.image_tests import ImageDomainObjectTests
20 from snaps.openstack.utils.tests.glance_utils_tests import GlanceSmokeTests, GlanceUtilsTests
21 from snaps.openstack.tests.create_flavor_tests import CreateFlavorTests
22 from snaps.tests.file_utils_tests import FileUtilsTests
23 from snaps.openstack.tests.create_security_group_tests import CreateSecurityGroupTests, \
24     SecurityGroupRuleSettingsUnitTests, SecurityGroupSettingsUnitTests
25 from snaps.openstack.tests.create_project_tests import CreateProjectSuccessTests, ProjectSettingsUnitTests, \
26     CreateProjectUserTests
27 from snaps.openstack.tests.create_user_tests import UserSettingsUnitTests, CreateUserSuccessTests
28 from snaps.openstack.utils.tests.keystone_utils_tests import KeystoneSmokeTests, KeystoneUtilsTests
29 from snaps.openstack.utils.tests.neutron_utils_tests import NeutronSmokeTests, NeutronUtilsNetworkTests, \
30     NeutronUtilsSubnetTests, NeutronUtilsRouterTests, NeutronUtilsSecurityGroupTests
31 from snaps.openstack.tests.create_image_tests import CreateImageSuccessTests, CreateImageNegativeTests, \
32     ImageSettingsUnitTests, CreateMultiPartImageTests
33 from snaps.openstack.tests.create_keypairs_tests import CreateKeypairsTests, KeypairSettingsUnitTests
34 from snaps.openstack.tests.create_network_tests import CreateNetworkSuccessTests, NetworkSettingsUnitTests, \
35     PortSettingsUnitTests, SubnetSettingsUnitTests, CreateNetworkTypeTests
36 from snaps.openstack.tests.create_router_tests import CreateRouterSuccessTests, CreateRouterNegativeTests
37 from snaps.openstack.tests.create_instance_tests import CreateInstanceSingleNetworkTests, \
38     CreateInstancePubPrivNetTests, CreateInstanceOnComputeHost, CreateInstanceSimpleTests, \
39     FloatingIpSettingsUnitTests, InstanceSecurityGroupTests, VmInstanceSettingsUnitTests, \
40     CreateInstancePortManipulationTests, SimpleHealthCheck, CreateInstanceFromThreePartImage
41 from snaps.provisioning.tests.ansible_utils_tests import AnsibleProvisioningTests
42 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase, OSIntegrationTestCase
43 from snaps.openstack.utils.tests.nova_utils_tests import NovaSmokeTests, NovaUtilsKeypairTests, NovaUtilsFlavorTests
44
45 __author__ = 'spisarski'
46
47
48 def add_unit_tests(suite):
49     """
50     Adds tests that do not require external resources
51     :param suite: the unittest.TestSuite object to which to add the tests
52     :return: None as the tests will be adding to the 'suite' parameter object
53     """
54     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(FileUtilsTests))
55     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(SecurityGroupRuleSettingsUnitTests))
56     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(SecurityGroupSettingsUnitTests))
57     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(ImageSettingsUnitTests))
58     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(ImageDomainObjectTests))
59     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(KeypairSettingsUnitTests))
60     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(UserSettingsUnitTests))
61     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(ProjectSettingsUnitTests))
62     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(NetworkSettingsUnitTests))
63     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(SubnetSettingsUnitTests))
64     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(PortSettingsUnitTests))
65     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(FloatingIpSettingsUnitTests))
66     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(VmInstanceSettingsUnitTests))
67
68
69 def add_openstack_client_tests(suite, os_creds, ext_net_name, use_keystone=True, log_level=logging.INFO):
70     """
71     Adds tests written to exercise OpenStack client retrieval
72     :param suite: the unittest.TestSuite object to which to add the tests
73     :param os_creds: and instance of OSCreds that holds the credentials required by OpenStack
74     :param ext_net_name: the name of an external network on the cloud under test
75     :param use_keystone: when True, tests requiring direct access to Keystone are added as these need to be running on
76                          a host that has access to the cloud's private network
77     :param log_level: the logging level
78     :return: None as the tests will be adding to the 'suite' parameter object
79     """
80     # Basic connection tests
81     suite.addTest(OSComponentTestCase.parameterize(GlanceSmokeTests, os_creds=os_creds, ext_net_name=ext_net_name,
82                                                    log_level=log_level))
83
84     if use_keystone:
85         suite.addTest(OSComponentTestCase.parameterize(KeystoneSmokeTests, os_creds=os_creds, ext_net_name=ext_net_name,
86                                                        log_level=log_level))
87
88     suite.addTest(OSComponentTestCase.parameterize(NeutronSmokeTests, os_creds=os_creds, ext_net_name=ext_net_name,
89                                                    log_level=log_level))
90     suite.addTest(OSComponentTestCase.parameterize(NovaSmokeTests, os_creds=os_creds, ext_net_name=ext_net_name,
91                                                    log_level=log_level))
92
93
94 def add_openstack_api_tests(suite, os_creds, ext_net_name, use_keystone=True, log_level=logging.INFO):
95     """
96     Adds tests written to exercise all existing OpenStack APIs
97     :param suite: the unittest.TestSuite object to which to add the tests
98     :param os_creds: and instance of OSCreds that holds the credentials required by OpenStack
99     :param ext_net_name: the name of an external network on the cloud under test
100     :param use_keystone: when True, tests requiring direct access to Keystone are added as these need to be running on
101                          a host that has access to the cloud's private network
102     :param log_level: the logging level
103     :return: None as the tests will be adding to the 'suite' parameter object
104     """
105     # Tests the OpenStack API calls
106     if use_keystone:
107         suite.addTest(OSComponentTestCase.parameterize(
108             KeystoneUtilsTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
109         suite.addTest(OSComponentTestCase.parameterize(
110             CreateUserSuccessTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
111         suite.addTest(OSComponentTestCase.parameterize(
112             CreateProjectSuccessTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
113         suite.addTest(OSComponentTestCase.parameterize(
114             CreateProjectUserTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
115
116     suite.addTest(OSComponentTestCase.parameterize(
117         GlanceUtilsTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
118     suite.addTest(OSComponentTestCase.parameterize(
119         NeutronUtilsNetworkTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
120     suite.addTest(OSComponentTestCase.parameterize(
121         NeutronUtilsSubnetTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
122     suite.addTest(OSComponentTestCase.parameterize(
123         NeutronUtilsRouterTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
124     suite.addTest(OSComponentTestCase.parameterize(
125         NeutronUtilsSecurityGroupTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
126     suite.addTest(OSComponentTestCase.parameterize(
127         NovaUtilsKeypairTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
128     suite.addTest(OSComponentTestCase.parameterize(
129         NovaUtilsFlavorTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
130     suite.addTest(OSComponentTestCase.parameterize(
131         CreateFlavorTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))
132
133
134 def add_openstack_integration_tests(suite, os_creds, ext_net_name, use_keystone=True, flavor_metadata=None,
135                                     image_metadata=None, use_floating_ips=True, log_level=logging.INFO):
136     """
137     Adds tests written to exercise all long-running OpenStack integration tests meaning they will be creating VM
138     instances and potentially performing some SSH functions through floating IPs
139     :param suite: the unittest.TestSuite object to which to add the tests
140     :param os_creds: and instance of OSCreds that holds the credentials required by OpenStack
141     :param ext_net_name: the name of an external network on the cloud under test
142     :param use_keystone: when True, tests requiring direct access to Keystone are added as these need to be running on
143                          a host that has access to the cloud's private network
144     :param image_metadata: dict() object containing metadata for creating an image with custom config:
145                            (i.e. {'hw_video_model' : 'vga'}). It can be used to override the default url and
146                            create 3-part images by passing kerner_url and ramdisk_url info
147     :param flavor_metadata: dict() object containing the metadata required by your flavor based on your configuration:
148                             (i.e. {'hw:mem_page_size': 'large'})
149     :param use_floating_ips: when true, all tests requiring Floating IPs will be added to the suite
150     :param log_level: the logging level
151     :return: None as the tests will be adding to the 'suite' parameter object
152     """
153     # Tests the OpenStack API calls via a creator. If use_keystone, objects will be created with a custom user
154     # and project
155
156     # Creator Object tests
157     suite.addTest(OSIntegrationTestCase.parameterize(
158         CreateSecurityGroupTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
159         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
160     suite.addTest(OSIntegrationTestCase.parameterize(
161         CreateImageSuccessTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
162         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
163     suite.addTest(OSIntegrationTestCase.parameterize(
164         CreateImageNegativeTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
165         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
166     suite.addTest(OSIntegrationTestCase.parameterize(
167         CreateMultiPartImageTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
168         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
169     suite.addTest(OSIntegrationTestCase.parameterize(
170         CreateKeypairsTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
171         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
172     suite.addTest(OSIntegrationTestCase.parameterize(
173         CreateNetworkSuccessTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
174         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
175     suite.addTest(OSIntegrationTestCase.parameterize(
176         CreateRouterSuccessTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
177         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
178     suite.addTest(OSIntegrationTestCase.parameterize(
179         CreateRouterNegativeTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
180         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
181
182     # VM Instances
183     suite.addTest(OSIntegrationTestCase.parameterize(
184         SimpleHealthCheck, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
185         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
186     suite.addTest(OSIntegrationTestCase.parameterize(
187         CreateInstanceSimpleTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
188         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
189     suite.addTest(OSIntegrationTestCase.parameterize(
190         CreateInstancePortManipulationTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
191         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
192     suite.addTest(OSIntegrationTestCase.parameterize(
193         InstanceSecurityGroupTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
194         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
195     suite.addTest(OSIntegrationTestCase.parameterize(
196         CreateInstanceOnComputeHost, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
197         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
198     suite.addTest(OSIntegrationTestCase.parameterize(
199         CreateInstanceFromThreePartImage, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
200         flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
201
202     if use_floating_ips:
203         suite.addTest(OSIntegrationTestCase.parameterize(
204             CreateInstanceSingleNetworkTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
205             flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
206         suite.addTest(OSIntegrationTestCase.parameterize(
207             CreateInstancePubPrivNetTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
208             flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
209         suite.addTest(OSIntegrationTestCase.parameterize(
210             AnsibleProvisioningTests, os_creds=os_creds, ext_net_name=ext_net_name, use_keystone=use_keystone,
211             flavor_metadata=flavor_metadata, image_metadata=image_metadata, log_level=log_level))
212
213
214 def add_openstack_staging_tests(suite, os_creds, ext_net_name, log_level=logging.INFO):
215     """
216     Adds tests that are still in development have not been designed to run successfully against all OpenStack pods
217     :param suite: the unittest.TestSuite object to which to add the tests
218     :param os_creds: and instance of OSCreds that holds the credentials required by OpenStack
219     :param ext_net_name: the name of an external network on the cloud under test
220     :param log_level: the logging level
221     :return: None as the tests will be adding to the 'suite' parameter object
222     """
223     suite.addTest(OSComponentTestCase.parameterize(
224         CreateNetworkTypeTests, os_creds=os_creds, ext_net_name=ext_net_name, log_level=log_level))