1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
|
# test mod_md notify support
import os
import time
import pytest
from .md_conf import MDConf, MDConf
from .md_env import MDTestEnv
@pytest.mark.skipif(condition=not MDTestEnv.has_acme_server(),
reason="no ACME test server configured")
class TestNotify:
notify_cmd = None
notify_log = None
domain = None
@pytest.fixture(autouse=True, scope='class')
def _class_scope(self, env, acme):
acme.start(config='default')
env.check_acme()
env.clear_store()
@pytest.fixture(autouse=True, scope='function')
def _method_scope(self, env, request):
self.domain = env.get_request_domain(request)
self.notify_cmd = os.path.join(env.test_dir, "../modules/md/notify.py")
self.notify_log = os.path.join(env.gen_dir, "notify.log")
if os.path.isfile(self.notify_log):
os.remove(self.notify_log)
def configure_httpd(self, env, domain, add_lines=""):
conf = MDConf(env)
conf.add(add_lines)
conf.add_md([domain])
conf.add_vhost(domain)
conf.install()
return domain
# test: invalid notify cmd, check error
def test_md_900_001(self, env):
command = "blablabla"
args = ""
self.configure_httpd(env, self.domain, f"""
MDNotifyCmd {command} {args}
""")
assert env.apache_restart() == 0, f'{env.apachectl_stderr}'
assert env.await_error(self.domain)
stat = env.get_md_status(self.domain)
assert stat["renewal"]["last"]["problem"] == "urn:org:apache:httpd:log:AH10108:"
#
env.httpd_error_log.ignore_recent(
matches = [
r'.*urn:org:apache:httpd:log:AH10108:.*'
]
)
# test: valid notify cmd that fails, check error
def test_md_900_002(self, env):
command = "%s/notifail.py" % env.test_dir
args = ""
self.configure_httpd(env, self.domain, f"""
MDNotifyCmd {command} {args}
""")
assert env.apache_restart() == 0, f'{env.apachectl_stderr}'
assert env.await_error(self.domain)
stat = env.get_md_status(self.domain)
assert stat["renewal"]["last"]["problem"] == "urn:org:apache:httpd:log:AH10108:"
#
env.httpd_error_log.ignore_recent(
matches = [
r'.*urn:org:apache:httpd:log:AH10108:.*',
r'.*urn:org:apache:httpd:log:AH10109:.*'
r'.*problem\[challenge-setup-failure\].*',
]
)
# test: valid notify that logs to file
def test_md_900_010(self, env):
command = self.notify_cmd
args = self.notify_log
self.configure_httpd(env, self.domain, f"""
MDNotifyCmd {command} {args}
""")
assert env.apache_restart() == 0, f'{env.apachectl_stderr}'
assert env.await_completion([self.domain], restart=False)
time.sleep(1)
stat = env.get_md_status(self.domain)
assert stat["renewal"]["last"]["status"] == 0
time.sleep(1)
nlines = open(self.notify_log).readlines()
assert 1 == len(nlines)
assert ("['%s', '%s', '%s']" % (command, args, self.domain)) == nlines[0].strip()
# test: signup with working notify cmd and see that it is called with the
# configured extra arguments
def test_md_900_011(self, env):
command = self.notify_cmd
args = self.notify_log
extra_arg = "test_900_011_extra"
self.configure_httpd(env, self.domain, f"""
MDNotifyCmd {command} {args} {extra_arg}
""")
assert env.apache_restart() == 0, f'{env.apachectl_stderr}'
assert env.await_completion([self.domain], restart=False)
time.sleep(1)
stat = env.get_md_status(self.domain)
assert stat["renewal"]["last"]["status"] == 0
nlines = open(self.notify_log).readlines()
assert ("['%s', '%s', '%s', '%s']" % (command, args, extra_arg, self.domain)) == nlines[0].strip()
# test: signup with working notify cmd for 2 MD and expect it to be called twice
def test_md_900_012(self, env):
md1 = "a-" + self.domain
domains1 = [md1, "www." + md1]
md2 = "b-" + self.domain
domains2 = [md2, "www." + md2]
command = self.notify_cmd
args = self.notify_log
conf = MDConf(env)
conf.add(f"MDNotifyCmd {command} {args}")
conf.add_md(domains1)
conf.add_md(domains2)
conf.add_vhost(domains1)
conf.add_vhost(domains2)
conf.install()
assert env.apache_restart() == 0, f'{env.apachectl_stderr}'
assert env.await_completion([md1, md2], restart=False)
time.sleep(1)
stat = env.get_md_status(md1)
assert stat["renewal"]["last"]["status"] == 0
stat = env.get_md_status(md2)
assert stat["renewal"]["last"]["status"] == 0
nlines = open(args).readlines()
assert 2 == len(nlines)
|