def Run(self, args):
if args.devsite_dir:
walker_util.DevSiteGenerator(self._cli_power_users_only,
args.devsite_dir).Walk(
args.hidden, args.restrict)
if args.help_text_dir:
walker_util.HelpTextGenerator(
self._cli_power_users_only, args.help_text_dir).Walk(args.hidden,
args.restrict)
if args.html_dir:
walker_util.HtmlGenerator(
self._cli_power_users_only, args.html_dir).Walk(args.hidden,
args.restrict)
tree = walker_util.CommandTreeGenerator(
self._cli_power_users_only).Walk(args.hidden, args.restrict)
with io.open(os.path.join(args.html_dir, '_menu_.html'), 'wt') as out:
WriteHtmlMenu(tree, out)
for file_name in _HELP_HTML_DATA_FILES:
with io.open(os.path.join(args.html_dir, file_name), 'wb') as out:
file_contents = pkg_resources.GetResource(
'googlecloudsdk.api_lib.meta.help_html_data.', file_name)
out.write(file_contents)
if args.manpage_dir:
walker_util.ManPageGenerator(
self._cli_power_users_only, args.manpage_dir).Walk(args.hidden,
args.restrict)
if args.update_help_text_dir:
# The help text golden files are always ascii.
console_attr.ResetConsoleAttr(encoding='ascii')
changes = help_util.HelpTextUpdater(
self._cli_power_users_only, args.update_help_text_dir,
test=args.test).Update(args.restrict)
if changes and args.test:
raise HelpTextOutOfDateError('Help text files must be updated.')
def Generate(kind, generator, directory, encoding='utf8', hidden=False):
"""Runs generator and optionally updates help docs in directory."""
console_attr.ResetConsoleAttr(encoding)
if not args.update:
generator(self._cli_power_users_only, directory).Walk(
hidden, args.restrict)
elif help_util.HelpUpdater(
self._cli_power_users_only, directory, generator,
test=args.test, hidden=hidden).Update(args.restrict):
out_of_date.add(kind)
def testReadySymbolAndColor_utf8(self, ready_status, expected):
console_attr.ResetConsoleAttr("utf-8")
self.assertEqual(
kubernetesobject.KubernetesObject({
"status": {
"conditions": [{
"type": "Ready",
"status": ready_status
}, {
"type": "RoutesReady",
"status": "True"
}]
}
}).ReadySymbolAndColor(), expected)
def testReadySymbol(self, ready_status, expected):
console_attr.ResetConsoleAttr("ascii")
self.assertEqual(
kubernetesobject.KubernetesObject({
"status": {
"conditions": [{
"type": "Ready",
"status": ready_status
}, {
"type": "RoutesReady",
"status": "True"
}]
}
}).ready_symbol, expected)
def testConsoleAttrStateResetTermSize(self):
console_attr.ConsoleAttr._CONSOLE_ATTR_STATE = None
console_attr.GetConsoleAttr()
self.assertEqual(self._get_term_size_windows_mock.call_count, 1)
console_attr.GetConsoleAttr()
self.assertEqual(self._get_term_size_windows_mock.call_count, 1)
console_attr.GetConsoleAttr(encoding='mock')
self.assertEqual(self._get_term_size_windows_mock.call_count, 2)
console_attr.GetConsoleAttr()
self.assertEqual(self._get_term_size_windows_mock.call_count, 2)
console_attr.ResetConsoleAttr()
self.assertEqual(self._get_term_size_windows_mock.call_count, 3)
console_attr.GetConsoleAttr()
self.assertEqual(self._get_term_size_windows_mock.call_count, 3)
def tearDown(self): console_attr.ResetConsoleAttr() super(KubernetesObjectTest, self)
def TearDown(self):
console_attr.ResetConsoleAttr()
def TearDown(self):
console_attr.ResetConsoleAttr()
if self.failures:
self.fail('%d/%d subtests failed:\n%s\n' % (len(self.failures),
self.subtests,
'\n'.join(self.failures)))
def testResetEncodingWin(self):
attr = console_attr.ResetConsoleAttr(encoding='win')
self.assertEqual(attr.GetEncoding(), 'cp437')
attr = console_attr.ResetConsoleAttr(encoding='cp437')
self.assertEqual(attr.GetEncoding(), 'cp437')
def testResetEncodingUtf8(self):
attr = console_attr.ResetConsoleAttr(encoding='utf8')
self.assertEqual(attr.GetEncoding(), 'utf8')
def testResetEncodingAscii(self):
attr = console_attr.ResetConsoleAttr(encoding='ascii')
self.assertEqual(attr.GetEncoding(), 'ascii')
def testColorizerBlueScreenDefaultOutImplicitInit(self):
console_attr.ResetConsoleAttr(encoding='UTF-8')
s = 'Am I blue?'
colorize = console_attr.Colorizer(s, 'blue')
colorize.Render(sys.stdout)
self.AssertOutputEquals('\x1b[34;1m{0}\x1b[39;0m'.format(s))
