def testRestartInChrootWithChrootArgs(self): """Verify args and chroot args from exception are used.""" # Setup Mocks/Fakes rc = self.StartPatcher(cros_test_lib.RunCommandMock()) rc.SetDefaultCmdResult() def findTarget(_): """ScriptWrapperMain needs a function to find a function to run.""" def raiseChrootRequiredError(_args): raise commandline.ChrootRequiredError(self.CMD_ARGS, self.CHROOT_ARGS) return raiseChrootRequiredError # Run Test commandline.ScriptWrapperMain(findTarget, ['unrelated']) # Verify Results rc.assertCommandContains(enter_chroot=True) rc.assertCommandContains(self.CMD_ARGS) rc.assertCommandContains(chroot_args=self.CHROOT_ARGS)
def testRestartInChrootPreserveArgs(self): """Verify args to ScriptWrapperMain are passed through to chroot..""" # Setup Mocks/Fakes rc = self.StartPatcher(cros_build_lib_unittest.RunCommandMock()) rc.SetDefaultCmdResult() def findTarget(target): """ScriptWrapperMain needs a function to find a function to run.""" def raiseChrootRequiredError(args): raise commandline.ChrootRequiredError(args) self.lastTargetFound = target return raiseChrootRequiredError # Run Test commandline.ScriptWrapperMain(findTarget, self.SYS_ARGV) # Verify Results rc.assertCommandContains(enter_chroot=True) rc.assertCommandContains(self.CMD_ARGS) self.assertEqual('/cmd', self.lastTargetFound)
return issues issues += RunLinters(gnfile, data, settings) return issues def main(argv): parser = GetParser() opts = parser.parse_args(argv) if not opts.files: logging.warning('No files provided to lint. Doing nothing.') return 0 extensions = set(opts.extensions.split(',')) num_files = 0 for gnfile in FilterPaths(opts.files, extensions): logging.debug('Checking %s', gnfile) issues = CheckGnFile(gnfile) if issues: logging.error('**** %s: found %i issue(s)', gnfile, len(issues)) for issue in issues: logging.log(issue.type, '%s: %s', issue.linter, issue.msg) num_files += 1 if num_files: logging.error('%i file(s) failed linting', num_files) return 1 if num_files else 0 if __name__ == '__main__': commandline.ScriptWrapperMain(lambda _: main)
def DoMain(): commandline.ScriptWrapperMain(FindTarget)