23 lines
619 B
Python
23 lines
619 B
Python
def test_term_error_and_info_format(monkeypatch):
|
|
from certctl import term
|
|
|
|
class Fake:
|
|
def isatty(self):
|
|
return True
|
|
written = ''
|
|
def write(self, s):
|
|
Fake.written += s
|
|
|
|
fake = Fake()
|
|
# error
|
|
e = term.format_error('danger', stream=fake)
|
|
assert '\033[1;31m' in e and '\033[0m' in e
|
|
term.print_error('err', stream=fake)
|
|
assert '\033[1;31m' in Fake.written
|
|
|
|
# info
|
|
i = term.format_info('ok', stream=fake)
|
|
assert '\033[1;32m' in i and '\033[0m' in i
|
|
term.print_info('info', stream=fake)
|
|
assert '\033[1;32m' in Fake.written
|