diff options
Diffstat (limited to 'tests/unit/test_exceptions.py')
-rw-r--r-- | tests/unit/test_exceptions.py | 50 |
1 files changed, 22 insertions, 28 deletions
diff --git a/tests/unit/test_exceptions.py b/tests/unit/test_exceptions.py index 514b9ce1f..fb2834058 100644 --- a/tests/unit/test_exceptions.py +++ b/tests/unit/test_exceptions.py @@ -1,42 +1,36 @@ # SPDX-License-Identifier: AGPL-3.0-or-later # pylint: disable=missing-module-docstring +from parameterized import parameterized from tests import SearxTestCase import searx.exceptions from searx import get_setting class TestExceptions(SearxTestCase): # pylint: disable=missing-class-docstring - def test_default_suspend_time(self): - with self.assertRaises(searx.exceptions.SearxEngineAccessDeniedException) as e: - raise searx.exceptions.SearxEngineAccessDeniedException() + @parameterized.expand( + [ + searx.exceptions.SearxEngineAccessDeniedException, + searx.exceptions.SearxEngineCaptchaException, + searx.exceptions.SearxEngineTooManyRequestsException, + ] + ) + def test_default_suspend_time(self, exception): + with self.assertRaises(exception) as e: + raise exception() self.assertEqual( e.exception.suspended_time, - get_setting(searx.exceptions.SearxEngineAccessDeniedException.SUSPEND_TIME_SETTING), + get_setting(exception.SUSPEND_TIME_SETTING), ) - with self.assertRaises(searx.exceptions.SearxEngineCaptchaException) as e: - raise searx.exceptions.SearxEngineCaptchaException() - self.assertEqual( - e.exception.suspended_time, get_setting(searx.exceptions.SearxEngineCaptchaException.SUSPEND_TIME_SETTING) - ) - - with self.assertRaises(searx.exceptions.SearxEngineTooManyRequestsException) as e: - raise searx.exceptions.SearxEngineTooManyRequestsException() - self.assertEqual( - e.exception.suspended_time, - get_setting(searx.exceptions.SearxEngineTooManyRequestsException.SUSPEND_TIME_SETTING), - ) - - def test_custom_suspend_time(self): - with self.assertRaises(searx.exceptions.SearxEngineAccessDeniedException) as e: - raise searx.exceptions.SearxEngineAccessDeniedException(suspended_time=1337) + @parameterized.expand( + [ + searx.exceptions.SearxEngineAccessDeniedException, + searx.exceptions.SearxEngineCaptchaException, + searx.exceptions.SearxEngineTooManyRequestsException, + ] + ) + def test_custom_suspend_time(self, exception): + with self.assertRaises(exception) as e: + raise exception(suspended_time=1337) self.assertEqual(e.exception.suspended_time, 1337) - - with self.assertRaises(searx.exceptions.SearxEngineCaptchaException) as e: - raise searx.exceptions.SearxEngineCaptchaException(suspended_time=1409) - self.assertEqual(e.exception.suspended_time, 1409) - - with self.assertRaises(searx.exceptions.SearxEngineTooManyRequestsException) as e: - raise searx.exceptions.SearxEngineTooManyRequestsException(suspended_time=1543) - self.assertEqual(e.exception.suspended_time, 1543) |