fix: tests and pin polars requirement

Author: joseph-sentry

conftest.py

@@ -5,6 +5,7 @@
 import vcr
 from django.conf import settings
 from shared.reports.resources import Report, ReportFile, ReportLine
+from shared.storage.memory import MemoryStorageService
 from shared.utils.sessions import Session
 
 # we need to enable this in the test environment since we're often creating

graphql_api/tests/test_test_analytics.py

@@ -6,6 +6,7 @@
 from shared.django_apps.codecov_auth.tests.factories import OwnerFactory
 from shared.django_apps.core.tests.factories import RepositoryFactory
 from shared.storage.exceptions import BucketAlreadyExistsError
+from shared.storage.memory import MemoryStorageService
 
 from graphql_api.types.enums import (
     OrderingDirection,
@@ -23,6 +24,15 @@
 
 from .helper import GraphQLTestHelper
 
+
+@pytest.fixture
+def mock_storage(mocker):
+    m = mocker.patch("utils.test_results.StorageService")
+    storage_server = MemoryStorageService({})
+    m.return_value = storage_server
+    yield storage_server
+
+
 base_gql_query = """
     query {
         owner(username: "%s") {
@@ -119,22 +129,21 @@ def store_in_redis(repository):
 
 
 @pytest.fixture
-def store_in_storage(repository):
-    storage = StorageService()
+def store_in_storage(repository, mock_storage):
     try:
-        storage.create_root_storage("codecov")
+        mock_storage.create_root_storage("codecov")
     except BucketAlreadyExistsError:
         pass
 
-    storage.write_file(
+    mock_storage.write_file(
         "codecov",
         f"test_results/rollups/{repository.repoid}/{repository.branch}/30",
         test_results_table.write_ipc(None).getvalue(),
     )
 
     yield
 
-    storage.delete_file(
+    mock_storage.delete_file(
         "codecov",
         f"test_results/rollups/{repository.repoid}/{repository.branch}/30",
     )
@@ -144,17 +153,24 @@ class TestAnalyticsTestCase(
     GraphQLTestHelper,
 ):
     def test_get_test_results(
-        self, transactional_db, repository, store_in_redis, store_in_storage
+        self,
+        transactional_db,
+        repository,
+        store_in_redis,
+        store_in_storage,
+        mock_storage,
     ):
         results = get_results(repository.repoid, repository.branch, 30)
         assert results is not None
         assert results.equals(test_results_table)
 
-    def test_get_test_results_no_storage(self, transactional_db, repository):
+    def test_get_test_results_no_storage(
+        self, transactional_db, repository, mock_storage
+    ):
         assert get_results(repository.repoid, repository.branch, 30) is None
 
     def test_get_test_results_no_redis(
-        self, mocker, transactional_db, repository, store_in_storage
+        self, mocker, transactional_db, repository, store_in_storage, mock_storage
     ):
         m = mocker.patch("services.task.TaskService.cache_test_results_redis")
         results = get_results(repository.repoid, repository.branch, 30)
@@ -163,7 +179,9 @@ def test_get_test_results_no_redis(
 
         m.assert_called_once_with(repository.repoid, repository.branch)
 
-    def test_test_results(self, transactional_db, repository, store_in_redis):
+    def test_test_results(
+        self, transactional_db, repository, store_in_redis, mock_storage
+    ):
         test_results = generate_test_results(
             repoid=repository.repoid,
             ordering=TestResultsOrderingParameter.UPDATED_AT,
@@ -191,7 +209,9 @@ def test_test_results(self, transactional_db, repository, store_in_redis):
             },
         )
 
-    def test_test_results_asc(self, transactional_db, repository, store_in_redis):
+    def test_test_results_asc(
+        self, transactional_db, repository, store_in_redis, mock_storage
+    ):
         test_results = generate_test_results(
             repoid=repository.repoid,
             ordering=TestResultsOrderingParameter.UPDATED_AT,
@@ -255,6 +275,7 @@ def test_test_results_pagination(
         rows,
         repository,
         store_in_redis,
+        mock_storage,
     ):
         test_results = generate_test_results(
             repoid=repository.repoid,
@@ -331,6 +352,7 @@ def test_test_results_pagination_asc(
         rows,
         repository,
         store_in_redis,
+        mock_storage,
     ):
         test_results = generate_test_results(
             repoid=repository.repoid,
@@ -371,7 +393,7 @@ def test_test_results_pagination_asc(
             },
         )
 
-    def test_test_analytics_term_filter(self, repository, store_in_redis):
+    def test_test_analytics_term_filter(self, repository, store_in_redis, mock_storage):
         test_results = generate_test_results(
             repoid=repository.repoid,
             term="test1",
@@ -421,7 +443,7 @@ def test_test_analytics_testsuite_filter(self, repository, store_in_redis):
             },
         )
 
-    def test_test_analytics_flag_filter(self, repository, store_in_redis):
+    def test_test_analytics_flag_filter(self, repository, store_in_redis, mock_storage):
         test_results = generate_test_results(
             repoid=repository.repoid,
             flags=["flag1"],
@@ -446,7 +468,7 @@ def test_test_analytics_flag_filter(self, repository, store_in_redis):
             },
         )
 
-    def test_gql_query(self, repository, store_in_redis):
+    def test_gql_query(self, repository, store_in_redis, mock_storage):
         query = base_gql_query % (
             repository.author.username,
             repository.name,
@@ -492,7 +514,7 @@ def test_gql_query(self, repository, store_in_redis):
             },
         ]
 
-    def test_gql_query_aggregates(self, repository, store_in_redis):
+    def test_gql_query_aggregates(self, repository, store_in_redis, mock_storage):
         query = base_gql_query % (
             repository.author.username,
             repository.name,
@@ -519,7 +541,7 @@ def test_gql_query_aggregates(self, repository, store_in_redis):
             "totalSlowTests": 1,
         }
 
-    def test_gql_query_flake_aggregates(self, repository, store_in_redis):
+    def test_gql_query_flake_aggregates(self, repository, store_in_redis, mock_storage):
         query = base_gql_query % (
             repository.author.username,
             repository.name,

requirements.in

@@ -29,7 +29,7 @@ minio
 opentelemetry-instrumentation-django>=0.45b0
 opentelemetry-sdk>=1.24.0
 opentracing
-polars
+polars==1.12.0
 pre-commit
 psycopg2
 PyJWT

requirements.txt

@@ -2,7 +2,7 @@
 # This file is autogenerated by pip-compile with Python 3.12
 # by the following command:
 #
-#    pip-compile requirements.in
+#    pip-compile
 #
 aiodataloader==0.4.0
     # via -r requirements.in