def test_data_create_or_update_with_new_record(base_app, db, es):
data = faker.record("dat")
record = DataRecord.create_or_update(data)
control_number = str(record["control_number"])
record_db = RecordMetadata.query.filter_by(id=record.id).one()
assert record == record_db.json
record_pid = PersistentIdentifier.query.filter_by(
pid_type="dat", pid_value=str(control_number)).one()
assert record.model.id == record_pid.object_uuid
assert control_number == record_pid.pid_value
def test_data_create_or_update_with_existing_record(base_app, db, es):
data = faker.record("dat", with_control_number=True)
record = DataRecord.create(data)
assert data["control_number"] == record["control_number"]
data_update = {"dois": [{"value": "10.1000/UPDATED"}]}
data.update(data_update)
record_updated = DataRecord.create_or_update(data)
control_number = str(record_updated["control_number"])
assert record["control_number"] == record_updated["control_number"]
record_updated_db = RecordMetadata.query.filter_by(
id=record_updated.id).one()
assert data == record_updated_db.json
record_updated_pid = PersistentIdentifier.query.filter_by(
pid_type="dat", pid_value=str(control_number)).one()
assert record_updated.model.id == record_updated_pid.object_uuid
assert control_number == record_updated_pid.pid_value