def test_conf_commands():
entry = Entrypoint(conf='configs/base.yml')
checks = [
('/tmp/OK', 'TEST'),
('/tmp/OKOK', 'TEST2'),
('/tmp/OKOKOK', 'TEST3'),
('/tmp/OKOKOKOK', 'TEST4'),
('/tmp/OKOKOKOKOK', 'TEST5'),
('/tmp/user', '1000'),
('/tmp/group', '1000'),
('/tmp/debug', 'true'),
]
os.environ['ENTRYPOINT_PRECONF_COMMAND'] = 'echo TEST4 > /tmp/OKOKOKOK'
os.environ['ENTRYPOINT_POSTCONF_COMMAND'] = 'echo TEST5 > /tmp/OKOKOKOKOK'
entry.config.set_to_env()
entry.run_pre_conf_cmds()
entry.run_post_conf_cmds()
for filename, value in checks:
with open(filename) as f:
line = f.readline()
print(line)
assert line.startswith(value)
def _reloader_check(conf, command):
entry = Entrypoint(conf=conf)
entry.apply_conf()
entry.config.reload.run(ret=True)
subprocess.check_call(command)
sleep(1)
entry.config.reload.stop()
def test_disabled_reloader():
os.environ['ENTRYPOINT_DISABLE_RELOAD'] = 'true'
with mock.patch('os.kill') as os_kill:
entry = Entrypoint(conf='configs/reloader/reloader.yml')
entry.apply_conf()
assert entry.config.reload is None
assert not os_kill.called
def test_templates():
test_confs = ['configs/base.yml']
for test_conf in test_confs:
entry = Entrypoint(conf=test_conf)
conf = entry.config
entry.apply_conf()
for _, config_file in conf.get_templates():
with open(config_file, mode='r') as r:
test = load(stream=r, Loader=FullLoader)
assert len(set(test['All links'])) == 4
assert len(set(test['All links 1'])) == 2
assert len(set(test['All links 2'])) == 2
assert fnmatch.fnmatch(test['Links 2 800'][0], 'udp://*:800')
# test environment
assert test['All environ']['FOO'] == 'bar'
assert test['All links 2 environ']['FOO'] == 'bar'
test_names = [
'test1',
'test2',
'test3',
'test4',
]
# test names
for test_name in test_names:
assert test_name in test['All names']
# test id
for id in test['ID']:
int(id, base=16)
# test env
assert test['ENV']['SECRET'].strip() == 'nothing'
assert test['ENVIRON']['SECRET'].strip() == 'nothing'
# test yaml
assert test['YAML']['yaml'] == 'ok'
# test json
assert test['JSON']['json'] == 'ok'
# test envtobool
assert test['ENVTOBOOL']['ok']
assert not test['ENVTOBOOL']['ko']
# test toml
assert test['TOML'].strip() == '[toml]\nvalue = "ok"'
# test configparser
assert test['CONFIGPARSER'].strip() == '[ini]\nvalue = ok'
def test_user_env():
os.environ['ENTRYPOINT_USER'] = '******'
entry = Entrypoint(conf='configs/base.yml')
assert entry.config.user == 100
os.environ['ENTRYPOINT_USER'] = '******'
entry = Entrypoint(conf='configs/base.yml')
assert entry.config.user == 1009
def test_group_env():
os.environ['ENTRYPOINT_GROUP'] = '100'
entry = Entrypoint(conf='configs/base.yml')
assert entry.config.group == 100
os.environ['ENTRYPOINT_GROUP'] = 'testgroup'
entry = Entrypoint(conf='configs/base.yml')
assert entry.config.group == 1010
def test_disabled_service():
os.environ['ENTRYPOINT_DISABLE_SERVICE'] = 'true'
entry = Entrypoint(conf='configs/base.yml')
assert entry.is_disabled
with pytest.raises(SystemExit):
entry.exit_if_disabled()
del os.environ['ENTRYPOINT_DISABLE_SERVICE']
def test_disabled_service():
os.environ['ENTRYPOINT_DISABLE_SERVICE'] = 'true'
entry = Entrypoint(conf='configs/base.yml')
assert entry.is_disabled
with pytest.raises(SystemExit):
entry.exit_if_disabled()
del os.environ['ENTRYPOINT_DISABLE_SERVICE']
def test_commands_handling():
cat = Entrypoint(conf='configs/commands.yml', args=['cat', 'hello'])
sleep = Entrypoint(conf='configs/commands.yml', args=['sleep', '1'])
bash = Entrypoint(conf='configs/commands.yml', args=['bash'])
zsh = Entrypoint(conf='configs/commands.yml', args=['zsh', '-c', 'exit'])
empty = Entrypoint(conf='configs/empty.yml', args=['zsh', '-c', 'exit'])
assert cat.is_handled
assert sleep.is_handled
assert bash.is_handled
assert not zsh.is_handled
assert empty.is_handled
def test_reloader():
if 'ENTRYPOINT_DISABLE_RELOAD' in os.environ:
os.environ.pop('ENTRYPOINT_DISABLE_RELOAD')
with mock.patch('os.kill') as os_kill:
entry = Entrypoint(conf='configs/reloader/reloader.yml')
entry.apply_conf()
entry.config.reload.run(ret=True)
subprocess.check_call(['touch', '/tmp/reload'])
sleep(1)
entry.config.reload.stop()
os_kill.assert_called_once_with(1, SIGHUP)
def test_debug_env():
os.environ['ENTRYPOINT_DEBUG'] = 'true'
entry = Entrypoint(conf='configs/empty.yml')
assert entry.config.debug
del os.environ['ENTRYPOINT_DEBUG']
def test_quiet_env():
os.environ['ENTRYPOINT_QUIET'] = 'true'
entry = Entrypoint(conf='configs/empty.yml')
assert entry.config.quiet
del os.environ['ENTRYPOINT_QUIET']
def test_config_file():
os.environ['ENTRYPOINT_CONFIG'] = 'configs/base.yml'
entry = Entrypoint()
assert entry.config.has_config
del os.environ['ENTRYPOINT_CONFIG']
def test_templates():
test_confs = ['configs/base.yml']
for test_conf in test_confs:
entry = Entrypoint(conf='configs/base.yml')
conf = entry.config
entry.apply_conf()
for _, config_file in conf.get_templates():
with open(config_file, mode='r') as r:
test = load(stream=r, Loader=Loader)
assert len(set(test['All links'])) == 4
assert len(set(test['All links 1'])) == 2
assert len(set(test['All links 2'])) == 2
assert fnmatch.fnmatch(test['Links 2 800'][0], 'udp://*:800')
# test environment
assert test['All environ']['FOO'] == 'bar'
assert test['All links 2 environ']['FOO'] == 'bar'
test_names = [
'test1',
'test2',
'test3',
'test4',
]
# test names
for test_name in test_names:
assert test_name in test['All names']
# test id
for id in test['ID']:
int(id, base=16)
# test env
assert test['ENV']['SECRET'] == 'nothing'
assert test['ENVIRON']['SECRET'] == 'nothing'
# test yaml
assert test['YAML']['yaml'] == 'ok'
# test json
assert test['JSON']['json'] == 'ok'
def test_templates():
test_confs = ['configs/base.yml']
for test_conf in test_confs:
entry = Entrypoint(conf='configs/base.yml')
conf = entry.config
entry.apply_conf()
for _, config_file in conf.get_templates():
with open(config_file, mode='r') as r:
test = load(stream=r, Loader=Loader)
assert len(set(test['All links'])) == 4
assert len(set(test['All links 1'])) == 2
assert len(set(test['All links 2'])) == 2
assert fnmatch.fnmatch(test['Links 2 800'][0], 'udp://*:800')
# test environment
assert test['All environ']['FOO'] == 'bar'
assert test['All links 2 environ']['FOO'] == 'bar'
test_names = [
'test1',
'test2',
'test3',
'test4',
]
# test names
for test_name in test_names:
assert test_name in test['All names']
# test id
for id in test['ID']:
int(id, base=16)
# test env
assert test['ENV']['SECRET'] == 'nothing'
assert test['ENVIRON']['SECRET'] == 'nothing'
# test yaml
assert test['YAML']['yaml'] == 'ok'
# test json
assert test['JSON']['json'] == 'ok'
def test_display_raw():
entry = Entrypoint()
assert not entry.raw_output
os.environ['ENTRYPOINT_RAW'] = 'True'
assert entry.raw_output
del os.environ['ENTRYPOINT_RAW']
def test_force_config():
entry = Entrypoint()
assert not entry.should_config
os.environ['ENTRYPOINT_FORCE'] = 'True'
assert entry.should_config
del os.environ['ENTRYPOINT_FORCE']
def test_entrypoint_links():
entry = Entrypoint(conf='configs/base.yml')
links = entry.config.links
assert len(links.all) == 4
assert len(links.test1) == 2
assert links.test2_800.port == 800
def test_runner_parallele():
run = [
(Process(target=Entrypoint(conf='configs/runner_parallele.yml',
args=['sleep', '5']).launch), 0, 0),
]
for proc, uid, gid in run:
proc.start()
proc.join()
assert compare_timestamps() < 1
assert os.stat('/tmp/timestamp1').st_uid == uid
assert os.stat('/tmp/timestamp1').st_gid == gid
def test_conf_commands():
entry = Entrypoint(conf='configs/base.yml')
checks = [
('/tmp/OK', 'TEST'),
('/tmp/OKOK', 'TEST2'),
('/tmp/OKOKOK', 'TEST3'),
('/tmp/OKOKOKOK', 'TEST4'),
('/tmp/OKOKOKOKOK', 'TEST5'),
('/tmp/user', '1000'),
('/tmp/group', '1000'),
('/tmp/debug', 'true'),
]
os.environ['ENTRYPOINT_PRECONF_COMMAND'] = 'echo TEST4 > /tmp/OKOKOKOK'
os.environ['ENTRYPOINT_POSTCONF_COMMAND'] = 'echo TEST5 > /tmp/OKOKOKOKOK'
entry.config.set_to_env()
entry.run_pre_conf_cmds()
entry.run_post_conf_cmds()
for filename, value in checks:
with open(filename) as f:
line = f.readline()
print(line)
assert line.startswith(value)
def test_command():
run = [
# ((Process instance), (file to check), (uid), (gid))
(Process(target=Entrypoint(conf='configs/base.yml',
args=['-c', 'echo OK > /tmp/CMD']).launch),
'/tmp/CMD', 1000, 1000),
(Process(target=Entrypoint(
conf='configs/base.yml',
args=['bash', '-c', 'echo ${SECRET}OK > /tmp/CMD2']).launch),
'/tmp/CMD2', 1000, 1000),
(Process(target=Entrypoint(conf='configs/usernames.yml',
args=['bash', '-c', 'echo OK > /tmp/CMD3'
]).launch), '/tmp/CMD3', 1009, 1010),
(Process(target=Entrypoint(conf='configs/unhandled.yml',
args=['bash', '-c', 'echo OK > /tmp/CMD4'
]).launch), '/tmp/CMD4', 0, 0),
(Process(target=Entrypoint(conf='/dontexist',
args=['bash', '-c', 'echo OK > /tmp/CMD5'
]).launch), '/tmp/CMD5', 0, 0),
(Process(target=Entrypoint(
conf='configs/secret_env.yml',
args=['bash', '-c', 'echo ${SECRET}OK > /tmp/CMD6']).launch),
'/tmp/CMD6', 0, 0),
]
for proc, test, uid, gid in run:
proc.start()
proc.join()
with open(test, 'r') as f:
assert f.readline().startswith('OK')
assert os.stat(test).st_uid == uid
assert os.stat(test).st_gid == gid
assert not os.path.isfile('/.dockerenv')
assert not os.path.isfile('/.dockerinit')
def test_runner():
run = [
(Process(target=Entrypoint(conf='configs/runner.yml',
args=['sleep', '5']).launch),
'/tmp/runner_test', 0, 0),
]
for proc, test, uid, gid in run:
proc.start()
proc.join()
with open(test, 'r') as f:
assert f.readline().startswith('OK')
assert os.stat(test).st_uid == uid
assert os.stat(test).st_gid == gid
def test_conf_commands():
entry = Entrypoint(conf='configs/base.yml')
os.environ['ENTRYPOINT_PRECONF_COMMAND'] = 'echo TEST4 > /tmp/OKOKOKOK'
os.environ['ENTRYPOINT_POSTCONF_COMMAND'] = 'echo TEST5 > /tmp/OKOKOKOKOK'
entry.run_pre_conf_cmds()
entry.run_post_conf_cmds()
with open('/tmp/OK') as f:
assert f.readline().startswith('TEST')
with open('/tmp/OKOK') as f:
assert f.readline().startswith('TEST2')
with open('/tmp/OKOKOK') as f:
assert f.readline().startswith('TEST3')
with open('/tmp/OKOKOKOK') as f:
assert f.readline().startswith('TEST4')
with open('/tmp/OKOKOKOKOK') as f:
assert f.readline().startswith('TEST5')
def test_command_matching_setup():
bash = Entrypoint(conf='configs/matching_command.yml', args=['bash'])
zsh = Entrypoint(conf='configs/matching_command.yml', args=['zsh'])
assert bash.config.user == 1000
assert zsh.config.user == 1001
assert bash.config.group == 1002
assert zsh.config.group == 1003
assert bash.config.config_files == [
'file1.tpl',
{
'file2': 'file3'
},
'file4',
'file9',
{
'file10': 'file11'
},
]
assert zsh.config.config_files == [
'file5.tpl',
{
'file6': 'file7'
},
'file8',
'file9',
{
'file10': 'file11'
},
]
assert bash.config.secret_env == [
'secret1',
'secret2',
]
assert zsh.config.secret_env == [
'secret1',
'secret3',
]
assert bash.config.pre_conf_commands == [
'cmd1',
'cmd3',
]
assert zsh.config.pre_conf_commands == [
'cmd2',
'cmd3',
]
assert bash.config.post_conf_commands == [
'cmd4',
'cmd6',
]
assert zsh.config.post_conf_commands == [
'cmd4',
'cmd5',
]
assert bash.config.set_environment == [{'ENV_1': 'echo set ENV_1'}]
assert zsh.config.set_environment == [{'ENV_2': 'echo set ENV_2'}]
assert bash.config.post_run_commands == [
'cmd7',
'cmd8',
]
assert zsh.config.post_run_commands == [
'cmd8',
'cmd9',
]
assert bash.config.debug
assert zsh.config.debug
assert bash.config.clean_env
assert not zsh.config.clean_env
assert bash.config.remove_dockerenv
assert zsh.config.remove_dockerenv