add decorator for cli context
This commit is contained in:
@@ -0,0 +1,58 @@
|
||||
import click
|
||||
import yaml
|
||||
|
||||
import dbyaml.cli
|
||||
from dbyaml.cli import pass_context, uses_decorator # ty: ignore[unresolved-import]
|
||||
|
||||
|
||||
def test_uses_decorator_cli():
|
||||
assert uses_decorator("main", "pass_context", dbyaml.cli.__file__)
|
||||
assert uses_decorator("list", "pass_context", dbyaml.cli.__file__)
|
||||
|
||||
|
||||
def test_uses_decorator_wo():
|
||||
@click.command
|
||||
@pass_context
|
||||
def cli_wo(ctx): ...
|
||||
|
||||
assert uses_decorator("cli_wo", "pass_context", __file__)
|
||||
|
||||
|
||||
def test_uses_decorator_wo_long():
|
||||
@click.command
|
||||
@dbyaml.cli.pass_context # ty: ignore[unresolved-attribute]
|
||||
def cli_wh_long(ctx): ...
|
||||
|
||||
assert dbyaml.cli.uses_decorator("cli_wh_long", "pass_context", __file__) # ty: ignore[unresolved-attribute]
|
||||
|
||||
|
||||
def test_uses_decorator_call():
|
||||
@click.command
|
||||
@pass_context()
|
||||
def cli_call(ctx): ...
|
||||
|
||||
assert uses_decorator("cli_call", "pass_context()", __file__)
|
||||
|
||||
|
||||
def test_uses_decorator_obj():
|
||||
@click.command
|
||||
@pass_context(False)
|
||||
def cli_obj(ctx): ...
|
||||
|
||||
assert uses_decorator("cli_obj", "pass_context()", __file__)
|
||||
|
||||
|
||||
def test_uses_decorator_mod():
|
||||
@click.command
|
||||
@pass_context(yaml)
|
||||
def cli_mod(ctx): ...
|
||||
|
||||
assert uses_decorator("cli_mod", "pass_context()", __file__)
|
||||
|
||||
|
||||
def test_uses_decorator_func():
|
||||
@click.command
|
||||
@pass_context(yaml.dump)
|
||||
def cli_func(ctx): ...
|
||||
|
||||
assert uses_decorator("cli_func", "pass_context()", __file__)
|
||||
Reference in New Issue
Block a user