Skip to content

Commit

Permalink
test: fix missing test
Browse files Browse the repository at this point in the history
  • Loading branch information
pkucmus committed Jan 31, 2025
1 parent 40eca4d commit ee43396
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 4 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ dependencies = [
]

[project.optional-dependencies]
dev = ["black", "mypy", "pylint"]
dev = ["black<25", "mypy", "pylint"]
test = [
"pytest",
"pytest-asyncio",
Expand Down
21 changes: 18 additions & 3 deletions tests/relay/test_objects.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,10 @@
from pytest_mock import MockFixture

from ariadne import make_executable_schema
from ariadne.contrib.relay.arguments import ConnectionArguments
from ariadne.contrib.relay.arguments import (
ConnectionArguments,
ForwardConnectionArguments,
)
from ariadne.contrib.relay.connection import RelayConnection
from ariadne.contrib.relay.objects import (
RelayObjectType,
Expand Down Expand Up @@ -129,6 +132,7 @@ def test_relay_object_resolve_wrapper(mocker: MockFixture, friends_connection):
mock_resolver.assert_called_once_with(
None, None, mock_connection_arguments, first=10
)
mock_connection_arguments_class.assert_called_once_with(first=10)


@pytest.mark.asyncio
Expand All @@ -154,8 +158,19 @@ async def resolver(*_, **__):
}


def test_relay_object_resolve_wrapper_with_custom_arguments():
pass
def test_relay_object_resolve_wrapper_with_custom_arguments(mocker: MockFixture):
object_type = RelayObjectType(
"User", connection_arguments_class=ForwardConnectionArguments
)
mock_resolver = mocker.Mock()

wrapped_resolver = object_type.resolve_wrapper(mock_resolver)
wrapped_resolver(None, None, first=10, after="VXNlcjox")

connection_arg_call = mock_resolver.call_args_list[0].args[2]

assert connection_arg_call.first == 10
assert connection_arg_call.after == "VXNlcjox"


def test_relay_object_connection_decorator(mocker: MockFixture):
Expand Down

0 comments on commit ee43396

Please sign in to comment.