salt/tests/integration/modules/test_dockermod.py
2018-06-05 10:57:09 -05:00

100 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
'''
Integration tests for the docker_container states
'''
# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals
import functools
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.helpers import destructiveTest
from tests.support.mixins import SaltReturnAssertsMixin
# Import Salt Libs
import salt.utils.path
# Import 3rd-party libs
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
def with_random_name(func):
'''
generate a randomized name for a container
'''
@functools.wraps(func)
def wrapper(self, *args, **kwargs):
name = _random_name(prefix='salt_')
return func(self, _random_name(prefix='salt_test_'), *args, **kwargs)
return wrapper
@destructiveTest
@skipIf(not salt.utils.path.which('dockerd'), 'Docker not installed')
class DockerCallTestCase(ModuleCase, SaltReturnAssertsMixin):
'''
Test docker_container states
'''
@with_random_name
def setUp(self, name):
'''
setup docker.call tests
'''
# Create temp dir
self.random_name = name
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)