def test02_should_sub_cmd_object_is_resource():
from schema_sugar.client.parser import Resource
with mock.patch.object(
Resource,
"parse_args",
return_value=Namespace(disks="", list=""),
):
cmd = Resource()
sub = cmd.add_cmd("disks")
assert isinstance(sub, Resource)
def test01_should_func_called():
from schema_sugar.client.parser import Resource
status = {"is_run": False}
def func(**kwargs):
status["is_run"] = True
with mock.patch(
"sys.argv",
["script", "disks"]
):
cmd = Resource()
cmd.add_cmd("disks", func=func)
cmd.run()
assert status['is_run']
