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']
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)