Update tests

Author: rohitvinnakota-codecov

graphql_api/tests/test_owner.py

@@ -1196,25 +1196,35 @@ def test_fetch_available_plans_is_enterprise_plan(self):
 
     @patch("services.self_hosted.get_config")
     def test_ai_enabled_repositories(self, get_config_mock):
-        current_org = OwnerFactory(
-            username="random-plan-user",
-            service="github",
-        )
-
         get_config_mock.return_value = [
             {"service": "github", "ai_features_app_id": 12345},
         ]
 
+        ai_app_installation = GithubAppInstallation(
+            name="ai-features",
+            owner=self.owner,
+            repository_service_ids=[],
+            installation_id=12345,
+        )
+
+        ai_app_installation.save()
+
         query = """{
             owner(username: "%s") {
-                aiEnabledRepos
+                aiEnabledRepositories(first: 20) {
+                    edges {
+                        node {
+                            name
+                        }
+                    }
+                }
             }
         }
 
-        """ % (current_org.username)
-        data = self.gql_request(query, owner=current_org)
-        assert data["owner"]["aiEnabledRepos"] is None
-
+        """ % (self.owner.username)
+        data = self.gql_request(query, owner=self.owner)
+        reps = paginate_connection(data["owner"]["aiEnabledRepositories"])
+        assert reps == [{'name': 'a'}, {'name': 'b'}]
 
     @patch("services.self_hosted.get_config")
     def test_ai_enabled_repositories_app_not_configured(self, get_config_mock):
@@ -1229,13 +1239,19 @@ def test_ai_enabled_repositories_app_not_configured(self, get_config_mock):
 
         query = """{
             owner(username: "%s") {
-                aiEnabledRepos
+                aiEnabledRepositories {
+                    edges {
+                        node {
+                            name
+                        }
+                    }
+                }
             }
         }
 
         """ % (current_org.username)
         data = self.gql_request(query, owner=current_org)
-        assert data["owner"]["aiEnabledRepos"] is None
+        assert data["owner"]["aiEnabledRepositories"] is None
 
     def test_fetch_owner_with_no_service(self):
         current_org = OwnerFactory(

graphql_api/types/owner/owner.graphql

@@ -47,7 +47,7 @@ type Owner {
     after: String
     last: Int
     before: String
-  ): RepositoryConnection! @cost(complexity: 25, multipliers: ["first", "last"])
+  ): RepositoryConnection @cost(complexity: 25, multipliers: ["first", "last"])
   uploadTokenRequired: Boolean
   activatedUserCount: Int
 }

graphql_api/types/owner/owner.py

@@ -32,6 +32,7 @@
     Connection,
     build_connection_graphql,
     queryset_to_connection,
+    queryset_to_connection_sync,
 )
 from graphql_api.helpers.mutation import (
     require_part_of_org,
@@ -406,7 +407,9 @@ def resolve_activated_user_count(owner: Owner, info: GraphQLResolveInfo) -> int:
 def resolve_billing(owner: Owner, info: GraphQLResolveInfo) -> dict | None:
     return owner
 
+
 @owner_bindable.field("aiEnabledRepositories")
+@sync_to_async
 def resolve_ai_enabled_repositories(
     owner: Owner,
     info: GraphQLResolveInfo,
@@ -423,15 +426,14 @@ def resolve_ai_enabled_repositories(
 
     current_owner = info.context["request"].current_owner
     queryset = Repository.objects.filter(author=owner).viewable_repos(current_owner)
-
     if ai_features_app_install.repository_service_ids:
         queryset = queryset.filter(
             service_id__in=ai_features_app_install.repository_service_ids
         )
 
-    return queryset_to_connection(
+    return queryset_to_connection_sync(
         queryset,
         ordering=(ordering, RepositoryOrdering.ID),
         ordering_direction=ordering_direction,
         **kwargs,
-    )
+    )