summaryrefslogtreecommitdiffstats
path: root/sfc/lib/config.py
blob: 49cebc7ddb5cd388c20f5fa1b80d7c4669410b5c (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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/python
#
# Copyright (c) 2016 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 logging
import os
import yaml
import functest

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

import sfc
import sfc.lib.test_utils as test_utils


logger = logging.getLogger(__name__)


class CommonConfig(object):
    """
    Common configuration parameters across testcases
    """

    def __init__(self):
        self.line_length = 30
        self.test_db = os.environ['TEST_DB_URL']
        self.functest_repo_path = os.path.dirname(functest.__file__)
        self.functest_logging_api = os.path.join(self.functest_repo_path,
                                                 "ci", "logging.ini")
        self.sfc_repo_path = os.path.dirname(sfc.__file__)
        self.sfc_test_dir = os.path.join(
            self.sfc_repo_path, "tests", "functest")
        self.vnfd_dir = os.path.join(self.sfc_test_dir, "vnfd-templates")
        self.vnfd_default_params_file = os.path.join(
            self.sfc_test_dir, "vnfd-default-params-file")
        self.vnffgd_dir = os.path.join(self.sfc_test_dir, "vnffgd-templates")
        self.functest_results_dir = os.path.join(
            getattr(CONST, 'dir_results'), "odl-sfc")
        self.config_file = os.path.join(self.sfc_test_dir, "config.yaml")
        self.vim_file = os.path.join(self.sfc_test_dir, "register-vim.json")

        self.installer_type = env.get('INSTALLER_TYPE')

        self.installer_fields = test_utils.fill_installer_dict(
            self.installer_type)

        self.installer_ip = env.get('INSTALLER_IP')

        self.installer_user = ft_utils.get_parameter_from_yaml(
            self.installer_fields['user'], self.config_file)

        try:
            self.installer_password = ft_utils.get_parameter_from_yaml(
                self.installer_fields['password'], self.config_file)
        except Exception:
            self.installer_password = None

        try:
            self.installer_key_file = ft_utils.get_parameter_from_yaml(
                self.installer_fields['pkey_file'], self.config_file)
        except Exception:
            self.installer_key_file = None

        try:
            self.installer_cluster = ft_utils.get_parameter_from_yaml(
                self.installer_fields['cluster'], self.config_file)
        except Exception:
            self.installer_cluster = None

        self.flavor = ft_utils.get_parameter_from_yaml(
            "defaults.flavor", self.config_file)
        self.ram_size_in_mb = ft_utils.get_parameter_from_yaml(
            "defaults.ram_size_in_mb", self.config_file)
        self.disk_size_in_gb = ft_utils.get_parameter_from_yaml(
            "defaults.disk_size_in_gb", self.config_file)
        self.vcpu_count = ft_utils.get_parameter_from_yaml(
            "defaults.vcpu_count", self.config_file)
        self.image_name = ft_utils.get_parameter_from_yaml(
            "defaults.image_name", self.config_file)
        self.image_format = ft_utils.get_parameter_from_yaml(
            "defaults.image_format", self.config_file)
        self.image_url = ft_utils.get_parameter_from_yaml(
            "defaults.image_url", self.config_file)
        self.dir_functest_data = ft_utils.get_functest_config(
            "general.dir.functest_data")


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 %s configuration is not present in %s',
                testcase, common_config.config_file)
        # Update class fields with configuration variables dynamically
        self.__dict__.update(**test_config)