def __init__(self, **kwargs):
""" Initializes environment.
`args`
List of environment arguments. Default: ``None``
`io`
IO object for data streams. Default: ``None``
`data_dir`
Home directory for focus user data. Default: ~/.focus or value
defined in $FOCUS_HOME env var.
`task`
``Task`` instance. Default: ``None``
"""
# argument list
self._args = kwargs.get('args', list())
# io stream
self._io = kwargs.get('io', IOStream())
# path to focus user data directory (for config files, plugins, etc.)
# first: check provided arg, then: $FOCUS_HOME env variable,
# finally: use default user homedir path
self._data_dir = (
kwargs.get('data_dir') or os.environ.get('FOCUS_HOME') or
os.path.expanduser(self.DEF_DATA_DIR) # ~/.focus
)
self._data_dir = os.path.realpath(self._data_dir)
self._task = kwargs.get('task')
self._loaded = False
class TestIOStream(FocusTestCase):
class Mock(object):
def read(self):
return '!!TESTING!! Hello World'
def write(self, s):
self.test__data = s
def setUp(self):
super(TestIOStream, self).setUp()
self.io = IOStream(inputs=self.Mock(),
outputs=self.Mock(),
errors=self.Mock())
def tearDown(self):
self.io = None
super(TestIOStream, self).tearDown()
def test__read(self):
""" IOStream.read: reads data from input stream.
"""
self.assertEqual(self.io.read(), '!!TESTING!! Hello World')
def test__write(self):
""" IOStream.write: writes data to output stream.
"""
testval = 'TESTING...1.2.3!'
self.io.write(testval, newline=False)
self.assertEqual(self.io._output.test__data, testval)
self.io.write(testval, newline=True)
self.assertEqual(self.io._output.test__data, testval + os.linesep)
def test__success(self):
""" IOStream.success: writes success-colored data to output stream.
"""
testval = 'TESTING...1.2.3!'
color_testval = '{0}TESTING...1.2.3!{1}'.format(
self.io.ESCAPE_GREEN, self.io.ESCAPE_CLEAR)
# colored versions
self.io.set_colored(True)
self.io.success(testval, newline=False)
self.assertEqual(self.io._output.test__data, color_testval)
self.io.success(testval, newline=True)
self.assertEqual(self.io._output.test__data,
color_testval + os.linesep)
# non-colored versions
self.io.set_colored(False)
self.io.success(testval, newline=False)
self.assertEqual(self.io._output.test__data, testval)
self.io.success(testval, newline=True)
self.assertEqual(self.io._output.test__data, testval + os.linesep)
def test__error(self):
""" IOStream.error: writes error-colored data to error stream.
"""
testval = 'TESTING...1.2.3!'
color_testval = '{0}TESTING...1.2.3!{1}'.format(
self.io.ESCAPE_RED, self.io.ESCAPE_CLEAR)
# colored versions
self.io.set_colored(True)
self.io.error(testval, newline=False)
self.assertEqual(self.io._error.test__data, color_testval)
self.io.error(testval, newline=True)
self.assertEqual(self.io._error.test__data, color_testval + os.linesep)
# non-colored versions
self.io.set_colored(False)
self.io.error(testval, newline=False)
self.assertEqual(self.io._error.test__data, testval)
self.io.error(testval, newline=True)
self.assertEqual(self.io._error.test__data, testval + os.linesep)
def test__set_colored(self):
""" IOStream.set_colored: enables or disables coloring of output data.
"""
self.io.set_colored(False)
self.assertFalse(self.io._colored)
self.io.set_colored(True)
self.assertTrue(self.io._colored)
self.io.set_colored(0)
self.assertFalse(self.io._colored)
self.io.set_colored(1)
self.assertTrue(self.io._colored)
def test__isatty(self):
""" IOStream.isatty: is input stream a terminal.
"""
# `Dummy` object doesn't have method.. should return False
self.assertFalse(self.io.isatty)
# patch in different `isatty` method versions to test return values
self.io._input.isatty = types.MethodType(lambda self: True,
self.io._input)
self.assertTrue(self.io.isatty)
self.io._input.isatty = types.MethodType(lambda self: False,
self.io._input)
self.assertFalse(self.io.isatty)
def setUp(self):
super(TestIOStream, self).setUp()
self.io = IOStream(inputs=self.Mock(),
outputs=self.Mock(),
errors=self.Mock())
class TestIOStream(FocusTestCase):
class Mock(object):
def read(self):
return '!!TESTING!! Hello World'
def write(self, s):
self.test__data = s
def setUp(self):
super(TestIOStream, self).setUp()
self.io = IOStream(inputs=self.Mock(), outputs=self.Mock(),
errors=self.Mock())
def tearDown(self):
self.io = None
super(TestIOStream, self).tearDown()
def test__read(self):
""" IOStream.read: reads data from input stream.
"""
self.assertEqual(self.io.read(), '!!TESTING!! Hello World')
def test__write(self):
""" IOStream.write: writes data to output stream.
"""
testval = 'TESTING...1.2.3!'
self.io.write(testval, newline=False)
self.assertEqual(self.io._output.test__data, testval)
self.io.write(testval, newline=True)
self.assertEqual(self.io._output.test__data, testval + os.linesep)
def test__success(self):
""" IOStream.success: writes success-colored data to output stream.
"""
testval = 'TESTING...1.2.3!'
color_testval = '{0}TESTING...1.2.3!{1}'.format(self.io.ESCAPE_GREEN,
self.io.ESCAPE_CLEAR)
# colored versions
self.io.set_colored(True)
self.io.success(testval, newline=False)
self.assertEqual(self.io._output.test__data, color_testval)
self.io.success(testval, newline=True)
self.assertEqual(self.io._output.test__data, color_testval +
os.linesep)
# non-colored versions
self.io.set_colored(False)
self.io.success(testval, newline=False)
self.assertEqual(self.io._output.test__data, testval)
self.io.success(testval, newline=True)
self.assertEqual(self.io._output.test__data, testval + os.linesep)
def test__error(self):
""" IOStream.error: writes error-colored data to error stream.
"""
testval = 'TESTING...1.2.3!'
color_testval = '{0}TESTING...1.2.3!{1}'.format(self.io.ESCAPE_RED,
self.io.ESCAPE_CLEAR)
# colored versions
self.io.set_colored(True)
self.io.error(testval, newline=False)
self.assertEqual(self.io._error.test__data, color_testval)
self.io.error(testval, newline=True)
self.assertEqual(self.io._error.test__data, color_testval + os.linesep)
# non-colored versions
self.io.set_colored(False)
self.io.error(testval, newline=False)
self.assertEqual(self.io._error.test__data, testval)
self.io.error(testval, newline=True)
self.assertEqual(self.io._error.test__data, testval + os.linesep)
def test__set_colored(self):
""" IOStream.set_colored: enables or disables coloring of output data.
"""
self.io.set_colored(False)
self.assertFalse(self.io._colored)
self.io.set_colored(True)
self.assertTrue(self.io._colored)
self.io.set_colored(0)
self.assertFalse(self.io._colored)
self.io.set_colored(1)
self.assertTrue(self.io._colored)
def test__isatty(self):
""" IOStream.isatty: is input stream a terminal.
"""
# `Dummy` object doesn't have method.. should return False
self.assertFalse(self.io.isatty)
# patch in different `isatty` method versions to test return values
self.io._input.isatty = types.MethodType(lambda self: True,
self.io._input)
self.assertTrue(self.io.isatty)
self.io._input.isatty = types.MethodType(lambda self: False,
self.io._input)
self.assertFalse(self.io.isatty)
def setUp(self):
super(TestIOStream, self).setUp()
self.io = IOStream(inputs=self.Mock(), outputs=self.Mock(),
errors=self.Mock())