|
18 | 18 |
|
19 | 19 | from pact.v3.pact import Pact |
20 | 20 | from tests.v3.compatibility_suite.util import ( |
21 | | - InteractionDefinition, |
22 | 21 | parse_horizontal_markdown_table, |
23 | 22 | parse_markdown_table, |
24 | 23 | ) |
| 24 | +from tests.v3.compatibility_suite.util.interaction_definition import ( |
| 25 | + InteractionDefinition, |
| 26 | + InteractionState, |
| 27 | +) |
25 | 28 | from tests.v3.compatibility_suite.util.provider import ( |
26 | 29 | a_provider_is_started_that_can_generate_the_message, |
27 | 30 | a_provider_state_callback_is_configured, |
@@ -229,7 +232,7 @@ def a_pact_file_for_is_to_be_verified_with_the_following( |
229 | 232 | interaction_definition = InteractionDefinition( |
230 | 233 | type="Async", |
231 | 234 | description=name, |
232 | | - **table, |
| 235 | + **table, # type: ignore[arg-type] |
233 | 236 | ) |
234 | 237 | interaction_definition.add_to_pact(pact, name) |
235 | 238 | (temp_dir / "pacts").mkdir(exist_ok=True, parents=True) |
@@ -284,16 +287,14 @@ def a_pact_file_for_is_to_be_verified_with_provider_state( |
284 | 287 | description=name, |
285 | 288 | body=fixture, |
286 | 289 | ) |
287 | | - interaction_definition.states = [InteractionDefinition.State(provider_state)] |
| 290 | + interaction_definition.states = [InteractionState(provider_state)] |
288 | 291 | interaction_definition.add_to_pact(pact, name) |
289 | 292 | (temp_dir / "pacts").mkdir(exist_ok=True, parents=True) |
290 | 293 | pact.write_file(temp_dir / "pacts") |
291 | 294 | verifier.add_source(temp_dir / "pacts") |
292 | 295 | with (temp_dir / "provider_states").open("w") as f: |
293 | 296 | logger.debug("Writing provider state to %s", temp_dir / "provider_states") |
294 | | - json.dump( |
295 | | - [s.as_dict() for s in [InteractionDefinition.State(provider_state)]], f |
296 | | - ) |
| 297 | + json.dump([s.as_dict() for s in [InteractionState(provider_state)]], f) |
297 | 298 |
|
298 | 299 |
|
299 | 300 | @given( |
|
0 commit comments