Fix Pep8 issues related to \
[functest.git] / functest / opnfv_tests / openstack / examples / create_instance_and_ip.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 # This script boots an instance and assigns a floating ip
11 #
12
13 import argparse
14 import os
15 import sys
16
17 import functest.utils.functest_logger as ft_logger
18 import functest.utils.openstack_utils as os_utils
19 import functest.utils.functest_constants as ft_constants
20
21 parser = argparse.ArgumentParser()
22
23 parser.add_argument("-r", "--report",
24                     help="Create json result file",
25                     action="store_true")
26
27 args = parser.parse_args()
28
29 """ logging configuration """
30 logger = ft_logger.Logger("create_instance_and_ip").getLogger()
31
32 HOME = ft_constants.HOME + "/"
33
34 VM_BOOT_TIMEOUT = 180
35
36 EXAMPLE_INSTANCE_NAME = ft_constants.EXAMPLE_INSTANCE_NAME
37 EXAMPLE_FLAVOR = ft_constants.EXAMPLE_FLAVOR
38 EXAMPLE_IMAGE_NAME = ft_constants.EXAMPLE_IMAGE_NAME
39 IMAGE_FILENAME = ft_constants.GLANCE_IMAGE_FILENAME
40 IMAGE_FORMAT = ft_constants.GLANCE_IMAGE_FORMAT
41 IMAGE_PATH = os.path.join(ft_constants.FUNCTEST_DATA_DIR, IMAGE_FILENAME)
42
43 # NEUTRON Private Network parameters
44
45 EXAMPLE_PRIVATE_NET_NAME = ft_constants.EXAMPLE_PRIVATE_NET_NAME
46 EXAMPLE_PRIVATE_SUBNET_NAME = ft_constants.EXAMPLE_PRIVATE_SUBNET_NAME
47 EXAMPLE_PRIVATE_SUBNET_CIDR = ft_constants.EXAMPLE_PRIVATE_SUBNET_CIDR
48 EXAMPLE_ROUTER_NAME = ft_constants.EXAMPLE_ROUTER_NAME
49
50 EXAMPLE_SECGROUP_NAME = ft_constants.EXAMPLE_SECGROUP_NAME
51 EXAMPLE_SECGROUP_DESCR = ft_constants.EXAMPLE_SECGROUP_DESCR
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     image_id = os_utils.create_glance_image(glance_client,
61                                             EXAMPLE_IMAGE_NAME,
62                                             IMAGE_PATH,
63                                             disk=IMAGE_FORMAT,
64                                             container="bare",
65                                             public=True)
66
67     network_dic = os_utils.create_network_full(neutron_client,
68                                                EXAMPLE_PRIVATE_NET_NAME,
69                                                EXAMPLE_PRIVATE_SUBNET_NAME,
70                                                EXAMPLE_ROUTER_NAME,
71                                                EXAMPLE_PRIVATE_SUBNET_CIDR)
72     if not network_dic:
73         logger.error(
74             "There has been a problem when creating the neutron network")
75         sys.exit(-1)
76
77     network_id = network_dic["net_id"]
78
79     sg_id = os_utils.create_security_group_full(neutron_client,
80                                                 EXAMPLE_SECGROUP_NAME,
81                                                 EXAMPLE_SECGROUP_DESCR)
82
83     # boot INTANCE
84     logger.info("Creating instance '%s'..." % EXAMPLE_INSTANCE_NAME)
85     logger.debug(
86         "Configuration:\n name=%s \n flavor=%s \n image=%s \n "
87         "network=%s \n"
88         % (EXAMPLE_INSTANCE_NAME, EXAMPLE_FLAVOR, image_id, network_id))
89     instance = \
90         os_utils.create_instance_and_wait_for_active(EXAMPLE_FLAVOR,
91                                                      image_id,
92                                                      network_id,
93                                                      EXAMPLE_INSTANCE_NAME)
94
95     if instance is None:
96         logger.error("Error while booting instance.")
97         sys.exit(-1)
98     # Retrieve IP of INSTANCE
99     instance_ip = instance.networks.get(EXAMPLE_PRIVATE_NET_NAME)[0]
100     logger.debug("Instance '%s' got private ip '%s'." %
101                  (EXAMPLE_INSTANCE_NAME, instance_ip))
102
103     logger.info("Adding '%s' to security group '%s'..."
104                 % (EXAMPLE_INSTANCE_NAME, EXAMPLE_SECGROUP_NAME))
105     os_utils.add_secgroup_to_instance(nova_client, instance.id, sg_id)
106
107     logger.info("Creating floating IP for VM '%s'..." % EXAMPLE_INSTANCE_NAME)
108     floatip_dic = os_utils.create_floating_ip(neutron_client)
109     floatip = floatip_dic['fip_addr']
110     # floatip_id = floatip_dic['fip_id']
111
112     if floatip is None:
113         logger.error("Cannot create floating IP.")
114         sys.exit(-1)
115     logger.info("Floating IP created: '%s'" % floatip)
116
117     logger.info("Associating floating ip: '%s' to VM '%s' "
118                 % (floatip, EXAMPLE_INSTANCE_NAME))
119     if not os_utils.add_floating_ip(nova_client, instance.id, floatip):
120         logger.error("Cannot associate floating IP to VM.")
121         sys.exit(-1)
122
123     sys.exit(0)
124
125
126 if __name__ == '__main__':
127     main()