@@ -76,27 +76,6 @@ def consumer_read_document_reference_step(
7676 context .response = client .read (doc_ref_id )
7777
7878
79- @when ("consumer '{ods_code}' sends HEAD request to {endpoint} endpoint" )
80- def consumer_head_request_step (context : Context , ods_code : str , endpoint : str ):
81- client = consumer_client_from_context (context , ods_code )
82- context .response = client .head (endpoint )
83-
84-
85- @when (
86- "consumer '{ods_code}' sends HEAD request to '{endpoint}' endpoint with parameters"
87- )
88- def consumer_head_request_step_with_parameters (
89- context : Context , ods_code : str , endpoint : str
90- ):
91- if not context .table :
92- raise ValueError ("No search query table provided" )
93-
94- items = {row ["parameter" ]: row ["value" ] for row in context .table }
95-
96- client = consumer_client_from_context (context , ods_code )
97- context .response = client .head (endpoint , items )
98-
99-
10079@when ("producer '{ods_code}' creates a DocumentReference with values" )
10180def create_post_document_reference_step (context : Context , ods_code : str ):
10281 client = producer_client_from_context (context , ods_code )
@@ -268,3 +247,32 @@ def producer_search_document_reference_step(context: Context, ods_code: str):
268247 pointer_type = pointer_type ,
269248 extra_params = items ,
270249 )
250+
251+
252+ @when ("{consumer_or_producer} '{ods_code}' sends HEAD request to '{endpoint}' endpoint" )
253+ def consumer_head_request_step (
254+ context : Context , consumer_or_producer : str , ods_code : str , endpoint : str
255+ ):
256+ if consumer_or_producer == "producer" :
257+ client = producer_client_from_context (context , ods_code )
258+ elif consumer_or_producer == "consumer" :
259+ client = consumer_client_from_context (context , ods_code )
260+ context .response = client .head (endpoint )
261+
262+
263+ @when (
264+ "{consumer_or_producer} '{ods_code}' sends HEAD request to '{endpoint}' endpoint with parameters"
265+ )
266+ def consumer_head_request_step_with_parameters (
267+ context : Context , consumer_or_producer : str , ods_code : str , endpoint : str
268+ ):
269+ if not context .table :
270+ raise ValueError ("No search query table provided" )
271+
272+ items = {row ["parameter" ]: row ["value" ] for row in context .table }
273+
274+ if consumer_or_producer == "producer" :
275+ client = producer_client_from_context (context , ods_code )
276+ elif consumer_or_producer == "consumer" :
277+ client = consumer_client_from_context (context , ods_code )
278+ context .response = client .head (endpoint , items )
0 commit comments