/
usr
/
lib
/
imh-dnskeyapi
/
tests
/
/usr/lib/imh-dnskeyapi/tests
mkdir
upload
Name
Size
Mode
Actions
conftest.py
891
0644
edit
dl
rm
test_api.py
1024
0644
edit
dl
rm
__init__.py
0
0644
edit
dl
rm
Edit:
/usr/lib/imh-dnskeyapi/tests/conftest.py
(891B)
"""Pytest fixtures""" import os import shutil import pytest CONF = "/usr/local/cpanel/whostmgr/etc/imhapi/allowed_hosts.conf" # pylint:disable=redefined-outer-name @pytest.fixture def allow_local(): """Add 127.0.0.1 to ``CONF`` under authorized_ips""" shutil.copy(CONF, f"{CONF}.bak") with open(CONF, encoding='utf-8') as file: data = file.read().splitlines(keepends=True) with open(CONF, 'w', encoding='utf-8') as file: for line in data: if line.startswith('authorized_ips='): line = f"{line.rstrip()},127.0.0.1\n" file.write(line) yield os.rename(f"{CONF}.bak", CONF) @pytest.fixture def access_hash(): """Read /root/.accesshash or raise IOError""" with open('/root/.accesshash', encoding='utf-8') as hash_file: hash_lines = hash_file.read().splitlines() return ''.join(hash_lines)
Save
cmd:
run