99aac66b19469f9d079a7d0c1556e389e618d02e
[functest.git] / testcases / features / sfc / sfc.py
1 import argparse
2 import os
3 import subprocess
4 import sys
5 import time
6 import functest.utils.functest_logger as ft_logger
7 import functest.utils.functest_utils as ft_utils
8 import functest.utils.openstack_utils as os_utils
9 import paramiko
10
11
12 parser = argparse.ArgumentParser()
13
14 parser.add_argument("-r", "--report",
15                     help="Create json result file",
16                     action="store_true")
17
18 args = parser.parse_args()
19
20 """ logging configuration """
21 logger = ft_logger.Logger("create_instance_and_ip").getLogger()
22
23 REPO_PATH = os.environ['repos_dir'] + '/functest/'
24 HOME = os.environ['HOME'] + "/"
25
26 VM_BOOT_TIMEOUT = 180
27 INSTANCE_NAME = "client"
28 FLAVOR = "m1.small"
29 IMAGE_NAME = "sf_summit2016"
30 IMAGE_FILENAME = "sf_summit2016.qcow2"
31 IMAGE_FORMAT = "qcow2"
32 IMAGE_PATH = "/home/opnfv/functest/data" + "/" + IMAGE_FILENAME
33
34 # NEUTRON Private Network parameters
35
36 NET_NAME = "example-net"
37 SUBNET_NAME = "example-subnet"
38 SUBNET_CIDR = "11.0.0.0/24"
39 ROUTER_NAME = "example-router"
40
41 SECGROUP_NAME = "example-sg"
42 SECGROUP_DESCR = "Example Security group"
43
44 INSTANCE_NAME_2 = "server"
45
46 # TEST_DB = ft_utils.get_parameter_from_yaml("results.test_db_url")
47
48 PRE_SETUP_SCRIPT = 'sfc_pre_setup.bash'
49 TACKER_SCRIPT = 'sfc_tacker.bash'
50 TEARDOWN_SCRIPT = "sfc_teardown.bash"
51 TACKER_CHANGECLASSI = "sfc_change_classi.bash"
52
53
54 def main():
55
56     nova_client = os_utils.get_nova_client()
57     neutron_client = os_utils.get_neutron_client()
58     glance_client = os_utils.get_glance_client()
59
60 # Download the image
61
62     if not os.path.isfile(IMAGE_PATH):
63         logger.info("Downloading image")
64         ft_utils.download_url(
65             "http://artifacts.opnfv.org/sfc/demo/sf_summit2016.qcow2",
66             "/home/opnfv/functest/data/")
67     else:
68         logger.info("Using old image")
69
70 # Allow any port so that tacker commands reaches the server.
71 # CHECK IF THIS STILL MAKES SENSE WHEN TACKER IS INCLUDED IN OPNFV INSTALATION
72
73     controller_command = ("sshpass -p r00tme ssh root@10.20.0.2"
74                           " 'fuel node'|grep controller|awk '{print $10}'")
75     logger.info("Executing tacker script: '%s'" % controller_command)
76     process = subprocess.Popen(controller_command,
77                                shell=True,
78                                stdout=subprocess.PIPE)
79     ip = process.stdout.readline()
80
81     iptable_command1 = ("sshpass -p r00tme ssh root@10.20.0.2 ssh"
82                         " " + ip + " iptables -P INPUT ACCEPT ")
83     iptable_command2 = ("sshpass -p r00tme ssh root@10.20.0.2 ssh"
84                         " " + ip + "iptables -t nat -P INPUT ACCEPT ")
85
86     subprocess.call(iptable_command1, shell=True)
87     subprocess.call(iptable_command2, shell=True)
88
89 # Create glance image and the neutron network
90
91     image_id = os_utils.create_glance_image(glance_client,
92                                             IMAGE_NAME,
93                                             IMAGE_PATH,
94                                             disk=IMAGE_FORMAT,
95                                             container="bare",
96                                             public=True)
97
98     network_dic = os_utils.create_network_full(neutron_client,
99                                                NET_NAME,
100                                                SUBNET_NAME,
101                                                ROUTER_NAME,
102                                                SUBNET_CIDR)
103     if not network_dic:
104         logger.error(
105             "There has been a problem when creating the neutron network")
106         sys.exit(-1)
107
108     network_id = network_dic["net_id"]
109
110     sg_id = os_utils.create_security_group_full(neutron_client,
111                                                 SECGROUP_NAME, SECGROUP_DESCR)
112
113     # boot INTANCE
114     logger.info("Creating instance '%s'..." % INSTANCE_NAME)
115     logger.debug(
116         "Configuration:\n name=%s \n flavor=%s \n image=%s \n "
117         "network=%s \n" % (INSTANCE_NAME, FLAVOR, image_id, network_id))
118     instance = os_utils.create_instance_and_wait_for_active(FLAVOR,
119                                                             image_id,
120                                                             network_id,
121                                                             INSTANCE_NAME)
122
123     if instance is None:
124         logger.error("Error while booting instance.")
125         sys.exit(-1)
126     # Retrieve IP of INSTANCE
127     instance_ip = instance.networks.get(NET_NAME)[0]
128     logger.debug("Instance '%s' got private ip '%s'." %
129                  (INSTANCE_NAME, instance_ip))
130
131     logger.info("Adding '%s' to security group '%s'..."
132                 % (INSTANCE_NAME, SECGROUP_NAME))
133     os_utils.add_secgroup_to_instance(nova_client, instance.id, sg_id)
134
135     logger.info("Creating floating IP for VM '%s'..." % INSTANCE_NAME)
136     floatip_dic = os_utils.create_floating_ip(neutron_client)
137     floatip_client = floatip_dic['fip_addr']
138     # floatip_id = floatip_dic['fip_id']
139
140     if floatip_client is None:
141         logger.error("Cannot create floating IP.")
142         sys.exit(-1)
143     logger.info("Floating IP created: '%s'" % floatip_client)
144
145     logger.info("Associating floating ip: '%s' to VM '%s' "
146                 % (floatip_client, INSTANCE_NAME))
147     if not os_utils.add_floating_ip(nova_client, instance.id, floatip_client):
148         logger.error("Cannot associate floating IP to VM.")
149         sys.exit(-1)
150
151 # STARTING SECOND VM (server) ###
152
153     # boot INTANCE
154     logger.info("Creating instance '%s'..." % INSTANCE_NAME)
155     logger.debug(
156         "Configuration:\n name=%s \n flavor=%s \n image=%s \n "
157         "network=%s \n" % (INSTANCE_NAME, FLAVOR, image_id, network_id))
158     instance_2 = os_utils.create_instance_and_wait_for_active(FLAVOR,
159                                                               image_id,
160                                                               network_id,
161                                                               INSTANCE_NAME_2)
162
163     if instance_2 is None:
164         logger.error("Error while booting instance.")
165         sys.exit(-1)
166     # Retrieve IP of INSTANCE
167     instance_ip_2 = instance_2.networks.get(NET_NAME)[0]
168     logger.debug("Instance '%s' got private ip '%s'." %
169                  (INSTANCE_NAME_2, instance_ip_2))
170
171     logger.info("Adding '%s' to security group '%s'..."
172                 % (INSTANCE_NAME_2, SECGROUP_NAME))
173     os_utils.add_secgroup_to_instance(nova_client, instance_2.id, sg_id)
174
175     logger.info("Creating floating IP for VM '%s'..." % INSTANCE_NAME_2)
176     floatip_dic = os_utils.create_floating_ip(neutron_client)
177     floatip_server = floatip_dic['fip_addr']
178     # floatip_id = floatip_dic['fip_id']
179
180     if floatip_server is None:
181         logger.error("Cannot create floating IP.")
182         sys.exit(-1)
183     logger.info("Floating IP created: '%s'" % floatip_server)
184
185     logger.info("Associating floating ip: '%s' to VM '%s' "
186                 % (floatip_server, INSTANCE_NAME_2))
187
188     if not os_utils.add_floating_ip(nova_client,
189                                     instance_2.id,
190                                     floatip_server):
191         logger.error("Cannot associate floating IP to VM.")
192         sys.exit(-1)
193
194     # CREATION OF THE 2 SF ####
195
196     tacker_script = "/home/opnfv/repos/functest/testcases/features/sfc/" + \
197         TACKER_SCRIPT
198     logger.info("Executing tacker script: '%s'" % tacker_script)
199     subprocess.call(tacker_script, shell=True)
200
201     # SSH CALL TO START HTTP SERVER
202     ssh = paramiko.SSHClient()
203     ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
204
205     try:
206         ssh.connect(floatip_server, username="root",
207                     password="opnfv", timeout=2)
208         command = "python -m SimpleHTTPServer 80 > /dev/null 2>&1 &"
209         logger.info("Starting HTTP server")
210         (stdin, stdout, stderr) = ssh.exec_command(command)
211     except:
212         logger.debug("Waiting for %s..." % floatip_server)
213         time.sleep(6)
214         # timeout -= 1
215
216     instances = nova_client.servers.list(search_opts={'all_tenants': 1})
217     ips = []
218     try:
219         for instance in instances:
220             if "server" not in instance.name:
221                 if "client" not in instance.name:
222                     logger.debug(
223                         "This is the instance name: %s " % instance.name)
224                     floatip_dic = os_utils.create_floating_ip(neutron_client)
225                     floatip = floatip_dic['fip_addr']
226                     ips.append(floatip)
227                     instance.add_floating_ip(floatip)
228     except:
229         logger.debug("Problems assigning floating IP to SFs")
230
231     logger.debug("Floating IPs for SFs: %s..." % ips)
232     # SSH TO START THE VXLAN_TOOL ON SF1
233     logger.info("Configuring the SFs")
234     try:
235         ssh.connect(ips[0], username="root",
236                     password="opnfv", timeout=2)
237         command = ("nohup python vxlan_tool.py -i eth0 "
238                    "-d forward -v off -f -b 80 &")
239         (stdin, stdout, stderr) = ssh.exec_command(command)
240     except:
241         logger.debug("Waiting for %s..." % ips[0])
242         time.sleep(6)
243         # timeout -= 1
244
245     # SSH TO START THE VXLAN_TOOL ON SF2
246     try:
247         ssh.connect(ips[1], username="root",
248                     password="opnfv", timeout=2)
249         command = ("nohup python vxlan_tool.py -i eth0 "
250                    "-d forward -v off -f -b 22 &")
251         (stdin, stdout, stderr) = ssh.exec_command(command)
252     except:
253         logger.debug("Waiting for %s..." % ips[1])
254         time.sleep(6)
255         # timeout -= 1
256
257     # SSH TO EXECUTE cmd_client
258
259     logger.info("TEST STARTED")
260     try:
261         ssh.connect(floatip_client, username="root",
262                     password="opnfv", timeout=2)
263         command = "nc -w 5 -zv " + floatip_server + " 22 2>&1"
264         (stdin, stdout, stderr) = ssh.exec_command(command)
265     except:
266         logger.debug("Waiting for %s..." % floatip_client)
267         time.sleep(6)
268         # timeout -= 1
269
270     # WRITE THE CORRECT WAY TO DO LOGGING
271     i = 0
272     logger.info("First output: %s" % stdout.readlines())
273     if "timed out" in stdout.readlines()[0]:
274         logger.info('\033[92m' + "TEST 1 [PASSED] "
275                     "==> SSH BLOCKED" + '\033[0m')
276         i = i + 1
277     else:
278         logger.debug('\033[91m' + "TEST 1 [FAILED] "
279                      "==> SSH NOT BLOCKED" + '\033[0m')
280         return
281
282     # SSH TO EXECUTE cmd_client
283
284     try:
285         ssh.connect(floatip_client, username="root",
286                     password="opnfv", timeout=2)
287         command = "nc -w 5 -zv " + floatip_server + " 80 2>&1"
288         (stdin, stdout, stderr) = ssh.exec_command(command)
289     except:
290         logger.debug("Waiting for %s..." % floatip_client)
291         time.sleep(6)
292         # timeout -= 1
293
294     if "succeeded" in stdout.readlines()[0]:
295         logger.info('\033[92m' + "TEST 2 [PASSED] "
296                     "==> HTTP WORKS" + '\033[0m')
297         i = i + 1
298     else:
299         logger.debug('\033[91m' + "TEST 2 [FAILED] "
300                      "==> HTTP BLOCKED" + '\033[0m')
301         return
302
303     # CHANGE OF CLASSIFICATION #
304     logger.info("Changing the classification")
305     tacker_classi = "/home/opnfv/repos/functest/testcases/features/sfc/" + \
306         TACKER_CHANGECLASSI
307     subprocess.call(tacker_classi, shell=True)
308
309     # SSH TO EXECUTE cmd_client
310
311     try:
312         ssh.connect(floatip_client, username="root",
313                     password="opnfv", timeout=2)
314         command = "nc -w 5 -zv " + floatip_server + " 80 2>&1"
315         (stdin, stdout, stderr) = ssh.exec_command(command)
316     except:
317         logger.debug("Waiting for %s..." % floatip_client)
318         time.sleep(6)
319         # timeout -= 1
320
321     if "timed out" in stdout.readlines()[0]:
322         logger.info('\033[92m' + "TEST 3 [WORKS] "
323                     "==> HTTP BLOCKED" + '\033[0m')
324         i = i + 1
325     else:
326         logger.debug('\033[91m' + "TEST 3 [FAILED] "
327                      "==> HTTP NOT BLOCKED" + '\033[0m')
328         return
329
330     # SSH TO EXECUTE cmd_client
331
332     try:
333         ssh.connect(floatip_client, username="root",
334                     password="opnfv", timeout=2)
335         command = "nc -w 5 -zv " + floatip_server + " 22 2>&1"
336         (stdin, stdout, stderr) = ssh.exec_command(command)
337     except:
338         logger.debug("Waiting for %s..." % floatip_client)
339         time.sleep(6)
340         # timeout -= 1
341
342     if "succeeded" in stdout.readlines()[0]:
343         logger.info('\033[92m' + "TEST 4 [WORKS] "
344                     "==> SSH WORKS" + '\033[0m')
345         i = i + 1
346     else:
347         logger.debug('\033[91m' + "TEST 4 [FAILED] "
348                      "==> SSH BLOCKED" + '\033[0m')
349         return
350
351     if i == 4:
352         for x in range(0, 5):
353             logger.info('\033[92m' + "SFC TEST WORKED"
354                         " :) \n" + '\033[0m')
355
356     sys.exit(0)
357
358 if __name__ == '__main__':
359     main()