salt/tests/integration/utils/test_reactor.py
Thomas Jackson 5bfef71240 Pylint fix
2015-01-28 18:43:43 -08:00

69 lines
1.7 KiB
Python

# -*- coding: utf-8 -*-
import time
import shutil
import tempfile
import os
from contextlib import contextmanager
import integration
from salt.utils.process import clean_proc
from salt.utils import event
from salttesting.mock import patch
@contextmanager
def reactor_process(opts, reactor):
opts = dict(opts)
opts['reactor'] = reactor
proc = event.Reactor(opts)
proc.start()
try:
if os.environ.get('TRAVIS_PYTHON_VERSION', None) is not None:
# Travis is slow
time.sleep(10)
else:
time.sleep(2)
yield
finally:
clean_proc(proc)
@contextmanager
def _args_sideffect(*args, **kwargs):
return args, kwargs
class TestReactor(integration.ModuleCase):
def setUp(self):
self.opts = self.get_config('master', from_scratch=True)
self.tempdir = tempfile.mkdtemp(dir=integration.SYS_TMP_DIR)
self.sls_name = os.path.join(self.tempdir, 'test.sls')
with open(self.sls_name, 'w') as fh:
fh.write('''
update_fileserver:
runner.fileserver.update
''')
def tearDown(self):
if os.path.isdir(self.tempdir):
shutil.rmtree(self.tempdir)
def test_basic(self):
reactor_config = [
{'salt/tagA': ['/srv/reactor/A.sls']},
{'salt/tagB': ['/srv/reactor/B.sls']},
{'*': ['/srv/reactor/all.sls']},
]
wrap = event.ReactWrap(self.opts)
with patch('salt.utils.event.ReactWrap.local', _args_sideffect):
ret = wrap.run({'fun': 'test.ping',
'state': 'local',
'order': 1,
'name': 'foo_action',
'__id__': 'foo_action'})
raise Exception(ret)