Update test_core.py

AndreiDrang · Dec 8, 2023 · e2fc51b · e2fc51b
1 parent 03f2d45
commit e2fc51b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -4,7 +4,7 @@
 
 from tests.conftest import BaseTest
 from python_rucaptcha.core.base import BaseCaptcha
-from python_rucaptcha.core.enums import MyEnum, ControlEnm
+from python_rucaptcha.core.enums import MyEnum, ControlEnm, ServiceEnm
 from python_rucaptcha.core.config import RETRIES, ASYNC_RETRIES, attempts_generator
 
 
@@ -104,3 +104,10 @@ def test_attempts_generator(self):
         for attempt in attempts:
             assert isinstance(attempt, int)
         assert attempt == 4
+
+
+class TestDeathbycaptcha(BaseTest):
+    def test_attempts_generator(self):
+        BaseCaptcha(
+            rucaptcha_key=self.RUCAPTCHA_KEY, service_type=ServiceEnm.DEATHBYCAPTCHA, method=ControlEnm.control.value
+        )