Files
asciinema/tests/test_helper.py

52 lines
1.0 KiB
Python
Raw Normal View History

import sys
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
stdout = None
def assert_printed(expected):
success = expected in stdout.getvalue()
assert success, 'expected text "%s" not printed' % expected
def assert_not_printed(expected):
success = expected not in stdout.getvalue()
assert success, 'not expected text "%s" printed' % expected
class Test(object):
def setUp(self):
global stdout
self.real_stdout = sys.stdout
sys.stdout = stdout = StringIO()
def tearDown(self):
sys.stdout = self.real_stdout
class FakeClock(object):
def __init__(self, values):
self.values = values
self.n = 0
def time(self):
value = self.values[self.n]
self.n += 1
return value
class FakeAsciicast(object):
def __init__(self, cmd=None, title=None, stdout=None, meta_data=None):
self.cmd = cmd
self.title = title
self.stdout = stdout
self.meta_data = meta_data or {}