diff --git a/wiremock/resources/scenarios/resource.py b/wiremock/resources/scenarios/resource.py index ae9b7b4..356bd2a 100644 --- a/wiremock/resources/scenarios/resource.py +++ b/wiremock/resources/scenarios/resource.py @@ -20,5 +20,14 @@ def reset_all_scenarios(cls, parameters={}): response = cls.REST_CLIENT.post(cls.get_base_uri(cls.endpoint()), headers=make_headers(), params=parameters) return cls.REST_CLIENT.handle_response(response) + @classmethod + def change_scenario_state(cls, scenario_name, state): + response = cls.REST_CLIENT.put( + cls.get_base_uri(f"{cls.endpoint_single()}/{scenario_name}/state"), + headers=make_headers(), + json={"state": state}, + ) + return cls.REST_CLIENT.handle_response(response) + __all__ = ["Scenarios"]