def run(arg):
mock_default = mock.Mock()
try:
main.get_reader(['self', arg], mock_default)
assert False, "Should have exited with usage information"
except SystemExit as se:
assert "Usage: " in se.message
assert " self " in se.message
def run(arg):
mock_default = mock.Mock()
try:
main.get_reader(['self', arg], mock_default)
assert False, "Should have exited with usage information"
except SystemExit as se:
assert "Usage: " in se.message
assert " self " in se.message
def test_get_reader_default(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = main.get_reader(['self'], mock_default)
assert reader is mock_default, 'Should have returned the file reader'
open_mock.assert_no_calls()
def test_get_reader_file(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = main.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')
def test_get_reader_default(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = main.get_reader(['self'], mock_default)
assert reader is mock_default, 'Should have returned the file reader'
open_mock.assert_no_calls()
def test_get_reader_file(open_mock):
mock_default = mock.Mock()
open_mock.return_value = open_return = mock.Mock()
file_name = 'bees'
reader = main.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')
def register_reader(_context, path, **kwargs):
reader = get_reader(path)
alsoProvides(reader, IDumpReader)
getGlobalSiteManager().registerUtility(reader)
