Update tests

alphatwirl · Jul 7, 2024 · 1d99189 · 1d99189
1 parent 835815f
commit 1d99189
Show file tree

Hide file tree

Showing 2 changed files with 81 additions and 13 deletions.
diff --git a/tests/presentation/test_base.py b/tests/presentation/test_base.py
@@ -35,17 +35,17 @@ def test_repr(obj: MockProgressBar) -> None:
 def test_present(obj: Presentation) -> None:
     i = uuid.uuid4()
     j = uuid.uuid4()
-    obj.present(dict(task_id=i, last=False))
+    obj.present(dict(task_id=i, last=False))  # type: ignore
     assert obj.active()
-    obj.present(dict(task_id=i, last=False))
+    obj.present(dict(task_id=i, last=False))  # type: ignore
     assert obj.active()
-    obj.present(dict(task_id=j, last=False))
+    obj.present(dict(task_id=j, last=False))  # type: ignore
     assert obj.active()
-    obj.present(dict(task_id=j, last=False))
+    obj.present(dict(task_id=j, last=False))  # type: ignore
     assert obj.active()
-    obj.present(dict(task_id=i, last=True))
+    obj.present(dict(task_id=i, last=True))  # type: ignore
     assert obj.active()
-    obj.present(dict(task_id=j, last=True))
+    obj.present(dict(task_id=j, last=True))  # type: ignore
     assert not obj.active()
 
 

diff --git a/tests/progress_report/test_reporter.py b/tests/progress_report/test_reporter.py
@@ -46,7 +46,14 @@ def test_report_need_to_report(
         task_id = uuid.uuid4()
         mock_time.time.return_value = current_time
         monkeypatch.setattr(obj, '_need_to_report', mock.Mock(return_value=True))
-        report = Report(task_id=task_id, done=0, total=10)
+        report = Report(
+            task_id=task_id,
+            name='',
+            done=0,
+            total=10,
+            first=True,
+            last=False,
+        )
         obj.report(report)
         assert [mock.call(report)] == mock_queue.put.call_args_list
         assert {task_id: current_time} == obj.last_time
@@ -62,7 +69,14 @@ def test_report_no_need_to_report(
         task_id = uuid.uuid4()
         mock_time.time.return_value = current_time
         monkeypatch.setattr(obj, '_need_to_report', mock.Mock(return_value=False))
-        report = Report(task_id=task_id, done=0, total=10)
+        report = Report(
+            task_id=task_id,
+            name='',
+            done=0,
+            total=10,
+            first=True,
+            last=False,
+        )
         obj.report(report)
         assert [] == mock_queue.put.call_args_list
         assert {} == obj.last_time
@@ -71,16 +85,70 @@ def test_report_no_need_to_report(
 task_id = uuid.uuid4()
 
 params = [
-    pytest.param(Report(task_id=task_id, first=True, last=False), {}, 10.0, True),
-    pytest.param(Report(task_id=task_id, first=False, last=True), {}, 10.0, True),
     pytest.param(
-        Report(task_id=task_id, first=True, last=False), {task_id: 10.0}, 10.0, True
+        Report(
+            task_id=task_id,
+            name='',
+            done=0,
+            total=10,
+            first=True,
+            last=False,
+        ),
+        {},
+        10.0,
+        True,
     ),
     pytest.param(
-        Report(task_id=task_id, first=False, last=False), {task_id: 10.0}, 30.0, True
+        Report(
+            task_id=task_id,
+            name='',
+            done=10,
+            total=10,
+            first=False,
+            last=True,
+        ),
+        {},
+        10.0,
+        True,
     ),
     pytest.param(
-        Report(task_id=task_id, first=False, last=False), {task_id: 10.0}, 15.0, False
+        Report(
+            task_id=task_id,
+            name='',
+            done=0,
+            total=10,
+            first=True,
+            last=False,
+        ),
+        {task_id: 10.0},
+        10.0,
+        True,
+    ),
+    pytest.param(
+        Report(
+            task_id=task_id,
+            name='',
+            done=5,
+            total=10,
+            first=False,
+            last=False,
+        ),
+        {task_id: 10.0},
+        30.0,
+        True,
+    ),
+    pytest.param(
+        Report(
+            task_id=task_id,
+            name='',
+            done=5,
+            total=10,
+            first=False,
+            last=False,
+        ),
+        {task_id: 10.0},
+        15.0,
+        False,
     ),
 ]
 param_names = 'report, last_time, current_time, expected'