test-ui-color.py
40 lines
| 988 B
| text/x-python
|
PythonLexer
/ tests / test-ui-color.py
|
r28915 | from __future__ import absolute_import, print_function | ||
|
r19322 | import os | ||
|
r28915 | from hgext import ( | ||
color, | ||||
) | ||||
from mercurial import ( | ||||
dispatch, | ||||
ui as uimod, | ||||
) | ||||
|
r11732 | |||
# ensure errors aren't buffered | ||||
testui = color.colorui() | ||||
testui.pushbuffer() | ||||
|
r17956 | testui.write(('buffered\n')) | ||
testui.warn(('warning\n')) | ||||
|
r11732 | testui.write_err('error\n') | ||
|
r28682 | print(repr(testui.popbuffer())) | ||
|
r14516 | |||
# test dispatch.dispatch with the same ui object | ||||
hgrc = open(os.environ["HGRCPATH"], 'w') | ||||
hgrc.write('[extensions]\n') | ||||
hgrc.write('color=\n') | ||||
hgrc.close() | ||||
|
r30559 | ui_ = uimod.ui.load() | ||
|
r14516 | ui_.setconfig('ui', 'formatted', 'True') | ||
|
r14614 | # we're not interested in the output, so write that to devnull | ||
ui_.fout = open(os.devnull, 'w') | ||||
|
r14516 | # call some arbitrary command just so we go through | ||
# color's wrapped _runcommand twice. | ||||
def runcmd(): | ||||
dispatch.dispatch(dispatch.request(['version', '-q'], ui_)) | ||||
runcmd() | ||||
|
r28682 | print("colored? " + str(issubclass(ui_.__class__, color.colorui))) | ||
|
r14516 | runcmd() | ||
|
r28682 | print("colored? " + str(issubclass(ui_.__class__, color.colorui))) | ||