36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
import typing as t
|
|
|
|
import click.testing
|
|
|
|
from tests.helpers import TestContext, temporary_root
|
|
from tutor.commands.cli import cli
|
|
|
|
|
|
class TestCommandMixin:
|
|
"""
|
|
Run CLI tests in an isolated test root.
|
|
"""
|
|
|
|
@staticmethod
|
|
def invoke(args: t.List[str]) -> click.testing.Result:
|
|
with temporary_root() as root:
|
|
return TestCommandMixin.invoke_in_root(root, args)
|
|
|
|
@staticmethod
|
|
def invoke_in_root(root: str, args: t.List[str]) -> click.testing.Result:
|
|
"""
|
|
Use this method for commands that all need to run in the same root:
|
|
|
|
with temporary_root() as root:
|
|
result1 = self.invoke_in_root(root, ...)
|
|
result2 = self.invoke_in_root(root, ...)
|
|
"""
|
|
runner = click.testing.CliRunner(
|
|
env={
|
|
"TUTOR_ROOT": root,
|
|
"TUTOR_IGNORE_ENTRYPOINT_PLUGINS": "1",
|
|
"TUTOR_IGNORE_DICT_PLUGINS": "1",
|
|
}
|
|
)
|
|
return runner.invoke(cli, args, obj=TestContext(root))
|