def run(arg):
mock_default = mock.Mock()
try:
program.get_reader(['self', arg], mock_default)
assert False, "Should have exited with usage information"
except SystemExit as se:
assert "Usage: " in se.args[0]
assert " self " in se.args[0]
def test_get_reader_help(self):
for arg in ('-h', '--help'):
with self.subTest(arg):
mock_default = mock.Mock()
with self.assertRaises(SystemExit) as cm:
program.get_reader(['self', arg], mock_default)
self.assertIn("Usage: ", cm.exception.args[0])
self.assertIn(" self ", cm.exception.args[0])
def test_get_reader_default(self, open_mock):
mock_default = mock.Mock()
open_mock.return_value = mock.Mock()
reader = program.get_reader(['self'], mock_default)
self.assertIs(mock_default, reader,
'Should have returned the file reader')
self.assertFalse(open_mock.called)
def test_get_reader_default(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = program.get_reader(['self'], mock_default)
assert reader is mock_default, 'Should have returned the file reader'
assert not open_mock.called
def test_get_reader_file(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = program.get_reader(['self', file_name], mock_default)
assert reader is open_return, 'Should have returned the file reader'
open_mock.assert_called_once_with(file_name, 'r')