Merge "Refactoring"
[sdnvpn.git] / test / functest / testcase_4.py
1 #!/usr/bin/python
2 #
3 # Copyright (c) 2015 All rights reserved
4 # This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 #
8 # http://www.apache.org/licenses/LICENSE-2.0
9 #
10
11 import argparse
12 import os
13 from random import randint
14 import sys
15 import time
16
17 import functest.utils.functest_logger as ft_logger
18 import functest.utils.functest_utils as ft_utils
19 import functest.utils.openstack_utils as os_utils
20
21 import utils as test_utils
22 from results import Results
23
24 parser = argparse.ArgumentParser()
25
26 parser.add_argument("-r", "--report",
27                     help="Create json result file",
28                     action="store_true")
29
30 args = parser.parse_args()
31
32 logger = ft_logger.Logger("sdnvpn-testcase-4").getLogger()
33
34 REPO_PATH = os.environ['repos_dir'] + '/sdnvpn/'
35
36 VM_BOOT_TIMEOUT = 180
37
38 config_file = REPO_PATH + 'test/functest/config.yaml'
39
40 INSTANCE_1_NAME = ft_utils.get_parameter_from_yaml(
41     "testcases.testcase_4.instance_1_name", config_file)
42 INSTANCE_2_NAME = ft_utils.get_parameter_from_yaml(
43     "testcases.testcase_4.instance_2_name", config_file)
44 INSTANCE_3_NAME = ft_utils.get_parameter_from_yaml(
45     "testcases.testcase_4.instance_3_name", config_file)
46 INSTANCE_4_NAME = ft_utils.get_parameter_from_yaml(
47     "testcases.testcase_4.instance_4_name", config_file)
48 INSTANCE_5_NAME = ft_utils.get_parameter_from_yaml(
49     "testcases.testcase_4.instance_5_name", config_file)
50 IMAGE_NAME = ft_utils.get_parameter_from_yaml(
51     "testcases.testcase_4.image_name", config_file)
52 IMAGE_FILENAME = ft_utils.get_functest_config(
53     "general.openstack.image_file_name")
54 IMAGE_FORMAT = ft_utils.get_functest_config(
55     "general.openstack.image_disk_format")
56 IMAGE_PATH = ft_utils.get_functest_config(
57     "general.directories.dir_functest_data") + "/" + IMAGE_FILENAME
58
59 # NEUTRON Private Network parameters
60
61 NET_1_NAME = ft_utils.get_parameter_from_yaml(
62     "testcases.testcase_4.net_1_name", config_file)
63 SUBNET_1_NAME = ft_utils.get_parameter_from_yaml(
64     "testcases.testcase_4.subnet_1_name", config_file)
65 SUBNET_1_CIDR = ft_utils.get_parameter_from_yaml(
66     "testcases.testcase_4.subnet_1_cidr", config_file)
67 ROUTER_1_NAME = ft_utils.get_parameter_from_yaml(
68     "testcases.testcase_4.router_1_name", config_file)
69 NET_2_NAME = ft_utils.get_parameter_from_yaml(
70     "testcases.testcase_4.net_2_name", config_file)
71 SUBNET_2_NAME = ft_utils.get_parameter_from_yaml(
72     "testcases.testcase_4.subnet_2_name", config_file)
73 SUBNET_2_CIDR = ft_utils.get_parameter_from_yaml(
74     "testcases.testcase_4.subnet_2_cidr", config_file)
75 ROUTER_2_NAME = ft_utils.get_parameter_from_yaml(
76     "testcases.testcase_4.router_2_name", config_file)
77 SECGROUP_NAME = ft_utils.get_parameter_from_yaml(
78     "testcases.testcase_4.sdnvpn_sg_name", config_file)
79 SECGROUP_DESCR = ft_utils.get_parameter_from_yaml(
80     "testcases.testcase_4.sdnvpn_sg_descr", config_file)
81 TARGETS_1 = ft_utils.get_parameter_from_yaml(
82     "testcases.testcase_4.targets1", config_file)
83 TARGETS_2 = ft_utils.get_parameter_from_yaml(
84     "testcases.testcase_4.targets2", config_file)
85 SUCCESS_CRITERIA = ft_utils.get_parameter_from_yaml(
86     "testcases.testcase_4.succes_criteria", config_file)
87 TEST_DB = ft_utils.get_functest_config("results.test_db_url")
88
89 LINE_LENGTH = 60  # length for the summary table
90
91
92 def main():
93     global LINE_LENGTH
94
95     results = Results(LINE_LENGTH)
96
97     results.add_to_summary(0, "=")
98     results.add_to_summary(2, "STATUS", "SUBTEST")
99     results.add_to_summary(0, "=")
100
101     nova_client = os_utils.get_nova_client()
102     neutron_client = os_utils.get_neutron_client()
103     glance_client = os_utils.get_glance_client()
104
105     image_id = os_utils.create_glance_image(glance_client,
106                                             IMAGE_NAME,
107                                             IMAGE_PATH,
108                                             disk=IMAGE_FORMAT,
109                                             container="bare",
110                                             public=True)
111     network_1_id, _, router_1_id = test_utils.create_network(neutron_client,
112                                                              NET_1_NAME,
113                                                              SUBNET_1_NAME,
114                                                              SUBNET_1_CIDR,
115                                                              ROUTER_1_NAME)
116     network_2_id, _, router_2_id = test_utils.create_network(neutron_client,
117                                                              NET_2_NAME,
118                                                              SUBNET_2_NAME,
119                                                              SUBNET_2_CIDR,
120                                                              ROUTER_2_NAME)
121     sg_id = os_utils.create_security_group_full(neutron_client,
122                                                 SECGROUP_NAME, SECGROUP_DESCR)
123
124     # Get hypervisors zones
125     compute_nodes = os_utils.get_hypervisors(nova_client)
126     num_compute_nodes = len(compute_nodes)
127     if num_compute_nodes < 2:
128         logger.error("There are %s compute nodes in the deployment. "
129                      "Minimum number of nodes to complete the test is 2."
130                      % num_compute_nodes)
131         sys.exit(-1)
132
133     logger.debug("Compute nodes: %s" % compute_nodes)
134     av_zone_1 = "nova:" + compute_nodes[0]
135     av_zone_2 = "nova:" + compute_nodes[1]
136
137     # boot INTANCES
138     vm_2 = test_utils.create_instance(nova_client,
139                                       INSTANCE_2_NAME,
140                                       image_id,
141                                       network_1_id,
142                                       sg_id,
143                                       secgroup_name=SECGROUP_NAME,
144                                       compute_node=av_zone_1)
145     vm_2_ip = vm_2.networks.itervalues().next()[0]
146     logger.debug("Instance '%s' booted successfully. IP='%s'." %
147                  (INSTANCE_2_NAME, vm_2_ip))
148
149     vm_3 = test_utils.create_instance(nova_client,
150                                       INSTANCE_3_NAME,
151                                       image_id,
152                                       network_1_id,
153                                       sg_id,
154                                       secgroup_name=SECGROUP_NAME,
155                                       compute_node=av_zone_2)
156     vm_3_ip = vm_3.networks.itervalues().next()[0]
157     logger.debug("Instance '%s' booted successfully. IP='%s'." %
158                  (INSTANCE_3_NAME, vm_3_ip))
159
160     vm_5 = test_utils.create_instance(nova_client,
161                                       INSTANCE_5_NAME,
162                                       image_id,
163                                       network_2_id,
164                                       sg_id,
165                                       secgroup_name=SECGROUP_NAME,
166                                       compute_node=av_zone_2)
167     vm_5_ip = vm_5.networks.itervalues().next()[0]
168     logger.debug("Instance '%s' booted successfully. IP='%s'." %
169                  (INSTANCE_5_NAME, vm_5_ip))
170
171     # We boot vm5 first because we need vm5_ip for vm4 userdata
172     u4 = test_utils.generate_ping_userdata([vm_5_ip])
173     vm_4 = test_utils.create_instance(nova_client,
174                                       INSTANCE_4_NAME,
175                                       image_id,
176                                       network_2_id,
177                                       sg_id,
178                                       secgroup_name=SECGROUP_NAME,
179                                       compute_node=av_zone_1,
180                                       userdata=u4)
181     vm_4_ip = vm_4.networks.itervalues().next()[0]
182     logger.debug("Instance '%s' booted successfully. IP='%s'." %
183                  (INSTANCE_4_NAME, vm_4_ip))
184
185     # We boot VM1 at the end because we need to get the IPs first to generate
186     # the userdata
187     u1 = test_utils.generate_ping_userdata([vm_2_ip,
188                                             vm_3_ip,
189                                             vm_4_ip,
190                                             vm_5_ip])
191     vm_1 = test_utils.create_instance(nova_client,
192                                       INSTANCE_1_NAME,
193                                       image_id,
194                                       network_1_id,
195                                       sg_id,
196                                       secgroup_name=SECGROUP_NAME,
197                                       compute_node=av_zone_1,
198                                       userdata=u1)
199     vm_1_ip = vm_1.networks.itervalues().next()[0]
200     logger.debug("Instance '%s' booted successfully. IP='%s'." %
201                  (INSTANCE_1_NAME, vm_1_ip))
202     msg = ("Create VPN with eRT<>iRT")
203     logger.info(msg)
204     results.add_to_summary(1, msg)
205     vpn_name = "sdnvpn-" + str(randint(100000, 999999))
206     kwargs = {"import_targets": TARGETS_1,
207               "export_targets": TARGETS_2,
208               "name": vpn_name}
209     bgpvpn = os_utils.create_bgpvpn(neutron_client, **kwargs)
210     bgpvpn_id = bgpvpn['bgpvpn']['id']
211     logger.debug("VPN created details: %s" % bgpvpn)
212
213     msg = ("Associate router '%s' to the VPN." % ROUTER_1_NAME)
214     logger.info(msg)
215     results.add_to_summary(1, msg)
216     results.add_to_summary(0, "-")
217
218     os_utils.create_router_association(
219         neutron_client, bgpvpn_id, router_1_id)
220
221     # Wait for VMs to get ips.
222     instances_up = test_utils.wait_for_instances_up(vm_1, vm_2,
223                                                     vm_3, vm_4,
224                                                     vm_5)
225
226     if not instances_up:
227         logger.error("One or more instances is down")
228         # TODO Handle appropriately
229
230     # Ping from VM1 to VM2 should work
231     results.get_ping_status(vm_1, vm_1_ip, vm_2, vm_2_ip,
232                             expected="PASS", timeout=200)
233     # Ping from VM1 to VM3 should work
234     results.get_ping_status(vm_1, vm_1_ip, vm_3, vm_3_ip,
235                             expected="PASS", timeout=30)
236     # Ping from VM1 to VM4 should not work
237     results.get_ping_status(vm_1, vm_1_ip, vm_4, vm_4_ip,
238                             expected="FAIL", timeout=30)
239
240     msg = ("Associate network '%s' to the VPN." % NET_2_NAME)
241     logger.info(msg)
242     results.add_to_summary(0, "-")
243     results.add_to_summary(1, msg)
244     results.add_to_summary(0, "-")
245     os_utils.create_network_association(
246         neutron_client, bgpvpn_id, network_2_id)
247
248     test_utils.wait_for_bgp_router_assoc(
249         neutron_client, bgpvpn_id, router_1_id)
250     test_utils.wait_for_bgp_net_assoc(
251         neutron_client, bgpvpn_id, network_2_id)
252
253     logger.info("Waiting for the VMs to connect to each other using the"
254                 " updated network configuration")
255     time.sleep(30)
256
257     # Ping from VM4 to VM5 should work
258     results.get_ping_status(vm_4, vm_4_ip, vm_5, vm_5_ip,
259                             expected="PASS", timeout=30)
260     # Ping from VM1 to VM4 should not work
261     results.get_ping_status(vm_1, vm_1_ip, vm_4, vm_4_ip,
262                             expected="FAIL", timeout=30)
263     # Ping from VM1 to VM5 should not work
264     results.get_ping_status(vm_1, vm_1_ip, vm_5, vm_5_ip,
265                             expected="FAIL", timeout=30)
266
267     msg = ("Update VPN with eRT=iRT ...")
268     logger.info(msg)
269     results.add_to_summary(0, "-")
270     results.add_to_summary(1, msg)
271     results.add_to_summary(0, "-")
272     kwargs = {"import_targets": TARGETS_1,
273               "export_targets": TARGETS_1,
274               "name": vpn_name}
275     bgpvpn = os_utils.update_bgpvpn(neutron_client, bgpvpn_id, **kwargs)
276
277     logger.info("Waiting for the VMs to connect to each other using the"
278                 " updated network configuration")
279     time.sleep(30)
280
281     # Ping from VM1 to VM4 should work
282     results.get_ping_status(vm_1, vm_1_ip, vm_4, vm_4_ip,
283                             expected="PASS", timeout=30)
284     # Ping from VM1 to VM5 should work
285     results.get_ping_status(vm_1, vm_1_ip, vm_5, vm_5_ip,
286                             expected="PASS", timeout=30)
287
288     results.add_to_summary(0, "=")
289     logger.info("\n%s" % results.summary)
290
291     if results.test_result == "PASS":
292         logger.info("All the ping tests have passed as expected.")
293     else:
294         logger.info("One or more ping tests have failed.")
295
296     status = "PASS"
297     success = 100 - \
298         (100 * int(results.num_tests_failed) / int(results.num_tests_failed))
299     if success < int(SUCCESS_CRITERIA):
300         status = "FAILED"
301
302     return {"status": status, "details": results.details}
303
304
305 if __name__ == '__main__':
306     main()