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 register_reader(_context, path, **kwargs): reader = get_reader(path) alsoProvides(reader, IDumpReader) getGlobalSiteManager().registerUtility(reader)