@@ -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
145145async 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