salt/tests/integration/modules/test_dockermod.py

# -*- coding: utf-8 -*-
'''
Integration tests for the docker_container states
'''

# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals
import random
import string
import sys

# Import Salt Testing Libs
from tests.support.unit import skipIf
from tests.support.case import ModuleCase
from tests.support.mixins import SaltReturnAssertsMixin

# Import Salt Libs
import salt.utils.path

# Import 3rd-party libs
import pytest
from salt.ext.six.moves import range  # pylint: disable=import-error,redefined-builtin


def _random_name(prefix=''):
    ret = prefix
    for _ in range(8):
        ret += random.choice(string.ascii_lowercase)
    return ret


@pytest.mark.destructive_test
@skipIf(not salt.utils.path.which('dockerd'), 'Docker not installed')
class DockerCallTestCase(ModuleCase, SaltReturnAssertsMixin):
    '''
    Test docker_container states
    '''

    def setUp(self):
        '''
        setup docker.call tests
        '''
        # Create temp dir
        self.random_name = _random_name(prefix='salt_test_')
        self.image_tag = sys.version_info[0]

        self.run_state('docker_image.present',
                       tag=self.image_tag,
                       name='python')
        self.run_state('docker_container.running',
                       name=self.random_name,
                       image='python:{0}'.format(self.image_tag),
                       entrypoint='tail -f /dev/null')

    def tearDown(self):
        '''
        teardown docker.call tests
        '''
        self.run_state('docker_container.absent',
                       name=self.random_name,
                       force=True)
        self.run_state('docker_image.absent',
                       images=['python:{0}'.format(self.image_tag)],
                       force=True)
        delattr(self, 'random_name')
        delattr(self, 'image_tag')

    def test_docker_call(self):
        '''
        check that docker.call works, and works with a container not running as root
        '''
        ret = self.run_function('docker.call', [self.random_name, 'test.ping'])
        assert ret is True

    def test_docker_sls(self):
        '''
        check that docker.sls works, and works with a container not running as root
        '''
        ret = self.run_function('docker.apply', [self.random_name, 'core'])
        self.assertSaltTrueReturn(ret)

    def test_docker_highstate(self):
        '''
        check that docker.highstate works, and works with a container not running as root
        '''
        ret = self.run_function('docker.apply', [self.random_name])
        self.assertSaltTrueReturn(ret)