def test_generate(self): with infra_libs.temporary_directory(prefix="new-tool-test-") as tempdir: toolname = "test_tool" new_tool.generate_tool_files("test_tool", tempdir) self.assertTrue(os.path.isdir(os.path.join(tempdir, toolname))) self.assertTrue(os.path.isfile(os.path.join(tempdir, toolname, "__init__.py"))) self.assertTrue(os.path.isfile(os.path.join(tempdir, toolname, "__main__.py"))) self.assertTrue(os.path.isdir(os.path.join(tempdir, toolname, "test"))) self.assertTrue(os.path.isfile(os.path.join(tempdir, toolname, "test", "__init__.py")))
def main(argv): parser = argparse.ArgumentParser( prog='new_tool', description=sys.modules['__main__'].__doc__) new_tool.add_argparse_options(parser) args = parser.parse_args(argv) return new_tool.generate_tool_files(args.name[0], args.base_dir)
def test_missing_destination_dir(self): with infra_libs.temporary_directory(prefix="new-tool-test-") as tempdir: # When destination directory does not exist, just do nothing and return # a non-zero value. self.assertTrue(new_tool.generate_tool_files("test_tool", os.path.join(tempdir, "missing-directory")))
def test_no_overwrite_tool(self): with infra_libs.temporary_directory(prefix="new-tool-test-") as tempdir: self.assertFalse(new_tool.generate_tool_files("test_tool", tempdir)) self.assertTrue(new_tool.generate_tool_files("test_tool", tempdir))