fix tests

Author: Swatinem

codecov_auth/tests/unit/test_authentication.py

@@ -1,6 +1,6 @@
 from datetime import datetime, timedelta
 from http.cookies import SimpleCookie
-from unittest.mock import call, patch
+from unittest.mock import AsyncMock, call, patch
 
 import pytest
 from django.conf import settings
@@ -198,7 +198,9 @@ def test_impersonation(self):
         assert res.json()["data"]["me"] == {"user": {"username": "impersonateme"}}
 
     @patch("core.commands.repository.repository.RepositoryCommands.fetch_repository")
-    def test_impersonation_with_okta(self, mock_call_to_fetch_repository):
+    def test_impersonation_with_okta(
+        self, mock_call_to_fetch_repository, new_callable=AsyncMock
+    ):
         repo = RepositoryFactory(author=self.owner_to_impersonate, private=True)
         query_repositories = """{ owner(username: "%s") { repository(name: "%s") { ... on Repository { name } } } }"""
         query = query_repositories % (repo.author.username, repo.name)
@@ -228,12 +230,16 @@ def test_impersonation_with_okta(self, mock_call_to_fetch_repository):
                     repo.name,
                     [],
                     exclude_okta_enforced_repos=True,
+                    needs_coverage=False,
+                    needs_commits=False,
                 ),
                 call(
                     self.owner_to_impersonate,
                     repo.name,
                     [],
                     exclude_okta_enforced_repos=False,
+                    needs_coverage=False,
+                    needs_commits=False,
                 ),
             ]
         )

core/commands/repository/tests/test_repository.py

@@ -17,9 +17,7 @@ def setUp(self):
     @patch("core.commands.repository.repository.FetchRepositoryInteractor.execute")
     def test_fetch_repository_to_interactor(self, interactor_mock):
         self.command.fetch_repository(self.org, self.repo.name, [])
-        interactor_mock.assert_called_once_with(
-            self.org, self.repo.name, [], exclude_okta_enforced_repos=True
-        )
+        interactor_mock.assert_called_once_with(self.org, self.repo.name, [])
 
     @patch("core.commands.repository.repository.FetchRepositoryInteractor.execute")
     def test_fetch_repository_to_interactor_with_enforcing_okta(self, interactor_mock):

graphql_api/helpers/requested_fields.py

@@ -4,9 +4,9 @@
 from graphql import GraphQLResolveInfo
 from graphql.language import (
     FieldNode,
-    SelectionNode,
-    InlineFragmentNode,
     FragmentSpreadNode,
+    InlineFragmentNode,
+    SelectionNode,
 )
 
 

graphql_api/tests/test_requested_fields.py

@@ -1,12 +1,12 @@
-from graphql_api.helpers.requested_fields import selected_fields
-from graphql.language import parse
-
 from graphql import GraphQLResolveInfo
 from graphql.language import (
-    OperationDefinitionNode,
     FragmentDefinitionNode,
+    OperationDefinitionNode,
+    parse,
 )
 
+from graphql_api.helpers.requested_fields import selected_fields
+
 
 def parse_into_resolveinfo(source: str) -> GraphQLResolveInfo:
     document = parse(source)

graphql_api/types/owner/owner.py

@@ -2,7 +2,6 @@
 from hashlib import sha1
 from typing import Any, Coroutine, Iterable, List, Optional
 
-from graphql_api.helpers.requested_fields import selected_fields
 import sentry_sdk
 import shared.rate_limits as rate_limits
 import stripe
@@ -39,6 +38,7 @@
     require_part_of_org,
     require_shared_account_or_part_of_org,
 )
+from graphql_api.helpers.requested_fields import selected_fields
 from graphql_api.types.enums import OrderingDirection, RepositoryOrdering
 from graphql_api.types.errors.errors import NotFoundError
 from graphql_api.types.repository.repository import TOKEN_UNAVAILABLE