Skip to content

Commit 874a097

Browse files
authored
[AL-6733] Add ADV specific assertion (#1199)
1 parent 08298ec commit 874a097

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

tests/integration/test_data_rows.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff 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

494501
def test_create_data_rows_with_metadata_missing_value(dataset, image_url):

0 commit comments

Comments
 (0)