Example #1
0
 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")))
Example #2
0
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)
Example #3
0
 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")))
Example #4
0
 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))