Merge "Auto Generated INFO.yaml file"
[snaps.git] / snaps / openstack / tests / openstack_tests.py
index 78e8601..f3a1df7 100644 (file)
@@ -15,8 +15,8 @@
 import logging
 import re
 
-import pkg_resources
 from snaps import file_utils
+from snaps.config.flavor import FlavorConfig
 from snaps.config.image import ImageConfig
 from snaps.config.network import NetworkConfig, SubnetConfig
 from snaps.config.router import RouterConfig
@@ -139,6 +139,10 @@ def get_credentials(os_env_file=None, proxy_settings_str=None,
     if overrides and isinstance(overrides, dict):
         creds_dict.update(overrides)
 
+    for key, value in creds_dict.items():
+        if value is not None and isinstance(value, str):
+            creds_dict[key] = value.replace('"', '').replace('\'', '')
+
     os_creds = OSCreds(**creds_dict)
     logger.info('OS Credentials = %s', os_creds.__str__)
     return os_creds
@@ -174,17 +178,15 @@ def create_image_settings(image_name, image_user, image_format, metadata,
         return ImageConfig(**metadata['config'])
 
     disk_file = None
-    if metadata:
+    if metadata and ('disk_url' in metadata or 'disk_file' in metadata):
         disk_url = metadata.get('disk_url')
         disk_file = metadata.get('disk_file')
     elif not disk_url:
         disk_url = default_url
-    else:
-        disk_url = disk_url
 
-    if metadata and \
-            ('kernel_file' in metadata or 'kernel_url' in metadata) and \
-            kernel_settings is None:
+    if (metadata
+            and ('kernel_file' in metadata or 'kernel_url' in metadata)
+            and kernel_settings is None):
         kernel_image_settings = ImageConfig(
             name=image_name + '-kernel', image_user=image_user,
             img_format=image_format, image_file=metadata.get('kernel_file'),
@@ -192,9 +194,9 @@ def create_image_settings(image_name, image_user, image_format, metadata,
     else:
         kernel_image_settings = kernel_settings
 
-    if metadata and \
-            ('ramdisk_file' in metadata or 'ramdisk_url' in metadata) and \
-            ramdisk_settings is None:
+    if (metadata
+            and ('ramdisk_file' in metadata or 'ramdisk_url' in metadata)
+            and ramdisk_settings is None):
         ramdisk_image_settings = ImageConfig(
             name=image_name + '-ramdisk', image_user=image_user,
             img_format=image_format,
@@ -273,15 +275,12 @@ def centos_image_settings(name, url=None, image_metadata=None,
     else:
         metadata = image_metadata
 
-    pb_path = pkg_resources.resource_filename(
-        'snaps.provisioning.ansible_pb.centos-network-setup.playbooks',
-        'configure_host.yml')
     return create_image_settings(
         image_name=name, image_user=CENTOS_USER,
         image_format=DEFAULT_IMAGE_FORMAT, metadata=metadata, disk_url=url,
         default_url=CENTOS_DEFAULT_IMAGE_URL,
         kernel_settings=kernel_settings, ramdisk_settings=ramdisk_settings,
-        public=public, nic_config_pb_loc=pb_path)
+        public=public)
 
 
 def ubuntu_image_settings(name, url=None, image_metadata=None,
@@ -306,49 +305,126 @@ def ubuntu_image_settings(name, url=None, image_metadata=None,
     else:
         metadata = image_metadata
 
-    pb_path = pkg_resources.resource_filename(
-        'snaps.provisioning.ansible_pb.ubuntu-network-setup.playbooks',
-        'configure_host.yml')
     return create_image_settings(
         image_name=name, image_user=UBUNTU_USER,
         image_format=DEFAULT_IMAGE_FORMAT, metadata=metadata, disk_url=url,
         default_url=UBUNTU_DEFAULT_IMAGE_URL,
         kernel_settings=kernel_settings, ramdisk_settings=ramdisk_settings,
-        public=public, nic_config_pb_loc=pb_path)
-
+        public=public)
 
-def get_priv_net_config(net_name, subnet_name, router_name=None,
-                        cidr='10.55.0.0/24', external_net=None):
-    return OSNetworkConfig(net_name, subnet_name, cidr, router_name,
-                           external_gateway=external_net)
 
+def get_priv_net_config(project_name, net_name, mtu=None, subnet_name=None,
+                        router_name=None, cidr='10.55.0.0/24',
+                        external_net=None, netconf_override=None):
+    return OSNetworkConfig(
+        project_name, net_name, mtu, subnet_name, cidr, router_name,
+        external_gateway=external_net, netconf_override=netconf_override)
+
+
+def get_pub_net_config(
+        project_name, net_name, mtu=None, subnet_name=None, router_name=None,
+        cidr='10.55.1.0/24', external_net=None, netconf_override=None):
+    return OSNetworkConfig(project_name, net_name, mtu, subnet_name, cidr,
+                           router_name, external_gateway=external_net,
+                           netconf_override=netconf_override)
+
+
+def get_flavor_config(name, ram, disk, vcpus, ephemeral=None, swap=None,
+                      rxtx_factor=None, is_public=None, metadata=None):
+    """This method replaces the hard coded basic element (e.g. ram, vcpu, disk
+     etc) with those are included in the new freeform dict() of metadata
+     parameter.
+
+    :param name: the flavor name (required)
+    :param ram: memory in MB to allocate to VM (required)
+    :param disk: disk storage in GB (required)
+    :param vcpus: the number of CPUs to allocate to VM (required)
+    :param ephemeral: the size of the ephemeral disk in GB (default=0)
+    :param swap: the size of the swap disk in GB (default=0)
+    :param rxtx_factor: the receive/transmit factor to be set on ports
+         if backend supports QoS extension (default=1.0)
+    :param is_public: flag that denotes whether or not other projects
+         can access image (default=True)
+    :param metadata: - freeform dict() for special metadata (optional)
+                     - freeform dict() for values of basic elements
+                     (e.g. ram, vcpu, disk, etc) could be added.
+                     As result the hard coded values of those elements will be
+                     overwritten by the new ones (optional)
+    :return: The FlavorConfig replacing the hard coded basic element values
+            (e.g. ram, vcpu, disk etc) with those are included in the metadata
+             dict [optional]. The metadata parameter in the FlavorConfig
+             consist of the metadata data only.
+    """
 
-def get_pub_net_config(net_name, subnet_name=None, router_name=None,
-                       cidr='10.55.1.0/24', external_net=None):
-    return OSNetworkConfig(net_name, subnet_name, cidr, router_name,
-                           external_gateway=external_net)
+    metadata_excl = metadata
+    if metadata:
+        if 'ram' in metadata:
+            ram = metadata['ram']
+            del metadata_excl['ram']
+        if 'disk' in metadata:
+            disk = metadata['disk']
+            del metadata_excl['disk']
+        if 'vcpus' in metadata:
+            vcpus = metadata['vcpus']
+            del metadata_excl['vcpus']
+        if 'ephemeral' in metadata:
+            ephemeral = metadata['ephemeral']
+            del metadata_excl['ephemeral']
+        if 'swap' in metadata:
+            swap = metadata['swap']
+            del metadata_excl['swap']
+        if 'rxtx_factor' in metadata:
+            rxtx_factor = metadata['rxtx_factor']
+            del metadata_excl['rxtx_factor']
+        if 'is_public' in metadata:
+            is_public = metadata['is_public']
+            del metadata_excl['is_public']
+        if 'metadata' in metadata:
+            metadata_excl = metadata['metadata']
+
+    return FlavorConfig(
+        name=name, ram=ram, disk=disk, vcpus=vcpus, ephemeral=ephemeral,
+        swap=swap, rxtx_factor=rxtx_factor, is_public=is_public,
+        metadata=metadata_excl)
 
 
 class OSNetworkConfig:
     """
     Represents the settings required for the creation of a network in OpenStack
+    where netconf_override is used to reconfigure the network_type,
+    physical_network and segmentation_id
     """
 
-    def __init__(self, net_name, subnet_name=None, subnet_cidr=None,
-                 router_name=None, external_gateway=None):
-
+    def __init__(self, project_name, net_name, mtu=None, subnet_name=None,
+                 subnet_cidr=None, router_name=None, external_gateway=None,
+                 netconf_override=None):
+        """
+        :param netconf_override: dict() containing the reconfigured
+                                 network_type, physical_network and
+                                 segmentation_id
+        """
         if subnet_name and subnet_cidr:
-            self.network_settings = NetworkConfig(
-                name=net_name, subnet_settings=[
+            network_conf = NetworkConfig(
+                name=net_name, mtu=mtu, subnet_settings=[
                     SubnetConfig(cidr=subnet_cidr, name=subnet_name)])
         else:
-            self.network_settings = NetworkConfig(name=net_name)
+            network_conf = NetworkConfig(name=net_name)
+        if netconf_override:
+            network_conf.network_type = netconf_override.get('network_type')
+            network_conf.physical_network = netconf_override.get(
+                'physical_network')
+            network_conf.segmentation_id = netconf_override.get(
+                'segmentation_id')
+        self.network_settings = network_conf
 
         if router_name:
             if subnet_name:
                 self.router_settings = RouterConfig(
                     name=router_name, external_gateway=external_gateway,
-                    internal_subnets=[subnet_name])
+                    internal_subnets=[{'subnet': {
+                        'project_name': project_name,
+                        'network_name': net_name,
+                        'subnet_name': subnet_name}}])
             else:
                 self.router_settings = RouterConfig(
                     name=router_name, external_gateway=external_gateway)