chore: add Type to test (#35942)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

api/tests/unit_tests/controllers/console/datasets/test_datasets_segments.py

@@ -68,7 +68,7 @@ def _segment():
     )
 
 
-def test_get_segment_with_summary(monkeypatch):
+def test_get_segment_with_summary(monkeypatch: pytest.MonkeyPatch):
     segment = _segment()
     summary = SimpleNamespace(summary_content="summary")
 

api/tests/unit_tests/controllers/console/datasets/test_hit_testing.py

@@ -3,6 +3,7 @@ from unittest.mock import MagicMock, PropertyMock, patch
 
 import pytest
 from flask import Flask
+from pytest_mock import MockerFixture
 from werkzeug.exceptions import NotFound
 
 from controllers.console import console_ns
@@ -35,7 +36,7 @@ def dataset():
 
 
 @pytest.fixture(autouse=True)
-def bypass_decorators(mocker):
+def bypass_decorators(mocker: MockerFixture):
     """Bypass all decorators on the API method."""
     mocker.patch(
         "controllers.console.datasets.hit_testing.setup_required",
@@ -56,7 +57,7 @@ def bypass_decorators(mocker):
 
 
 class TestHitTestingApi:
-    def test_hit_testing_success(self, app, dataset, dataset_id):
+    def test_hit_testing_success(self, app: Flask, dataset, dataset_id):
         api = HitTestingApi()
         method = unwrap(api.post)
 
@@ -99,7 +100,7 @@ class TestHitTestingApi:
         assert "records" in result
         assert result["records"] == []
 
-    def test_hit_testing_success_with_optional_record_fields(self, app, dataset, dataset_id):
+    def test_hit_testing_success_with_optional_record_fields(self, app: Flask, dataset, dataset_id):
         api = HitTestingApi()
         method = unwrap(api.post)
 
@@ -150,7 +151,7 @@ class TestHitTestingApi:
         assert result["query"] == payload["query"]
         assert result["records"] == records
 
-    def test_hit_testing_dataset_not_found(self, app, dataset_id):
+    def test_hit_testing_dataset_not_found(self, app: Flask, dataset_id):
         api = HitTestingApi()
         method = unwrap(api.post)
 
@@ -175,7 +176,7 @@ class TestHitTestingApi:
             with pytest.raises(NotFound, match="Dataset not found"):
                 method(api, dataset_id)
 
-    def test_hit_testing_invalid_args(self, app, dataset, dataset_id):
+    def test_hit_testing_invalid_args(self, app: Flask, dataset, dataset_id):
         api = HitTestingApi()
         method = unwrap(api.post)
 

api/tests/unit_tests/controllers/console/datasets/test_metadata.py

@@ -3,6 +3,7 @@ from unittest.mock import MagicMock, PropertyMock, patch
 
 import pytest
 from flask import Flask
+from pytest_mock import MockerFixture
 from werkzeug.exceptions import NotFound
 
 from controllers.console import console_ns
@@ -60,7 +61,7 @@ def metadata_id():
 
 
 @pytest.fixture(autouse=True)
-def bypass_decorators(mocker):
+def bypass_decorators(mocker: MockerFixture):
     """Bypass setup/login/license decorators."""
     mocker.patch(
         "controllers.console.datasets.metadata.setup_required",

api/tests/unit_tests/controllers/console/datasets/test_website.py

@@ -2,6 +2,7 @@ from unittest.mock import Mock, PropertyMock, patch
 
 import pytest
 from flask import Flask
+from pytest_mock import MockerFixture
 
 from controllers.console import console_ns
 from controllers.console.datasets.error import WebsiteCrawlError
@@ -31,7 +32,7 @@ def app():
 
 
 @pytest.fixture(autouse=True)
-def bypass_auth_and_setup(mocker):
+def bypass_auth_and_setup(mocker: MockerFixture):
     """Bypass setup/login/account decorators."""
     mocker.patch(
         "controllers.console.datasets.website.login_required",
@@ -48,7 +49,7 @@ def bypass_auth_and_setup(mocker):
 
 
 class TestWebsiteCrawlApi:
-    def test_crawl_success(self, app, mocker):
+    def test_crawl_success(self, app, mocker: MockerFixture):
         api = WebsiteCrawlApi()
         method = unwrap(api.post)
 
@@ -85,7 +86,7 @@ class TestWebsiteCrawlApi:
         assert status == 200
         assert result["job_id"] == "job-1"
 
-    def test_crawl_invalid_payload(self, app, mocker):
+    def test_crawl_invalid_payload(self, app, mocker: MockerFixture):
         api = WebsiteCrawlApi()
         method = unwrap(api.post)
 
@@ -113,7 +114,7 @@ class TestWebsiteCrawlApi:
             with pytest.raises(WebsiteCrawlError, match="invalid payload"):
                 method(api)
 
-    def test_crawl_service_error(self, app, mocker):
+    def test_crawl_service_error(self, app, mocker: MockerFixture):
         api = WebsiteCrawlApi()
         method = unwrap(api.post)
 
@@ -150,7 +151,7 @@ class TestWebsiteCrawlApi:
 
 
 class TestWebsiteCrawlStatusApi:
-    def test_get_status_success(self, app, mocker):
+    def test_get_status_success(self, app, mocker: MockerFixture):
         api = WebsiteCrawlStatusApi()
         method = unwrap(api.get)
 
@@ -181,7 +182,7 @@ class TestWebsiteCrawlStatusApi:
         assert status == 200
         assert result["status"] == "completed"
 
-    def test_get_status_invalid_provider(self, app, mocker):
+    def test_get_status_invalid_provider(self, app, mocker: MockerFixture):
         api = WebsiteCrawlStatusApi()
         method = unwrap(api.get)
 
@@ -203,7 +204,7 @@ class TestWebsiteCrawlStatusApi:
             with pytest.raises(WebsiteCrawlError, match="invalid provider"):
                 method(api, job_id)
 
-    def test_get_status_service_error(self, app, mocker):
+    def test_get_status_service_error(self, app, mocker: MockerFixture):
         api = WebsiteCrawlStatusApi()
         method = unwrap(api.get)
 

api/tests/unit_tests/controllers/console/datasets/test_wraps.py

@@ -1,6 +1,7 @@
 from unittest.mock import Mock
 
 import pytest
+from pytest_mock import MockerFixture
 
 from controllers.console.datasets.error import PipelineNotFoundError
 from controllers.console.datasets.wraps import get_rag_pipeline
@@ -16,7 +17,7 @@ class TestGetRagPipeline:
         with pytest.raises(ValueError, match="missing pipeline_id"):
             dummy_view()
 
-    def test_pipeline_not_found(self, mocker):
+    def test_pipeline_not_found(self, mocker: MockerFixture):
         @get_rag_pipeline
         def dummy_view(**kwargs):
             return "ok"
@@ -34,7 +35,7 @@ class TestGetRagPipeline:
         with pytest.raises(PipelineNotFoundError):
             dummy_view(pipeline_id="pipeline-1")
 
-    def test_pipeline_found_and_injected(self, mocker):
+    def test_pipeline_found_and_injected(self, mocker: MockerFixture):
         pipeline = Mock(spec=Pipeline)
         pipeline.id = "pipeline-1"
         pipeline.tenant_id = "tenant-1"
@@ -57,7 +58,7 @@ class TestGetRagPipeline:
 
         assert result is pipeline
 
-    def test_pipeline_id_removed_from_kwargs(self, mocker):
+    def test_pipeline_id_removed_from_kwargs(self, mocker: MockerFixture):
         pipeline = Mock(spec=Pipeline)
 
         @get_rag_pipeline
@@ -79,7 +80,7 @@ class TestGetRagPipeline:
 
         assert result == "ok"
 
-    def test_pipeline_id_cast_to_string(self, mocker):
+    def test_pipeline_id_cast_to_string(self, mocker: MockerFixture):
         pipeline = Mock(spec=Pipeline)
 
         @get_rag_pipeline