Exemplo n.º 1
0
 def test_os_env_no_value(self):
     open_mock = mock.mock_open(
         read_data='env_value: !os_env MYVAR\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     self.os_mock.environ = {}
     with self.assertRaises(error.DevopsError):
         templates.yaml_template_load('/path/to/my.yaml')
    def test_include_exception(self):
        mock_open1 = mock.mock_open(read_data='myinclude: !include file2.yaml')
        mock_open1.return_value.name = '/path/to/my.yaml'
        mock_open2 = mock.mock_open(read_data='10')
        mock_open2.return_value.name = '/path/to/file2.yaml'
        open_mock = self.patch('devops.helpers.templates.open',
                               new_callable=mock.mock_open,
                               create=True)
        open_mock.side_effect = (mock_open1.return_value,
                                 mock_open2.return_value)

        with self.assertRaises(error.DevopsError):
            templates.yaml_template_load('/path/to/my.yaml')
Exemplo n.º 3
0
    def test_include_exception(self):
        mock_open1 = mock.mock_open(read_data='myinclude: !include file2.yaml')
        mock_open1.return_value.name = '/path/to/my.yaml'
        mock_open2 = mock.mock_open(read_data='10')
        mock_open2.return_value.name = '/path/to/file2.yaml'
        open_mock = self.patch(
            'devops.helpers.templates.open',
            new_callable=mock.mock_open, create=True)
        open_mock.side_effect = (
            mock_open1.return_value, mock_open2.return_value)

        with self.assertRaises(error.DevopsError):
            templates.yaml_template_load('/path/to/my.yaml')
Exemplo n.º 4
0
 def test_os_env(self):
     open_mock = mock.mock_open(
         read_data='env_value: !os_env MYVAR\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     self.os_mock.environ = {'MYVAR': '30'}
     m = templates.yaml_template_load('/path/to/my.yaml')
     assert m['env_value'] == 30
Exemplo n.º 5
0
 def env(self):
     if self.__env is None:
         # hack before we find proper solution for loading devops templates
         external_config = os.environ.get("DEVOPS_SETTINGS_TEMPLATE", None)
         if external_config:
             config = yaml_template_load(external_config)
         else:
             config = self._devops_config
         self.__env = EnvironmentModel(config)
     return self.__env
Exemplo n.º 6
0
    def __init__(self):
        super(UpgradeCustom, self).__init__()
        data = yaml_template_load(
            settings.UPGRADE_TEST_TEMPLATE)['upgrade_data']
        self.upgrade_data = data

        logger.debug("Got following data from upgrade template:")
        for step in data:
            logger.debug("\n".join(
                ["{}:{}".format(key, value) for key, value in step.items()]))
Exemplo n.º 7
0
    def __init__(self):
        super(UpgradeCustom, self).__init__()
        data = yaml_template_load(
            settings.UPGRADE_TEST_TEMPLATE)['upgrade_data']
        self.upgrade_data = data

        logger.debug("Got following data from upgrade template:")
        for step in data:
            logger.debug("\n".join(
                ["{}:{}".format(key, value) for key, value in step.items()]))
Exemplo n.º 8
0
    def load_template(self, filename):
        """Method for reading file with devops config

        :param filename: string
        """
        if filename is not None:
            LOG.debug("Preparing to load config from template '{0}'".format(
                filename))
            self.config = templates.yaml_template_load(filename)
        else:
            LOG.error("Template filename is not set, loading config " +
                      "from template aborted.")
 def test_mapping_order(self):
     open_mock = mock.mock_open(read_data='mapping:\n'
                                '    one: 1\n'
                                '    two: 2\n'
                                '    three: 3\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     m = templates.yaml_template_load('/path/to/my.yaml')
     assert isinstance(m['mapping'], collections.OrderedDict)
     assert m['mapping'] == collections.OrderedDict([
         ('one', 1),
         ('two', 2),
         ('three', 3),
     ])
Exemplo n.º 10
0
 def test_mapping_order(self):
     open_mock = mock.mock_open(
         read_data='mapping:\n'
                   '    one: 1\n'
                   '    two: 2\n'
                   '    three: 3\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     m = templates.yaml_template_load('/path/to/my.yaml')
     assert isinstance(m['mapping'], collections.OrderedDict)
     assert m['mapping'] == collections.OrderedDict([
         ('one', 1),
         ('two', 2),
         ('three', 3),
     ])
    def test_include(self):
        self.isfiles.append('/path/to/file2.yaml')
        mock_open1 = mock.mock_open(read_data='myinclude: !include file2.yaml')
        mock_open1.return_value.name = '/path/to/my.yaml'
        mock_open2 = mock.mock_open(read_data='10')
        mock_open2.return_value.name = '/path/to/file2.yaml'
        open_mock = self.patch('devops.helpers.templates.open',
                               new_callable=mock.mock_open,
                               create=True)
        open_mock.side_effect = (mock_open1.return_value,
                                 mock_open2.return_value)

        m = templates.yaml_template_load('/path/to/my.yaml')
        assert m == {'myinclude': 10}
Exemplo n.º 12
0
    def test_include(self):
        self.isfiles.append('/path/to/file2.yaml')
        mock_open1 = mock.mock_open(read_data='myinclude: !include file2.yaml')
        mock_open1.return_value.name = '/path/to/my.yaml'
        mock_open2 = mock.mock_open(read_data='10')
        mock_open2.return_value.name = '/path/to/file2.yaml'
        open_mock = self.patch(
            'devops.helpers.templates.open',
            new_callable=mock.mock_open, create=True)
        open_mock.side_effect = (
            mock_open1.return_value, mock_open2.return_value)

        m = templates.yaml_template_load('/path/to/my.yaml')
        assert m == {'myinclude': 10}
Exemplo n.º 13
0
    def load_template(self, filename):
        """Method for reading file with devops config

        :param filename: string
        """
        if filename is not None:
            LOG.debug(
                "Preparing to load config from template '{0}'".format(
                    filename
                )
            )
            self.config = templates.yaml_template_load(filename)
        else:
            LOG.error("Template filename is not set, loading config " +
                      "from template aborted.")
Exemplo n.º 14
0
def create_config():
    env = os.environ

    conf_path = env['CONF_PATH']
    conf = yaml_template_load(conf_path)

    slaves_count = int(env['SLAVES_COUNT'])

    group = conf['template']['devops_settings']['groups'][0]
    defined = filter(lambda x: x['role'] == 'k8s-node', group['nodes'])
    node_params = filter(lambda x: x['name'].endswith('slave-0'),
                         group['nodes'])[0]['params']

    for i in range(len(defined), slaves_count):
        group['nodes'].append({
            'name': 'slave-{}'.format(i),
            'role': 'k8s-node',
            'params': deepcopy(node_params)
        })

    return conf
Exemplo n.º 15
0
def create_config():
    env = os.environ

    conf_path = env['CONF_PATH']
    conf = yaml_template_load(conf_path)

    slaves_count = int(env['SLAVES_COUNT'])

    group = conf['template']['devops_settings']['groups'][0]
    defined = filter(lambda x: x['role'] == 'k8s-node',
                     group['nodes'])
    node_params = filter(lambda x: x['name'].endswith('slave-0'),
                         group['nodes'])[0]['params']

    for i in range(len(defined), slaves_count):
        group['nodes'].append(
            {'name': 'slave-{}'.format(i),
             'role': 'k8s-node',
             'params': deepcopy(node_params)})

    return conf
 def test_not_file(self):
     with self.assertRaises(error.DevopsError):
         templates.yaml_template_load('/path/to/dir')
 def test_os_env(self):
     open_mock = mock.mock_open(read_data='env_value: !os_env MYVAR\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     self.os_mock.environ = {'MYVAR': '30'}
     m = templates.yaml_template_load('/path/to/my.yaml')
     assert m['env_value'] == 30
 def test_os_env_no_value(self):
     open_mock = mock.mock_open(read_data='env_value: !os_env MYVAR\n')
     self.patch('devops.helpers.templates.open', open_mock, create=True)
     self.os_mock.environ = {}
     with self.assertRaises(error.DevopsError):
         templates.yaml_template_load('/path/to/my.yaml')
Exemplo n.º 19
0
 def test_not_file(self):
     with self.assertRaises(error.DevopsError):
         templates.yaml_template_load('/path/to/dir')