tests/test-ui-color.py
author Idan Kamara <idankk86@gmail.com>
Thu, 02 Jun 2011 00:43:34 +0300
changeset 14516 842a9179132c
parent 12865 4c50552fc9bc
child 14614 afccc64eea73
permissions -rw-r--r--
color: check if ui is already a subclass of colorui before wrapping it since it's possible to reuse the ui object (see 80c599eee3f3) between dispatch calls, the ui might have been wrapped already. we could probably move this test up a bit but leaving it here allows updates to take effect by extstyles() and configstyles().

import os, sys
from hgext import color
from mercurial import dispatch, ui

# ensure errors aren't buffered
testui = color.colorui()
testui.pushbuffer()
testui.write('buffered\n')
testui.warn('warning\n')
testui.write_err('error\n')
print repr(testui.popbuffer())

# test dispatch.dispatch with the same ui object
hgrc = open(os.environ["HGRCPATH"], 'w')
hgrc.write('[extensions]\n')
hgrc.write('color=\n')
hgrc.close()

ui_ = ui.ui()
ui_.setconfig('ui', 'formatted', 'True')

# call some arbitrary command just so we go through
# color's wrapped _runcommand twice.
# we're not interested in the output, so write that to devnull
def runcmd():
    sys.stdout = open(os.devnull, 'w')
    dispatch.dispatch(dispatch.request(['version', '-q'], ui_))
    sys.stdout = sys.__stdout__

runcmd()
print "colored? " + str(issubclass(ui_.__class__, color.colorui))
runcmd()
print "colored? " + str(issubclass(ui_.__class__, color.colorui))