flake8/tests/unit/test_exceptions.py
2021-04-07 08:28:11 -07:00

34 lines
1.1 KiB
Python

"""Tests for the flake8.exceptions module."""
import pickle
import pytest
from flake8 import exceptions
@pytest.mark.parametrize(
'err',
(
exceptions.FailedToLoadPlugin(
plugin_name='plugin_name',
exception=ValueError('boom!'),
),
exceptions.InvalidSyntax(exception=ValueError('Unexpected token: $')),
exceptions.PluginRequestedUnknownParameters(
plugin={'plugin_name': 'plugin_name'},
exception=ValueError('boom!'),
),
exceptions.PluginExecutionFailed(
plugin={'plugin_name': 'plugin_name'},
exception=ValueError('boom!'),
)
),
)
def test_pickleable(err):
"""Ensure that our exceptions can cross pickle boundaries."""
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
new_err = pickle.loads(pickle.dumps(err, protocol=proto))
assert str(err) == str(new_err)
orig_e = err.original_exception
new_e = new_err.original_exception
assert (type(orig_e), orig_e.args) == (type(new_e), new_e.args)