test(queries): add unit tests for query modules (#1609)

* test(queries): add unit tests for query modules

- Add tests for build, checkout, issues and test query modules

Part of #1498

* test(queries): add unit tests for remaining query modules

- Add tests for hardware, notifications and tree modules

Closes #1498

Author: joao-faria-dev

backend/kernelCI_app/tests/unitTests/queries/build_test.py

@@ -0,0 +1,48 @@
+from unittest.mock import patch
+
+from kernelCI_app.queries.build import get_build_details, get_build_tests
+
+from kernelCI_app.tests.unitTests.queries.conftest import (
+    setup_mock_filter_values_queryset,
+    setup_mock_query_builder,
+)
+
+
+class TestGetBuildDetails:
+    @patch("kernelCI_app.queries.build.Query")
+    def test_get_build_details_success(self, mock_query_class):
+        expected_result = [{"id": "build", "checkout_id": "checkout"}]
+        mock_query = setup_mock_query_builder(mock_query_class, expected_result)
+
+        result = get_build_details("build")
+
+        assert result == expected_result
+        mock_query.where.assert_called_once_with(**{"builds.id__eq": "build"})
+
+    @patch("kernelCI_app.queries.build.Query")
+    def test_get_build_details_empty_result(self, mock_query_class):
+        setup_mock_query_builder(mock_query_class, [])
+
+        result = get_build_details("build")
+
+        assert result == []
+
+
+class TestGetBuildTests:
+    @patch("kernelCI_app.queries.build.Tests")
+    def test_get_build_tests_success(self, mock_tests_model):
+        expected_result = [{"id": "test", "status": "PASS"}]
+        setup_mock_filter_values_queryset(mock_tests_model, expected_result)
+
+        result = get_build_tests("build")
+
+        assert result == expected_result
+        mock_tests_model.objects.filter.assert_called_once_with(build_id="build")
+
+    @patch("kernelCI_app.queries.build.Tests")
+    def test_get_build_tests_empty_result(self, mock_tests_model):
+        setup_mock_filter_values_queryset(mock_tests_model, [])
+
+        result = get_build_tests("build")
+
+        assert result == []

backend/kernelCI_app/tests/unitTests/queries/checkout_test.py

@@ -0,0 +1,52 @@
+from unittest.mock import patch
+
+from kernelCI_app.queries.checkout import get_origins
+
+from kernelCI_app.tests.unitTests.queries.conftest import setup_mock_cursor
+
+
+class TestGetOrigins:
+    @patch("kernelCI_app.queries.checkout.get_query_cache")
+    def test_get_origins_from_cache(self, mock_get_cache):
+        cached_data = [{"origin": "maestro", "table": "checkouts"}]
+        mock_get_cache.return_value = cached_data
+
+        result = get_origins(7)
+
+        assert result == cached_data
+
+    @patch("kernelCI_app.queries.checkout.get_query_cache")
+    @patch("kernelCI_app.queries.checkout.set_query_cache")
+    @patch("kernelCI_app.queries.checkout.dict_fetchall")
+    @patch("kernelCI_app.queries.checkout.connection")
+    def test_get_origins_from_database(
+        self, mock_connection, mock_dict_fetchall, mock_set_cache, mock_get_cache
+    ):
+        mock_get_cache.return_value = None
+        expected_result = [{"origin": "maestro", "table": "checkouts"}]
+        mock_dict_fetchall.return_value = expected_result
+
+        mock_cursor = setup_mock_cursor(mock_connection)
+
+        result = get_origins(7)
+
+        assert result == expected_result
+        mock_cursor.execute.assert_called_once()
+        mock_set_cache.assert_called_once()
+
+    @patch("kernelCI_app.queries.checkout.get_query_cache")
+    @patch("kernelCI_app.queries.checkout.set_query_cache")
+    @patch("kernelCI_app.queries.checkout.dict_fetchall")
+    @patch("kernelCI_app.queries.checkout.connection")
+    def test_get_origins_empty_result_does_not_cache(
+        self, mock_connection, mock_dict_fetchall, mock_set_cache, mock_get_cache
+    ):
+        mock_get_cache.return_value = None
+        mock_dict_fetchall.return_value = []
+
+        setup_mock_cursor(mock_connection)
+
+        result = get_origins(7)
+
+        assert result == []
+        mock_set_cache.assert_not_called()

backend/kernelCI_app/tests/unitTests/queries/conftest.py

@@ -0,0 +1,63 @@
+from unittest.mock import MagicMock, Mock
+
+from kernelCI_app.typeModels.hardwareDetails import Tree
+
+
+def setup_mock_cursor(mock_connection):
+    mock_cursor = MagicMock()
+    mock_connection.cursor.return_value.__enter__.return_value = mock_cursor
+    return mock_cursor
+
+
+def setup_mock_queryset(mock_model, return_value):
+    mock_queryset = Mock()
+    mock_queryset.values.return_value = mock_queryset
+    mock_queryset.filter.return_value = mock_queryset
+    mock_queryset.order_by.return_value = mock_queryset
+    mock_queryset.first.return_value = return_value
+    mock_model.objects.values.return_value = mock_queryset
+    return mock_queryset
+
+
+def setup_mock_test_queryset(mock_tests_model):
+    mock_queryset = Mock()
+    mock_queryset.values.return_value = mock_queryset
+    mock_queryset.filter.return_value = mock_queryset
+    mock_queryset.order_by.return_value = mock_queryset
+    mock_sliced = Mock()
+    mock_sliced.__iter__ = Mock(return_value=iter([]))
+    mock_queryset.__getitem__ = Mock(return_value=mock_sliced)
+    mock_tests_model.objects.filter.return_value = mock_queryset
+    return mock_queryset
+
+
+TEST_TREE = Tree(
+    index="0",
+    tree_name="mainline",
+    origin="maestro",
+    git_repository_branch="master",
+    git_repository_url="https://my_url.com",
+    head_git_commit_name="v6.1",
+    head_git_commit_hash="abc123",
+    head_git_commit_tag=None,
+    selected_commit_status=None,
+    is_selected=None,
+)
+
+
+def setup_mock_query_builder(mock_query_class, return_value):
+    mock_query = Mock()
+    mock_query.from_table.return_value = mock_query
+    mock_query.join.return_value = mock_query
+    mock_query.where.return_value = mock_query
+    mock_query.select.return_value = return_value
+    mock_query_class.return_value = mock_query
+    return mock_query
+
+
+def setup_mock_filter_values_queryset(mock_model, return_value):
+    mock_filter_queryset = Mock()
+    mock_filter_queryset.annotate.return_value = mock_filter_queryset
+    mock_filter_queryset.values.return_value = return_value
+    mock_model.objects.filter.return_value = mock_filter_queryset
+    return mock_filter_queryset

backend/kernelCI_app/tests/unitTests/queries/hardware_queries_test.py

@@ -0,0 +1,224 @@
+from datetime import datetime
+from unittest.mock import patch
+
+from kernelCI_app.queries.hardware import (
+    get_hardware_listing_data,
+    get_hardware_details_data,
+    get_hardware_trees_data,
+    get_hardware_commit_history,
+    _generate_query_params,
+    query_records,
+)
+from kernelCI_app.typeModels.hardwareDetails import CommitHead
+
+from kernelCI_app.tests.unitTests.queries.conftest import (
+    TEST_TREE,
+    setup_mock_cursor,
+)
+
+START_DATE = datetime(2025, 11, 11)
+END_DATE = datetime(2025, 11, 12)
+
+
+class TestGetHardwareListingData:
+    @patch("kernelCI_app.queries.hardware.connection")
+    def test_get_hardware_listing_data_success(self, mock_connection):
+        expected_result = [
+            ("platform", ["compatible"], 10, 5, 0, 0, 0, 0, 0, 20, 10, 0, 0, 0, 0, 0)
+        ]
+        mock_cursor = setup_mock_cursor(mock_connection)
+        mock_cursor.fetchall.return_value = expected_result
+
+        result = get_hardware_listing_data(
+            start_date=datetime(2025, 11, 10),
+            end_date=datetime(2025, 11, 12),
+            origin="maestro",
+        )
+
+        assert result == expected_result
+        mock_cursor.execute.assert_called_once()
+
+
+class TestGetHardwareDetailsData:
+    @patch("kernelCI_app.queries.hardware.get_query_cache")
+    def test_get_hardware_details_data_from_cache(self, mock_get_cache):
+        cached_data = [{"id": "test", "status": "PASS"}]
+        mock_get_cache.return_value = cached_data
+
+        result = get_hardware_details_data(
+            hardware_id="hardware",
+            origin="maestro",
+            trees_with_selected_commits=[TEST_TREE],
+            start_datetime=START_DATE,
+            end_datetime=END_DATE,
+        )
+
+        assert result == cached_data
+
+    @patch("kernelCI_app.queries.hardware.get_query_cache")
+    @patch("kernelCI_app.queries.hardware.set_query_cache")
+    @patch("kernelCI_app.queries.hardware.query_records")
+    def test_get_hardware_details_data_from_database(
+        self, mock_query_records, mock_set_cache, mock_get_cache
+    ):
+        expected_data = [{"id": "test", "status": "PASS"}]
+        mock_get_cache.return_value = None
+        mock_query_records.return_value = expected_data
+
+        result = get_hardware_details_data(
+            hardware_id="hardware",
+            origin="maestro",
+            trees_with_selected_commits=[TEST_TREE],
+            start_datetime=START_DATE,
+            end_datetime=END_DATE,
+        )
+
+        assert result == expected_data
+        mock_query_records.assert_called_once()
+        mock_set_cache.assert_called_once()
+
+
+class TestGetHardwareTreesData:
+    @patch("kernelCI_app.queries.hardware.get_query_cache")
+    def test_get_hardware_trees_data_from_cache(self, mock_get_cache):
+        cached_trees = [TEST_TREE]
+        mock_get_cache.return_value = cached_trees
+
+        result = get_hardware_trees_data(
+            hardware_id="hardware",
+            origin="maestro",
+            start_datetime=START_DATE,
+            end_datetime=END_DATE,
+        )
+
+        assert result == cached_trees
+
+    @patch("kernelCI_app.queries.hardware.get_query_cache")
+    @patch("kernelCI_app.queries.hardware.set_query_cache")
+    @patch("kernelCI_app.queries.hardware.dict_fetchall")
+    @patch("kernelCI_app.queries.hardware.connection")
+    def test_get_hardware_trees_data_from_database(
+        self, mock_connection, mock_dict_fetchall, mock_set_cache, mock_get_cache
+    ):
+        tree_records = [
+            {
+                "tree_name": "mainline",
+                "origin": "maestro",
+                "git_repository_branch": "master",
+                "git_repository_url": "https://my_url.com",
+                "git_commit_name": "v6.1",
+                "git_commit_hash": "abc123",
+                "git_commit_tags": None,
+            }
+        ]
+        mock_get_cache.return_value = None
+        mock_dict_fetchall.return_value = tree_records
+        setup_mock_cursor(mock_connection)
+
+        result = get_hardware_trees_data(
+            hardware_id="hardware",
+            origin="maestro",
+            start_datetime=START_DATE,
+            end_datetime=END_DATE,
+        )
+
+        assert len(result) == 1
+        assert result[0].tree_name == "mainline"
+        mock_set_cache.assert_called_once()
+
+
+class TestGenerateQueryParams:
+    def test_generate_query_params_single_commit(self):
+        commit_heads = [
+            CommitHead(
+                treeName="mainline",
+                repositoryUrl="https://my_url.com",
+                branch="master",
+                commitHash="abc123",
+            )
+        ]
+
+        result = _generate_query_params(commit_heads)
+
+        assert "tuple_str" in result
+        assert "query_params" in result
+        assert result["query_params"]["tree_name0"] == "mainline"
+        assert result["query_params"]["git_commit_hash0"] == "abc123"
+
+    def test_generate_query_params_multiple_commits(self):
+        commit_heads = [
+            CommitHead(
+                treeName="mainline",
+                repositoryUrl="https://my_url.com",
+                branch="master",
+                commitHash="abc123",
+            ),
+            CommitHead(
+                treeName="next",
+                repositoryUrl="https://my_url.com",
+                branch="master",
+                commitHash="def456",
+            ),
+        ]
+
+        result = _generate_query_params(commit_heads)
+
+        assert len(result["query_params"]) == 8
+        assert "tree_name0" in result["query_params"]
+        assert "tree_name1" in result["query_params"]
+
+
+class TestGetHardwareCommitHistory:
+    @patch("kernelCI_app.queries.hardware.connection")
+    def test_get_hardware_commit_history_success(self, mock_connection):
+        expected_result = [("abc123", "v6.1", None, datetime(2025, 11, 12))]
+        mock_cursor = setup_mock_cursor(mock_connection)
+        mock_cursor.fetchall.return_value = expected_result
+
+        result = get_hardware_commit_history(
+            origin="maestro",
+            start_date=datetime(2025, 11, 11),
+            end_date=datetime(2025, 11, 12),
+            commit_heads=[
+                CommitHead(
+                    treeName="mainline",
+                    repositoryUrl="https://my_url.com",
+                    branch="master",
+                    commitHash="abc123",
+                )
+            ],
+        )
+
+        assert result == expected_result
+
+    @patch("kernelCI_app.queries.hardware.connection")
+    def test_get_hardware_commit_history_empty_commits(self, mock_connection):
+        result = get_hardware_commit_history(
+            origin="maestro",
+            start_date=datetime(2025, 11, 11),
+            end_date=datetime(2025, 11, 12),
+            commit_heads=[],
+        )
+
+        assert result is None
+        mock_connection.cursor.assert_not_called()
+
+
+class TestQueryRecords:
+    @patch("kernelCI_app.queries.hardware.dict_fetchall")
+    @patch("kernelCI_app.queries.hardware.connection")
+    def test_query_records_success(self, mock_connection, mock_dict_fetchall):
+        expected_result = [{"id": "test", "status": "PASS"}]
+        mock_dict_fetchall.return_value = expected_result
+        mock_cursor = setup_mock_cursor(mock_connection)
+
+        result = query_records(
+            hardware_id="hardware",
+            origin="maestro",
+            trees=[TEST_TREE],
+            start_date=START_DATE,
+            end_date=END_DATE,
+        )
+
+        assert result == expected_result
+        mock_cursor.execute.assert_called_once()