2020-05-19 22:27:14 -04:00
|
|
|
"""Unit tests: template_esh"""
|
2020-07-11 15:13:24 -04:00
|
|
|
import os
|
|
|
|
|
|
|
|
FILE_MODE = 0o754
|
2020-05-19 22:27:14 -04:00
|
|
|
|
|
|
|
LOCAL_CLASS = "esh_Test+@-!^Class"
|
2021-01-10 16:38:42 -05:00
|
|
|
LOCAL_CLASS2 = "esh_Test+@-|^2nd_Class withSpace"
|
2021-12-23 16:42:43 -05:00
|
|
|
LOCAL_ARCH = "esh_Test+@-!^Arch"
|
2020-05-19 22:27:14 -04:00
|
|
|
LOCAL_SYSTEM = "esh_Test+@-!^System"
|
|
|
|
LOCAL_HOST = "esh_Test+@-!^Host"
|
|
|
|
LOCAL_USER = "esh_Test+@-!^User"
|
|
|
|
LOCAL_DISTRO = "esh_Test+@-!^Distro"
|
|
|
|
TEMPLATE = f'''
|
|
|
|
start of template
|
|
|
|
esh class = ><%=$YADM_CLASS%><
|
2021-12-23 16:42:43 -05:00
|
|
|
esh arch = ><%=$YADM_ARCH%><
|
2020-05-19 22:27:14 -04:00
|
|
|
esh os = ><%=$YADM_OS%><
|
|
|
|
esh host = ><%=$YADM_HOSTNAME%><
|
|
|
|
esh user = ><%=$YADM_USER%><
|
|
|
|
esh distro = ><%=$YADM_DISTRO%><
|
|
|
|
<% if [ "$YADM_CLASS" = "wrongclass1" ]; then -%>
|
|
|
|
wrong class 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_CLASS" = "{LOCAL_CLASS}" ]; then -%>
|
|
|
|
Included section for class = <%=$YADM_CLASS%> (<%=$YADM_CLASS%> repeated)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_CLASS" = "wrongclass2" ]; then -%>
|
|
|
|
wrong class 2
|
|
|
|
<% fi -%>
|
2021-01-10 16:38:42 -05:00
|
|
|
<% echo "$YADM_CLASSES" | while IFS='' read cls; do
|
|
|
|
if [ "$cls" = "{LOCAL_CLASS2}" ]; then -%>
|
|
|
|
Included section for second class
|
|
|
|
<% fi; done -%>
|
2021-12-23 16:42:43 -05:00
|
|
|
<% if [ "$YADM_ARCH" = "wrongarch1" ]; then -%>
|
|
|
|
wrong arch 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_ARCH" = "{LOCAL_ARCH}" ]; then -%>
|
|
|
|
Included section for arch = <%=$YADM_ARCH%> (<%=$YADM_ARCH%> repeated)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_ARCH" = "wrongarch2" ]; then -%>
|
|
|
|
wrong arch 2
|
|
|
|
<% fi -%>
|
2020-05-19 22:27:14 -04:00
|
|
|
<% if [ "$YADM_OS" = "wrongos1" ]; then -%>
|
|
|
|
wrong os 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_OS" = "{LOCAL_SYSTEM}" ]; then -%>
|
|
|
|
Included section for os = <%=$YADM_OS%> (<%=$YADM_OS%> repeated)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_OS" = "wrongos2" ]; then -%>
|
|
|
|
wrong os 2
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_HOSTNAME" = "wronghost1" ]; then -%>
|
|
|
|
wrong host 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_HOSTNAME" = "{LOCAL_HOST}" ]; then -%>
|
|
|
|
Included section for host = <%=$YADM_HOSTNAME%> (<%=$YADM_HOSTNAME%> again)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_HOSTNAME" = "wronghost2" ]; then -%>
|
|
|
|
wrong host 2
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_USER" = "wronguser1" ]; then -%>
|
|
|
|
wrong user 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_USER" = "{LOCAL_USER}" ]; then -%>
|
|
|
|
Included section for user = <%=$YADM_USER%> (<%=$YADM_USER%> repeated)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_USER" = "wronguser2" ]; then -%>
|
|
|
|
wrong user 2
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_DISTRO" = "wrongdistro1" ]; then -%>
|
|
|
|
wrong distro 1
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_DISTRO" = "{LOCAL_DISTRO}" ]; then -%>
|
|
|
|
Included section for distro = <%=$YADM_DISTRO%> (<%=$YADM_DISTRO%> again)
|
|
|
|
<% fi -%>
|
|
|
|
<% if [ "$YADM_DISTRO" = "wrongdistro2" ]; then -%>
|
|
|
|
wrong distro 2
|
|
|
|
<% fi -%>
|
|
|
|
end of template
|
|
|
|
'''
|
|
|
|
EXPECTED = f'''
|
|
|
|
start of template
|
|
|
|
esh class = >{LOCAL_CLASS}<
|
2021-12-23 16:42:43 -05:00
|
|
|
esh arch = >{LOCAL_ARCH}<
|
2020-05-19 22:27:14 -04:00
|
|
|
esh os = >{LOCAL_SYSTEM}<
|
|
|
|
esh host = >{LOCAL_HOST}<
|
|
|
|
esh user = >{LOCAL_USER}<
|
|
|
|
esh distro = >{LOCAL_DISTRO}<
|
|
|
|
Included section for class = {LOCAL_CLASS} ({LOCAL_CLASS} repeated)
|
2021-01-10 16:38:42 -05:00
|
|
|
Included section for second class
|
2021-12-23 16:42:43 -05:00
|
|
|
Included section for arch = {LOCAL_ARCH} ({LOCAL_ARCH} repeated)
|
2020-05-19 22:27:14 -04:00
|
|
|
Included section for os = {LOCAL_SYSTEM} ({LOCAL_SYSTEM} repeated)
|
|
|
|
Included section for host = {LOCAL_HOST} ({LOCAL_HOST} again)
|
|
|
|
Included section for user = {LOCAL_USER} ({LOCAL_USER} repeated)
|
|
|
|
Included section for distro = {LOCAL_DISTRO} ({LOCAL_DISTRO} again)
|
|
|
|
end of template
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
|
|
def test_template_esh(runner, yadm, tmpdir):
|
|
|
|
"""Test processing by esh"""
|
|
|
|
|
|
|
|
input_file = tmpdir.join('input')
|
|
|
|
input_file.write(TEMPLATE, ensure=True)
|
2020-07-11 15:13:24 -04:00
|
|
|
input_file.chmod(FILE_MODE)
|
2020-05-19 22:27:14 -04:00
|
|
|
output_file = tmpdir.join('output')
|
|
|
|
|
2021-02-28 11:55:37 -05:00
|
|
|
# ensure overwrite works when file exists as read-only (there is some
|
|
|
|
# special processing when this is encountered because some environments do
|
|
|
|
# not properly overwrite read-only files)
|
|
|
|
output_file.write('existing')
|
|
|
|
output_file.chmod(0o400)
|
|
|
|
|
2020-05-19 22:27:14 -04:00
|
|
|
script = f"""
|
|
|
|
YADM_TEST=1 source {yadm}
|
|
|
|
local_class="{LOCAL_CLASS}"
|
2021-01-10 16:38:42 -05:00
|
|
|
local_classes=("{LOCAL_CLASS2}" "{LOCAL_CLASS}")
|
2021-12-23 16:42:43 -05:00
|
|
|
local_arch="{LOCAL_ARCH}"
|
2020-05-19 22:27:14 -04:00
|
|
|
local_system="{LOCAL_SYSTEM}"
|
|
|
|
local_host="{LOCAL_HOST}"
|
|
|
|
local_user="{LOCAL_USER}"
|
|
|
|
local_distro="{LOCAL_DISTRO}"
|
|
|
|
template_esh "{input_file}" "{output_file}"
|
|
|
|
"""
|
|
|
|
run = runner(command=['bash'], inp=script)
|
|
|
|
assert run.success
|
|
|
|
assert run.err == ''
|
|
|
|
assert output_file.read().strip() == str(EXPECTED).strip()
|
2020-07-11 15:13:24 -04:00
|
|
|
assert os.stat(output_file).st_mode == os.stat(input_file).st_mode
|
2020-05-19 22:27:14 -04:00
|
|
|
|
|
|
|
|
|
|
|
def test_source(runner, yadm, tmpdir):
|
|
|
|
"""Test YADM_SOURCE"""
|
|
|
|
|
|
|
|
input_file = tmpdir.join('input')
|
|
|
|
input_file.write('<%= $YADM_SOURCE %>', ensure=True)
|
2020-07-11 15:13:24 -04:00
|
|
|
input_file.chmod(FILE_MODE)
|
2020-05-19 22:27:14 -04:00
|
|
|
output_file = tmpdir.join('output')
|
|
|
|
|
|
|
|
script = f"""
|
|
|
|
YADM_TEST=1 source {yadm}
|
|
|
|
template_esh "{input_file}" "{output_file}"
|
|
|
|
"""
|
|
|
|
run = runner(command=['bash'], inp=script)
|
|
|
|
assert run.success
|
|
|
|
assert run.err == ''
|
|
|
|
assert output_file.read().strip() == str(input_file)
|
2020-07-11 15:13:24 -04:00
|
|
|
assert os.stat(output_file).st_mode == os.stat(input_file).st_mode
|