PyInventory/inventory_project/tests/test_project_setup.py

74 wiersze
1.9 KiB
Python

import os
import shutil
import subprocess
from pathlib import Path
from creole.setup_utils import update_rst_readme
import inventory
PACKAGE_ROOT = Path(inventory.__file__).parent.parent
def assert_file_contains_string(file_path, string):
with file_path.open('r') as f:
for line in f:
if string in line:
return
raise AssertionError(f'File {file_path} does not contain {string!r} !')
def test_version(package_root=None, version=None):
if package_root is None:
package_root = PACKAGE_ROOT
if version is None:
version = inventory.__version__
if 'dev' not in version and 'rc' not in version:
version_string = f'v{version}'
assert_file_contains_string(
file_path=Path(package_root, 'README.creole'),
string=version_string
)
assert_file_contains_string(
file_path=Path(package_root, 'README.rst'),
string=version_string
)
assert_file_contains_string(
file_path=Path(package_root, 'pyproject.toml'),
string=f'version = "{version}"'
)
def test_poetry_check(package_root=None):
if package_root is None:
package_root = PACKAGE_ROOT
poerty_bin = shutil.which('poetry')
output = subprocess.check_output(
[poerty_bin, 'check'],
universal_newlines=True,
env=os.environ,
stderr=subprocess.STDOUT,
cwd=str(package_root),
)
print(output)
assert output == 'All set!\n'
def test_update_rst_readme(capsys):
rest_readme_path = update_rst_readme(
package_root=PACKAGE_ROOT, filename='README.creole'
)
captured = capsys.readouterr()
assert captured.out == 'Generate README.rst from README.creole...nothing changed, ok.\n'
assert captured.err == ''
assert isinstance(rest_readme_path, Path)
assert str(rest_readme_path).endswith('/README.rst')