summaryrefslogtreecommitdiffstats
path: root/sdnvpn/lib/config.py
blob: 8d87f334eba4ea4b656bcf1f5f7ccd98a09548f8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/usr/bin/python
#
# Copyright (c) 2017 All rights reserved
# This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
#
# http://www.apache.org/licenses/LICENSE-2.0

import yaml
import logging
import pkg_resources

from functest.utils.constants import CONST
import functest.utils.functest_utils as ft_utils

logger = logging.getLogger('sdnvpn_test_config')


class CommonConfig(object):
    """
    Common configuration parameters across testcases
    """
    class __CommonConfig:
        """
        Inner class used to make CommonConfig as singleton
        """

        def __init__(self):
            self.config_file = pkg_resources.resource_filename(
                'sdnvpn', 'test/functest/config.yaml')
            self.keyfile_path = pkg_resources.resource_filename(
                'sdnvpn', 'artifacts/id_rsa')
            self.test_db = CONST.results_test_db_url
            self.quagga_setup_script_path = pkg_resources.resource_filename(
                'sdnvpn', 'artifacts/quagga_setup.sh')
            self.line_length = 90  # length for the summary table
            self.vm_boot_timeout = 180
            self.default_flavor = ft_utils.get_parameter_from_yaml(
                "defaults.flavor", self.config_file)
            self.image_filename = CONST.openstack_image_file_name
            self.image_format = CONST.openstack_image_disk_format
            self.image_path = '{0}/{1}'.format(CONST.dir_functest_images,
                                               self.image_filename)
            # This is the ubuntu image used by sfc
            # Basically vanilla ubuntu + some scripts in there
            # We can use it to setup a quagga instance
            # TODO does functest have an ubuntu image somewhere?
            self.ubuntu_image_name = "sdnvpn-ubuntu"
            self.ubuntu_image_path = '{0}/{1}'.format(
                CONST.dir_functest_images,
                "ubuntu-16.04-server-cloudimg-amd64-disk1.img")
            self.custom_flavor_name = 'm1.custom'
            self.custom_flavor_ram = 1024
            self.custom_flavor_disk = 10
            self.custom_flavor_vcpus = 1
            self.neutron_nw_quota = -1
            self.neutron_subnet_quota = -1
            self.neutron_port_quota = -1
            self.neutron_router_quota = -1
            self.nova_instances_quota_class = -1

    commonCfgInstance = None

    def __init__(self):
        if not CommonConfig.commonCfgInstance:
            CommonConfig.commonCfgInstance = CommonConfig.__CommonConfig()

    def __getattr__(self, name):
        return getattr(self.commonCfgInstance, name)


class TestcaseConfig(object):
    """
    Configuration for a testcase.
    Parse config.yaml into a dict and create an object out of it.
    """

    def __init__(self, testcase):
        common_config = CommonConfig()
        test_config = None
        with open(common_config.config_file) as f:
            testcases_yaml = yaml.safe_load(f)
            test_config = testcases_yaml['testcases'].get(testcase, None)
        if test_config is None:
            logger.error('Test {0} configuration is not present in {1}'
                         .format(testcase, common_config.config_file))
        # Update class fields with configuration variables dynamically
        self.__dict__.update(**test_config)