Examples with RecordType org.folio.dao.util.RecordType used on opensource projects

Example 1 with RecordType

use of org.folio.dao.util.RecordType in project mod-source-record-storage by folio-org.

the class RecordServiceTest method getMarcSourceRecordsByListOfIdsThatAreDeleted.

private void getMarcSourceRecordsByListOfIdsThatAreDeleted(TestContext context, Record.RecordType recordType, RecordType parsedRecordType) {
    Async async = context.async();
    List<Record> records = TestMocks.getRecords().stream().map(record -> {
        Record deletedRecord = new Record().withId(record.getId()).withSnapshotId(record.getSnapshotId()).withMatchedId(record.getMatchedId()).withRecordType(record.getRecordType()).withState(State.DELETED).withGeneration(record.getGeneration()).withOrder(record.getOrder()).withLeaderRecordStatus(record.getLeaderRecordStatus()).withRawRecord(record.getRawRecord()).withParsedRecord(record.getParsedRecord()).withAdditionalInfo(record.getAdditionalInfo()).withExternalIdsHolder(record.getExternalIdsHolder());
        if (Objects.nonNull(record.getMetadata())) {
            deletedRecord.withMetadata(record.getMetadata());
        }
        if (Objects.nonNull(record.getErrorRecord())) {
            deletedRecord.withErrorRecord(record.getErrorRecord());
        }
        return deletedRecord;
    }).collect(Collectors.toList());
    RecordCollection recordCollection = new RecordCollection().withRecords(records).withTotalRecords(records.size());
    saveRecords(recordCollection.getRecords()).onComplete(batch -> {
        if (batch.failed()) {
            context.fail(batch.cause());
        }
        List<String> ids = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(Record::getMatchedId).collect(Collectors.toList());
        recordService.getSourceRecords(ids, IdType.RECORD, parsedRecordType, true, TENANT_ID).onComplete(get -> {
            if (get.failed()) {
                context.fail(get.cause());
            }
            List<SourceRecord> expected = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(RecordDaoUtil::toSourceRecord).sorted(comparing(SourceRecord::getRecordId)).collect(Collectors.toList());
            get.result().getSourceRecords().sort(comparing(SourceRecord::getRecordId));
            context.assertEquals(expected.size(), get.result().getTotalRecords());
            compareSourceRecords(context, expected, get.result().getSourceRecords());
            async.complete();
        });
    });
}

Example 2 with RecordType

use of org.folio.dao.util.RecordType in project mod-source-record-storage by folio-org.

the class RecordServiceTest method getMarcSourceRecords.

private void getMarcSourceRecords(TestContext context, RecordType parsedRecordType, Record.RecordType recordType) {
    Async async = context.async();
    List<Record> records = TestMocks.getRecords();
    RecordCollection recordCollection = new RecordCollection().withRecords(records).withTotalRecords(records.size());
    saveRecords(recordCollection.getRecords()).onComplete(batch -> {
        if (batch.failed()) {
            context.fail(batch.cause());
        }
        Condition condition = DSL.trueCondition();
        List<OrderField<?>> orderFields = new ArrayList<>();
        recordService.getSourceRecords(condition, parsedRecordType, orderFields, 0, 10, TENANT_ID).onComplete(get -> {
            if (get.failed()) {
                context.fail(get.cause());
            }
            List<SourceRecord> expected = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(RecordDaoUtil::toSourceRecord).sorted(comparing(SourceRecord::getRecordId)).collect(Collectors.toList());
            get.result().getSourceRecords().sort(comparing(SourceRecord::getRecordId));
            context.assertEquals(expected.size(), get.result().getTotalRecords());
            compareSourceRecords(context, expected, get.result().getSourceRecords());
            async.complete();
        });
    });
}

Example 3 with RecordType

use of org.folio.dao.util.RecordType in project mod-source-record-storage by folio-org.

the class RecordServiceTest method getMarcSourceRecordsBetweenDates.

private void getMarcSourceRecordsBetweenDates(TestContext context, Record.RecordType recordType, RecordType parsedRecordType, OffsetDateTime earliestDate, OffsetDateTime latestDate) {
    Async async = context.async();
    List<Record> records = TestMocks.getRecords();
    RecordCollection recordCollection = new RecordCollection().withRecords(records).withTotalRecords(records.size());
    saveRecords(recordCollection.getRecords()).onComplete(batch -> {
        if (batch.failed()) {
            context.fail(batch.cause());
        }
        Condition condition = RECORDS_LB.CREATED_DATE.between(earliestDate, latestDate);
        List<OrderField<?>> orderFields = new ArrayList<>();
        recordService.getSourceRecords(condition, parsedRecordType, orderFields, 0, 10, TENANT_ID).onComplete(get -> {
            if (get.failed()) {
                context.fail(get.cause());
            }
            List<SourceRecord> expected = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(RecordDaoUtil::toSourceRecord).sorted(comparing(SourceRecord::getRecordId)).collect(Collectors.toList());
            get.result().getSourceRecords().sort(comparing(SourceRecord::getRecordId));
            context.assertEquals(expected.size(), get.result().getTotalRecords());
            compareSourceRecords(context, expected, get.result().getSourceRecords());
            async.complete();
        });
    });
}

Example 4 with RecordType

use of org.folio.dao.util.RecordType in project mod-source-record-storage by folio-org.

the class RecordServiceTest method getMarcSourceRecordsByListOfIds.

private void getMarcSourceRecordsByListOfIds(TestContext context, Record.RecordType recordType, RecordType parsedRecordType) {
    Async async = context.async();
    List<Record> records = TestMocks.getRecords();
    RecordCollection recordCollection = new RecordCollection().withRecords(records).withTotalRecords(records.size());
    saveRecords(recordCollection.getRecords()).onComplete(batch -> {
        if (batch.failed()) {
            context.fail(batch.cause());
        }
        List<String> ids = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(Record::getMatchedId).collect(Collectors.toList());
        recordService.getSourceRecords(ids, IdType.RECORD, parsedRecordType, false, TENANT_ID).onComplete(get -> {
            if (get.failed()) {
                context.fail(get.cause());
            }
            List<SourceRecord> expected = records.stream().filter(r -> r.getRecordType().equals(recordType)).map(RecordDaoUtil::toSourceRecord).sorted(comparing(SourceRecord::getRecordId)).collect(Collectors.toList());
            sortByRecordId(get);
            context.assertEquals(expected.size(), get.result().getTotalRecords());
            compareSourceRecords(context, expected, get.result().getSourceRecords());
            async.complete();
        });
    });
}

Example 5 with RecordType

use of org.folio.dao.util.RecordType in project mod-source-record-storage by folio-org.

the class RecordServiceTest method updateParsedMarcRecords.

private void updateParsedMarcRecords(TestContext context, Record.RecordType recordType) {
    Async async = context.async();
    List<Record> original = TestMocks.getRecords().stream().filter(record -> record.getRecordType().equals(recordType)).collect(Collectors.toList());
    RecordCollection recordCollection = new RecordCollection().withRecords(original).withTotalRecords(original.size());
    saveRecords(recordCollection.getRecords()).onComplete(batch -> {
        if (batch.failed()) {
            context.fail(batch.cause());
        }
        List<Record> updated = original.stream().map(record -> record.withExternalIdsHolder(record.getExternalIdsHolder().withInstanceId(UUID.randomUUID().toString()))).collect(Collectors.toList());
        recordCollection.withRecords(updated).withTotalRecords(updated.size());
        List<ParsedRecord> expected = updated.stream().map(Record::getParsedRecord).collect(Collectors.toList());
        recordService.updateParsedRecords(recordCollection, TENANT_ID).onComplete(update -> {
            if (update.failed()) {
                context.fail(update.cause());
            }
            context.assertEquals(0, update.result().getErrorMessages().size());
            context.assertEquals(expected.size(), update.result().getTotalRecords());
            expected.sort(comparing(ParsedRecord::getId));
            update.result().getParsedRecords().sort(comparing(ParsedRecord::getId));
            compareParsedRecords(context, expected, update.result().getParsedRecords());
            GenericCompositeFuture.all(updated.stream().map(record -> recordDao.getRecordByMatchedId(record.getMatchedId(), TENANT_ID).onComplete(get -> {
                if (get.failed()) {
                    context.fail(get.cause());
                }
                context.assertTrue(get.result().isPresent());
            })).collect(Collectors.toList())).onComplete(res -> {
                if (res.failed()) {
                    context.fail(res.cause());
                }
                async.complete();
            });
        });
    });
}