File tree Expand file tree Collapse file tree 1 file changed +12
-5
lines changed Expand file tree Collapse file tree 1 file changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -476,19 +476,26 @@ def create_data_row(data_rows):
476476 CUSTOM_TEXT_SCHEMA_NAME ].uid
477477
478478
479- def test_create_data_rows_with_invalid_metadata (dataset , image_url ):
479+ def test_create_data_rows_with_invalid_metadata (dataset , image_url ,
480+ is_adv_enabled ):
480481 fields = make_metadata_fields ()
481482 # make the payload invalid by providing the same schema id more than once
482483 fields .append (
483- DataRowMetadataField (schema_id = TEXT_SCHEMA_ID , value = ' some msg' ))
484+ DataRowMetadataField (schema_id = TEXT_SCHEMA_ID , value = " some msg" ))
484485
485486 task = dataset .create_data_rows ([{
486487 DataRow .row_data : image_url ,
487488 DataRow .metadata_fields : fields
488489 }])
489- task .wait_till_done ()
490- assert task .status == "FAILED"
491- assert len (task .failed_data_rows ) > 0
490+ task .wait_till_done (timeout_seconds = 60 )
491+ if is_adv_enabled :
492+ assert task .status == "COMPLETE"
493+ assert len (task .failed_data_rows ) == 1
494+ assert f"A schemaId can only be specified once per DataRow : [{ TEXT_SCHEMA_ID } ]" in task .failed_data_rows [
495+ 0 ]["message" ]
496+ else :
497+ assert task .status == "FAILED"
498+ assert len (task .failed_data_rows ) > 0
492499
493500
494501def test_create_data_rows_with_metadata_missing_value (dataset , image_url ):
You can’t perform that action at this time.
0 commit comments