mirror of
https://github.com/valitydev/salt.git
synced 2024-11-06 16:45:27 +00:00
Tests wait timer is now colored(option to disable colors is also available), improves readability.
This commit is contained in:
parent
c6b00c59c1
commit
04146f39f9
@ -22,6 +22,7 @@ import salt.config
|
||||
import salt.master
|
||||
import salt.minion
|
||||
import salt.runner
|
||||
from salt.utils import get_colors
|
||||
from salt.utils.verify import verify_env
|
||||
from saltunittest import TestCase
|
||||
|
||||
@ -130,7 +131,7 @@ class TestDaemon(object):
|
||||
|
||||
def __init__(self, opts=None):
|
||||
self.opts = opts
|
||||
self.clean = opts.clean
|
||||
self.colors = get_colors(opts.no_colors is False)
|
||||
|
||||
def __enter__(self):
|
||||
'''
|
||||
@ -325,7 +326,7 @@ class TestDaemon(object):
|
||||
'''
|
||||
Clean out the tmp files
|
||||
'''
|
||||
if not self.clean:
|
||||
if not self.opts.clean:
|
||||
return
|
||||
if os.path.isdir(self.sub_minion_opts['root_dir']):
|
||||
shutil.rmtree(self.sub_minion_opts['root_dir'])
|
||||
@ -346,9 +347,10 @@ class TestDaemon(object):
|
||||
print
|
||||
return True
|
||||
sys.stdout.write(
|
||||
' * [Quit in {0}] Waiting for {1}'.format(
|
||||
' * {YELLOW}[Quit in {0}]{ENDC} Waiting for {1}'.format(
|
||||
'{0}'.format(expire - now).rsplit('.', 1)[0],
|
||||
', '.join(running)
|
||||
', '.join(running),
|
||||
**self.colors
|
||||
)
|
||||
)
|
||||
sys.stdout.flush()
|
||||
|
@ -228,6 +228,10 @@ def parse_opts():
|
||||
action='store_true',
|
||||
help='Print some system information.'
|
||||
)
|
||||
parser.add_option('--no-colors',
|
||||
default=False,
|
||||
action='store_true',
|
||||
help='Disable colour printing.')
|
||||
|
||||
options, _ = parser.parse_args()
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user