test_model: add types

Author: deltragon

safeeyes/tests/test_model.py

@@ -16,26 +16,28 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+import pytest
 import random
+import typing
 from safeeyes import model
 
 
 class TestBreak:
-    def test_break_short(self):
+    def test_break_short(self) -> None:
         b = model.Break(model.BreakType.SHORT_BREAK, "test break", 15, 15, None, None)
 
         assert b.is_short_break()
         assert not b.is_long_break()
 
-    def test_break_long(self):
+    def test_break_long(self) -> None:
         b = model.Break(model.BreakType.LONG_BREAK, "long break", 75, 60, None, None)
 
         assert not b.is_short_break()
         assert b.is_long_break()
 
 
 class TestBreakQueue:
-    def test_create_empty(self):
+    def test_create_empty(self) -> None:
         config = {
             "short_breaks": [],
             "long_breaks": [],
@@ -46,7 +48,7 @@ def test_create_empty(self):
             "random_order": False,
         }
 
-        context = {}
+        context: dict[str, typing.Any] = {}
 
         bq = model.BreakQueue(config, context)
 
@@ -56,7 +58,9 @@ def test_create_empty(self):
         assert bq.next() is None
         assert bq.get_break() is None
 
-    def get_bq_only_short(self, monkeypatch, random_seed=None):
+    def get_bq_only_short(
+        self, monkeypatch: pytest.MonkeyPatch, random_seed: typing.Optional[int] = None
+    ) -> model.BreakQueue:
         if random_seed is not None:
             random.seed(random_seed)
 
@@ -78,13 +82,15 @@ def get_bq_only_short(self, monkeypatch, random_seed=None):
             "random_order": random_seed is not None,
         }
 
-        context = {
+        context: dict[str, typing.Any] = {
             "session": {},
         }
 
         return model.BreakQueue(config, context)
 
-    def get_bq_only_long(self, monkeypatch, random_seed=None):
+    def get_bq_only_long(
+        self, monkeypatch: pytest.MonkeyPatch, random_seed: typing.Optional[int] = None
+    ) -> model.BreakQueue:
         if random_seed is not None:
             random.seed(random_seed)
 
@@ -106,13 +112,15 @@ def get_bq_only_long(self, monkeypatch, random_seed=None):
             "random_order": random_seed is not None,
         }
 
-        context = {
+        context: dict[str, typing.Any] = {
             "session": {},
         }
 
         return model.BreakQueue(config, context)
 
-    def get_bq_full(self, monkeypatch, random_seed=None):
+    def get_bq_full(
+        self, monkeypatch: pytest.MonkeyPatch, random_seed: typing.Optional[int] = None
+    ) -> model.BreakQueue:
         if random_seed is not None:
             random.seed(random_seed)
 
@@ -139,20 +147,22 @@ def get_bq_full(self, monkeypatch, random_seed=None):
             "random_order": random_seed is not None,
         }
 
-        context = {
+        context: dict[str, typing.Any] = {
             "session": {},
         }
 
         return model.BreakQueue(config, context)
 
-    def test_create_only_short(self, monkeypatch):
+    def test_create_only_short(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_only_short(monkeypatch)
 
         assert not bq.is_empty()
         assert not bq.is_empty(model.BreakType.SHORT_BREAK)
         assert bq.is_empty(model.BreakType.LONG_BREAK)
 
-    def test_only_short_repeat_get_break_no_change(self, monkeypatch):
+    def test_only_short_repeat_get_break_no_change(
+        self, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
         bq = self.get_bq_only_short(monkeypatch)
 
         next = bq.get_break()
@@ -163,7 +173,7 @@ def test_only_short_repeat_get_break_no_change(self, monkeypatch):
 
         assert not bq.is_long_break()
 
-    def test_only_short_next_break(self, monkeypatch):
+    def test_only_short_next_break(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_only_short(monkeypatch)
 
         next = bq.get_break()
@@ -178,7 +188,9 @@ def test_only_short_next_break(self, monkeypatch):
         assert bq.next().name == "translated!: break 2"
         assert bq.next().name == "translated!: break 3"
 
-    def test_only_short_next_break_random(self, monkeypatch):
+    def test_only_short_next_break_random(
+        self, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
         random_seed = 5
         bq = self.get_bq_only_short(monkeypatch, random_seed)
 
@@ -190,14 +202,16 @@ def test_only_short_next_break_random(self, monkeypatch):
         assert bq.next().name == "translated!: break 3"
         assert bq.next().name == "translated!: break 1"
 
-    def test_create_only_long(self, monkeypatch):
+    def test_create_only_long(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_only_long(monkeypatch)
 
         assert not bq.is_empty()
         assert not bq.is_empty(model.BreakType.LONG_BREAK)
         assert bq.is_empty(model.BreakType.SHORT_BREAK)
 
-    def test_only_long_repeat_get_break_no_change(self, monkeypatch):
+    def test_only_long_repeat_get_break_no_change(
+        self, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
         bq = self.get_bq_only_long(monkeypatch)
 
         next = bq.get_break()
@@ -208,7 +222,7 @@ def test_only_long_repeat_get_break_no_change(self, monkeypatch):
 
         assert bq.is_long_break()
 
-    def test_only_long_next_break(self, monkeypatch):
+    def test_only_long_next_break(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_only_long(monkeypatch)
 
         next = bq.get_break()
@@ -223,7 +237,7 @@ def test_only_long_next_break(self, monkeypatch):
         assert bq.next().name == "translated!: long break 2"
         assert bq.next().name == "translated!: long break 3"
 
-    def test_only_long_next_break_random(self, monkeypatch):
+    def test_only_long_next_break_random(self, monkeypatch: pytest.MonkeyPatch) -> None:
         random_seed = 5
         bq = self.get_bq_only_long(monkeypatch, random_seed)
 
@@ -235,14 +249,16 @@ def test_only_long_next_break_random(self, monkeypatch):
         assert bq.next().name == "translated!: long break 3"
         assert bq.next().name == "translated!: long break 1"
 
-    def test_create_full(self, monkeypatch):
+    def test_create_full(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_full(monkeypatch)
 
         assert not bq.is_empty()
         assert not bq.is_empty(model.BreakType.LONG_BREAK)
         assert not bq.is_empty(model.BreakType.SHORT_BREAK)
 
-    def test_full_repeat_get_break_no_change(self, monkeypatch):
+    def test_full_repeat_get_break_no_change(
+        self, monkeypatch: pytest.MonkeyPatch
+    ) -> None:
         bq = self.get_bq_full(monkeypatch)
 
         next = bq.get_break()
@@ -253,7 +269,7 @@ def test_full_repeat_get_break_no_change(self, monkeypatch):
 
         assert not bq.is_long_break()
 
-    def test_full_next_break(self, monkeypatch):
+    def test_full_next_break(self, monkeypatch: pytest.MonkeyPatch) -> None:
         bq = self.get_bq_full(monkeypatch)
 
         next = bq.get_break()
@@ -297,7 +313,7 @@ def test_full_next_break(self, monkeypatch):
         assert bq.next().name == "translated!: break 4"
         assert bq.next().name == "translated!: long break 1"
 
-    def test_full_next_break_random(self, monkeypatch):
+    def test_full_next_break_random(self, monkeypatch: pytest.MonkeyPatch) -> None:
         random_seed = 5
         bq = self.get_bq_full(monkeypatch, random_seed)