def test_existent_file_fails(self) -> None:
""" Tests nonexistent file throws the right error """
with TemporaryDirectory() as tempdir:
filename = os.path.join(
tempdir,
"delete_this_1{}".format(os.path.basename(__file__)),
)
open(filename, "a").close()
run_cli_test(
self,
[
"gen",
"--name-tpl",
"{name}Config",
"--input-mapping",
"doctrans.pure_utils.simple_types",
"--type",
"class",
"--output-filename",
filename,
],
exception=OSError,
exit_code=2,
output=
"File exists and this is a destructive operation. Delete/move {!r} then rerun."
.format(filename),
)
def test_missing_argument_fails(self) -> None:
""" Tests missing argument throws the right error """
run_cli_test(
self,
["sync", "--truth", "class"],
exit_code=2,
output="--truth must be an existent file. Got: None\n",
)
def test_incorrect_arg_fails(self) -> None:
""" Tests CLI interface failure cases """
run_cli_test(
self,
["sync", "--wrong"],
exit_code=2,
output="the following arguments are required: --truth\n",
)
def test_version(self) -> None:
""" Tests CLI interface gives version """
run_cli_test(
self,
["--version"],
exit_code=0,
output=__version__,
output_checker=lambda output: output[output.rfind(" ") + 1:][:-1],
)
def test_gen_fails(self) -> None:
""" Tests CLI interface failure cases """
run_cli_test(
self,
["gen", "--wrong"],
exit_code=2,
output="the following arguments are required:"
" --name-tpl, --input-mapping, --type, --output-filename/-o\n",
)
def test_sync_properties_fails(self) -> None:
""" Tests CLI interface failure cases """
run_cli_test(
self,
["sync_properties", "--wrong"],
exit_code=2,
output="the following arguments are required:"
" --input-filename, --input-param, --output-filename, --output-param\n",
)
def test_non_existent_file_fails(self) -> None:
""" Tests nonexistent file throws the right error """
with TemporaryDirectory() as tempdir:
filename = os.path.join(
tempdir,
"delete_this_1{}".format(os.path.basename(__file__)),
)
run_cli_test(
self,
[
"sync_properties",
"--input-file",
filename,
"--input-param",
"Foo.g.f",
"--output-file",
filename,
"--output-param",
"f.h",
],
exit_code=2,
output="--input-file must be an existent file. Got: {!r}\n".
format(filename),
)
with TemporaryDirectory() as tempdir:
input_filename = os.path.join(
tempdir,
"input_filename.py",
)
output_filename = os.path.join(
tempdir,
"output_filename.py",
)
open(input_filename, "wt").close()
run_cli_test(
self,
[
"sync_properties",
"--input-file",
input_filename,
"--input-param",
"Foo.g.f",
"--output-file",
output_filename,
"--output-param",
"f.h",
],
exit_code=2,
output="--output-file must be an existent file. Got: {!r}\n".
format(output_filename),
)
def test_missing_argument_fails_insufficient_args(self) -> None:
""" Tests missing argument throws the right error """
with TemporaryDirectory() as tempdir:
filename = os.path.join(
os.path.realpath(tempdir),
"delete_this_2{}".format(os.path.basename(__file__)),
)
with open(filename, "wt") as f:
f.write(class_str)
run_cli_test(
self,
["sync", "--truth", "class", "--class", filename],
exit_code=2,
output=
"Two or more of `--argparse-function`, `--class`, and `--function` must be specified\n",
)
def test_sync_properties(self) -> None:
""" Tests CLI interface gets all the way to the sync_properties call without error """
with TemporaryDirectory() as tempdir:
input_filename = os.path.join(tempdir, "class_.py")
output_filename = os.path.join(tempdir, "method.py")
open(input_filename, "wt").close()
open(output_filename, "wt").close()
with patch("doctrans.__main__.sync_properties", mock_function):
self.assertTrue(
run_cli_test(
self,
[
"sync_properties",
"--input-file",
input_filename,
"--input-param",
"Foo.g.f",
"--output-file",
output_filename,
"--output-param",
"f.h",
],
exit_code=None,
output=None,
), )
def test_args_example1(self) -> None:
""" Tests CLI interface sets namespace correctly """
with TemporaryDirectory() as tempdir:
argparse_filename = os.path.join(
os.path.realpath(tempdir),
"argparse.py",
)
class_filename = os.path.join(
os.path.realpath(tempdir),
"class_.py",
)
method_filename = os.path.join(
os.path.realpath(tempdir),
"method.py",
)
with open(argparse_filename, "wt") as f:
f.write(argparse_func_str)
with open(class_filename, "wt") as f:
f.write(class_str)
with open(method_filename, "wt") as f:
f.write(class_with_method_types_str)
_, args = run_cli_test(
self,
[
"sync",
"--class",
class_filename,
"--class-name",
"ConfigClass",
"--function",
method_filename,
"--function-name",
"train",
"--argparse-function",
argparse_filename,
"--argparse-function-name",
"set_cli_args",
"--truth",
"function",
],
exit_code=None,
output=None,
return_args=True,
)
self.assertListEqual(args.argparse_functions, [argparse_filename])
self.assertListEqual(args.argparse_function_names,
["set_cli_args"])
self.assertListEqual(args.classes, [class_filename])
self.assertListEqual(args.class_names, ["ConfigClass"])
self.assertEqual(args.truth, "function")
def test_gen(self) -> None:
""" Tests CLI interface gets all the way to the gen call without error """
with TemporaryDirectory() as tempdir:
output_filename = os.path.join(tempdir, "classes.py")
with patch("doctrans.__main__.gen", mock_function):
run_cli_test(
self,
[
"gen",
"--name-tpl",
"{name}Config",
"--input-mapping",
"doctrans.pure_utils.simple_types",
"--type",
"class",
"--output-filename",
output_filename,
],
exit_code=None,
output=None,
)
def test_non_existent_file_fails(self) -> None:
""" Tests nonexistent file throws the right error """
with TemporaryDirectory() as tempdir:
filename = os.path.join(
os.path.realpath(tempdir),
"delete_this_1{}".format(os.path.basename(__file__)),
)
run_cli_test(
self,
[
"sync",
"--argparse-function",
filename,
"--class",
filename,
"--truth",
"class",
],
exit_code=2,
output="--truth must be an existent file. Got: {!r}\n".format(
filename),
)
def test_args_example0(self) -> None:
""" Tests CLI interface sets namespace correctly """
with TemporaryDirectory() as tempdir:
filename = os.path.join(
os.path.realpath(tempdir),
"delete_this_0{}".format(os.path.basename(__file__)),
)
with open(filename, "wt") as f:
f.write(class_str)
try:
_, args = run_cli_test(
self,
[
"sync",
"--class",
filename,
"--class-name",
"ConfigClass",
"--argparse-function",
filename,
"--argparse-function-name",
"set_cli_args",
"--truth",
"class",
],
exit_code=None,
output=None,
return_args=True,
)
finally:
if os.path.isfile(filename):
os.remove(filename)
self.assertListEqual(args.argparse_functions, [filename])
self.assertListEqual(args.argparse_function_names, ["set_cli_args"])
self.assertListEqual(args.classes, [filename])
self.assertListEqual(args.class_names, ["ConfigClass"])
self.assertEqual(args.truth, "class")