salt/tests/kitchen/test_kitchen.py

import os
import logging

from tests.support.unit import TestCase, skipIf
import setup

import salt.utils.path
from salt.modules import cmdmod as cmd

CURRENT_DIR = os.path.dirname(os.path.realpath(__file__))
log = logging.getLogger(__name__)


@skipIf(not salt.utils.path.which('bundle'), 'Bundler is not installed')
class KitchenTestCase(TestCase):
    '''
    Test kitchen environments
    '''
    @classmethod
    def setUpClass(cls):
        '''
        setup kitchen tests
        '''
        cls.topdir = '/' + os.path.join(*CURRENT_DIR.split('/')[:-2])
        cls.use_vt = int(os.environ.get('TESTS_LOG_LEVEL')) >= 3
        cmd.run('python setup.py sdist', cwd=cls.topdir)
        cmd.run('bundle install', cwd=CURRENT_DIR)

    @classmethod
    def tearDownClass(cls):
        del cls.topdir


def func_builder(testdir):
    def func(self):
        env = {
            'KITCHEN_YAML': os.path.join(CURRENT_DIR, '.kitchen.yml'),
            'SALT_SDIST_PATH': os.path.join(self.topdir, 'dist', 'salt-{0}.tar.gz'.format(setup.__version__)),
        }
        if 'TESTS_XML_OUTPUT_DIR' in os.environ:
            env['PYTEST_ADDOPTS'] = '--junit-xml {0}/kitchen.tests.{1}.xml'.format(
                os.environ.get('TESTS_XML_OUTPUT_DIR'),
                testdir,
            )
        self.assertEqual(
            cmd.retcode(
                'bundle exec kitchen create -c 4 all',
                cwd=os.path.join(CURRENT_DIR, 'tests', testdir),
                env=env,
                use_vt=self.use_vt,
            ),
            0
        )
        self.assertEqual(
            cmd.retcode(
                'bundle exec kitchen verify all',
                cwd=os.path.join(CURRENT_DIR, 'tests', testdir),
                env=env,
                use_vt=self.use_vt,
            ),
            0
        )
        self.assertEqual(
            cmd.retcode(
                'bundle exec kitchen destroy all',
                cwd=os.path.join(CURRENT_DIR, 'tests', testdir),
                env=env,
                use_vt=self.use_vt,
            ),
            0
        )
    return func

for testdir in os.listdir(os.path.join(CURRENT_DIR, 'tests')):
    setattr(KitchenTestCase, 'test_kitchen_{0}'.format(testdir), func_builder(testdir))