Skip to content

Commit 422dfff

Browse files
committed
e2e test rename
1 parent afa3b1d commit 422dfff

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

tests/integration/test_end_to_end.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def agent_card() -> AgentCard:
7979
)
8080

8181

82-
class TransportSetup(NamedTuple):
82+
class ClientSetup(NamedTuple):
8383
"""Holds the client and task_store for a given test."""
8484

8585
client: BaseClient
@@ -98,7 +98,7 @@ def base_e2e_setup():
9898

9999

100100
@pytest.fixture
101-
def rest_setup(agent_card, base_e2e_setup) -> TransportSetup:
101+
def rest_setup(agent_card, base_e2e_setup) -> ClientSetup:
102102
task_store, handler = base_e2e_setup
103103
app_builder = A2ARESTFastAPIApplication(agent_card, handler)
104104
app = app_builder.build()
@@ -112,14 +112,14 @@ def rest_setup(agent_card, base_e2e_setup) -> TransportSetup:
112112
)
113113
)
114114
client = factory.create(agent_card)
115-
return TransportSetup(
115+
return ClientSetup(
116116
client=client,
117117
task_store=task_store,
118118
)
119119

120120

121121
@pytest.fixture
122-
def jsonrpc_setup(agent_card, base_e2e_setup) -> TransportSetup:
122+
def jsonrpc_setup(agent_card, base_e2e_setup) -> ClientSetup:
123123
task_store, handler = base_e2e_setup
124124
app_builder = A2AFastAPIApplication(
125125
agent_card, handler, extended_agent_card=agent_card
@@ -135,7 +135,7 @@ def jsonrpc_setup(agent_card, base_e2e_setup) -> TransportSetup:
135135
)
136136
)
137137
client = factory.create(agent_card)
138-
return TransportSetup(
138+
return ClientSetup(
139139
client=client,
140140
task_store=task_store,
141141
)
@@ -144,7 +144,7 @@ def jsonrpc_setup(agent_card, base_e2e_setup) -> TransportSetup:
144144
@pytest_asyncio.fixture
145145
async def grpc_setup(
146146
agent_card: AgentCard, base_e2e_setup
147-
) -> AsyncGenerator[TransportSetup, None]:
147+
) -> AsyncGenerator[ClientSetup, None]:
148148
task_store, handler = base_e2e_setup
149149
server = grpc.aio.server()
150150
port = server.add_insecure_port('[::]:0')
@@ -172,7 +172,7 @@ async def grpc_setup(
172172
)
173173
)
174174
client = factory.create(grpc_agent_card)
175-
yield TransportSetup(
175+
yield ClientSetup(
176176
client=client,
177177
task_store=task_store,
178178
)
@@ -188,7 +188,7 @@ async def grpc_setup(
188188
pytest.param('grpc_setup', id='gRPC'),
189189
]
190190
)
191-
def transport_setups(request) -> TransportSetup:
191+
def transport_setups(request) -> ClientSetup:
192192
"""Parametrized fixture that runs tests against all supported transports."""
193193
return request.getfixturevalue(request.param)
194194

0 commit comments

Comments
 (0)