summaryrefslogtreecommitdiff
path: root/tests/unit/test_exceptions.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/test_exceptions.py')
-rw-r--r--tests/unit/test_exceptions.py50
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)