Search in sources :

Example 6 with CQL2PgJSON

use of org.folio.cql2pgjson.CQL2PgJSON in project raml-module-builder by folio-org.

the class ConnIT method streamGetDoneHandlerWithThrowable.

@Test
void streamGetDoneHandlerWithThrowable(VertxTestContext vtc) throws CQL2PgJSONException {
    CQLWrapper cqlWrapper = new CQLWrapper(new CQL2PgJSON("jsonb"), "key=b");
    with(randomUuid(), "a", randomUuid(), "b", trans -> {
        return trans.streamGet("t", Pojo.class, cqlWrapper, info -> info.result().handler(x -> {
            throw new RuntimeException("foo");
        }));
    }).onComplete(failingThenComplete(vtc, t -> assertThat(t).hasMessage("foo")));
}
Also used : VertxTestContext(io.vertx.junit5.VertxTestContext) BeforeEach(org.junit.jupiter.api.BeforeEach) CsvSource(org.junit.jupiter.params.provider.CsvSource) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) BiFunction(java.util.function.BiFunction) PostgresTesterContainer(org.folio.postgres.testing.PostgresTesterContainer) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) WithAssertions(org.assertj.core.api.WithAssertions) Function(java.util.function.Function) Supplier(java.util.function.Supplier) Timeout(io.vertx.junit5.Timeout) AfterAll(org.junit.jupiter.api.AfterAll) CQL2PgJSONException(org.folio.cql2pgjson.exception.CQL2PgJSONException) ExtendWith(org.junit.jupiter.api.extension.ExtendWith) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) BeforeAll(org.junit.jupiter.api.BeforeAll) JsonObject(io.vertx.core.json.JsonObject) AsyncResult(io.vertx.core.AsyncResult) Vertx(io.vertx.core.Vertx) UUID(java.util.UUID) VertxExtension(io.vertx.junit5.VertxExtension) Future(io.vertx.core.Future) TimeUnit(java.util.concurrent.TimeUnit) Test(org.junit.jupiter.api.Test) Mockito(org.mockito.Mockito) JsonArray(io.vertx.core.json.JsonArray) CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) PgConnection(io.vertx.pgclient.PgConnection) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) Handler(io.vertx.core.Handler) Collections(java.util.Collections) CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) Test(org.junit.jupiter.api.Test) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest)

Example 7 with CQL2PgJSON

use of org.folio.cql2pgjson.CQL2PgJSON in project raml-module-builder by folio-org.

the class PostgresClientIT method getCQLWrapperJsonbField.

@Test
public void getCQLWrapperJsonbField(TestContext context) throws FieldException {
    final String tableDefiniton = "id UUID PRIMARY KEY , jsonb JSONB NOT NULL, distinct_test_field TEXT";
    List<FacetField> facets = null;
    createTableWithPoLines(context, MOCK_POLINES_TABLE, tableDefiniton);
    CQL2PgJSON cql2pgJson = new CQL2PgJSON(MOCK_POLINES_TABLE + ".jsonb");
    {
        CQLWrapper cqlWrapper = new CQLWrapper(cql2pgJson, "cql.allRecords=1");
        Async async = context.async();
        postgresClient.get(MOCK_POLINES_TABLE, Object.class, "jsonb", cqlWrapper, true, true, facets, null, handler -> {
            context.assertTrue(handler.succeeded());
            ResultInfo resultInfo = handler.result().getResultInfo();
            context.assertEquals(6, resultInfo.getTotalRecords());
            async.complete();
        });
        async.awaitSuccess();
    }
    String distinctOn = "jsonb->>'order_format'";
    {
        CQLWrapper cqlWrapper = new CQLWrapper(cql2pgJson, "cql.allRecords=1");
        Async async = context.async();
        postgresClient.get(MOCK_POLINES_TABLE, Object.class, "jsonb", cqlWrapper, true, true, facets, distinctOn, handler -> {
            context.assertTrue(handler.succeeded());
            ResultInfo resultInfo = handler.result().getResultInfo();
            context.assertEquals(4, resultInfo.getTotalRecords());
            async.complete();
        });
        async.awaitSuccess();
    }
}
Also used : CoreMatchers.is(org.hamcrest.CoreMatchers.is) TestContext(io.vertx.ext.unit.TestContext) RowStream(io.vertx.sqlclient.RowStream) Arrays(java.util.Arrays) PgNotification(io.vertx.pgclient.PgNotification) TransactionRollbackException(io.vertx.sqlclient.TransactionRollbackException) VertxUtils(org.folio.rest.tools.utils.VertxUtils) Tuple(io.vertx.sqlclient.Tuple) UpdateSection(org.folio.rest.persist.Criteria.UpdateSection) CoreMatchers.instanceOf(org.hamcrest.CoreMatchers.instanceOf) RowIterator(io.vertx.sqlclient.RowIterator) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) SqlResult(io.vertx.sqlclient.SqlResult) After(org.junit.After) JsonObject(io.vertx.core.json.JsonObject) Offset(org.folio.rest.persist.Criteria.Offset) Collector(java.util.stream.Collector) Transaction(io.vertx.sqlclient.Transaction) AfterClass(org.junit.AfterClass) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) RowImpl(io.vertx.pgclient.impl.RowImpl) Set(java.util.Set) UUID(java.util.UUID) FieldException(org.folio.cql2pgjson.exception.FieldException) Future(io.vertx.core.Future) StandardCharsets(java.nio.charset.StandardCharsets) UncheckedIOException(java.io.UncheckedIOException) IOUtils(org.apache.commons.io.IOUtils) CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) Base64(java.util.Base64) List(java.util.List) Stream(java.util.stream.Stream) Criterion(org.folio.rest.persist.Criteria.Criterion) Results(org.folio.rest.persist.interfaces.Results) Facet(org.folio.rest.jaxrs.model.Facet) RowDesc(io.vertx.sqlclient.impl.RowDesc) Async(io.vertx.ext.unit.Async) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) BeforeClass(org.junit.BeforeClass) FacetField(org.folio.rest.persist.facets.FacetField) PostgresTesterContainer(org.folio.postgres.testing.PostgresTesterContainer) Criteria(org.folio.rest.persist.Criteria.Criteria) CoreMatchers.not(org.hamcrest.CoreMatchers.not) RunWith(org.junit.runner.RunWith) Timeout(io.vertx.ext.unit.junit.Timeout) ResultInfo(org.folio.rest.jaxrs.model.ResultInfo) Function(java.util.function.Function) TotaledResults(org.folio.rest.persist.PostgresClient.TotaledResults) ArrayList(java.util.ArrayList) PreparedStatement(io.vertx.sqlclient.PreparedStatement) HashSet(java.util.HashSet) CompositeFuture(io.vertx.core.CompositeFuture) Poline(org.folio.rest.persist.helpers.Poline) PrepareOptions(io.vertx.sqlclient.PrepareOptions) SqlConnection(io.vertx.sqlclient.SqlConnection) Limit(org.folio.rest.persist.Criteria.Limit) QueryHelper(org.folio.rest.persist.PostgresClient.QueryHelper) RowSet(io.vertx.sqlclient.RowSet) MatcherAssert.assertThat(org.hamcrest.MatcherAssert.assertThat) AsyncResult(io.vertx.core.AsyncResult) LinkedList(java.util.LinkedList) DatabaseMetadata(io.vertx.sqlclient.spi.DatabaseMetadata) CoreMatchers.nullValue(org.hamcrest.CoreMatchers.nullValue) Before(org.junit.Before) Files(java.nio.file.Files) Query(io.vertx.sqlclient.Query) Promise(io.vertx.core.Promise) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) Vertx(io.vertx.core.Vertx) PgPool(io.vertx.pgclient.PgPool) JsonProcessingException(com.fasterxml.jackson.core.JsonProcessingException) IOException(java.io.IOException) Test(org.junit.Test) VertxUnitRunner(io.vertx.ext.unit.junit.VertxUnitRunner) PreparedQuery(io.vertx.sqlclient.PreparedQuery) JsonArray(io.vertx.core.json.JsonArray) PgConnection(io.vertx.pgclient.PgConnection) Rule(org.junit.Rule) Paths(java.nio.file.Paths) Row(io.vertx.sqlclient.Row) LocalRowSet(org.folio.rest.persist.helpers.LocalRowSet) Handler(io.vertx.core.Handler) SimplePojo(org.folio.rest.persist.helpers.SimplePojo) Collections(java.util.Collections) InputStream(java.io.InputStream) CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) Async(io.vertx.ext.unit.Async) FacetField(org.folio.rest.persist.facets.FacetField) JsonObject(io.vertx.core.json.JsonObject) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) ResultInfo(org.folio.rest.jaxrs.model.ResultInfo) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) Test(org.junit.Test)

Example 8 with CQL2PgJSON

use of org.folio.cql2pgjson.CQL2PgJSON in project raml-module-builder by folio-org.

the class PostgresClientIT method streamGetWithSyntaxError.

@Test
public void streamGetWithSyntaxError(TestContext context) throws FieldException {
    createTableWithPoLines(context);
    CQLWrapper wrapper = new CQLWrapper(new CQL2PgJSON("jsonb"), "edition=");
    postgresClient.streamGet(MOCK_POLINES_TABLE, Object.class, "jsonb", wrapper, false, null, context.asyncAssertFailure());
}
Also used : CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) Test(org.junit.Test)

Example 9 with CQL2PgJSON

use of org.folio.cql2pgjson.CQL2PgJSON in project raml-module-builder by folio-org.

the class BooksDemoAPI method getRmbtestsTest.

@Validate
@Override
public void getRmbtestsTest(String query, RoutingContext routingContext, Map<String, String> okapiHeaders, Handler<AsyncResult<Response>> asyncResultHandler, Context vertxContext) {
    switch(query == null ? "null" : query) {
        case "badclass=true":
            PgUtil.streamGet(TABLE, /* can not be deserialized */
            StringBuilder.class, null, 0, 10, new LinkedList<>(), "books", routingContext, okapiHeaders, vertxContext);
            break;
        case "nullpointer=true":
            PgUtil.streamGet(TABLE, Book.class, null, 0, 10, null, "books", routingContext, /* okapiHeaders is null which results in exception */
            null, vertxContext);
            break;
        case "slim=true":
            PgUtil.streamGet(TABLE, SlimBook.class, null, 0, 10, new LinkedList<>(), "books", routingContext, okapiHeaders, vertxContext);
            break;
        case "wrapper=true":
            try {
                CQLWrapper wrapper = new CQLWrapper(new CQL2PgJSON("jsonb"), "cql.allRecords=true");
                PgUtil.streamGet(TABLE, Book.class, wrapper, null, "books", routingContext, okapiHeaders, vertxContext);
            } catch (FieldException e) {
                GetRmbtestsTestResponse.respond500WithTextPlain(e.getMessage());
            }
            break;
        default:
            PgUtil.streamGet(TABLE, Book.class, query, 0, 10, new LinkedList<>(), "books", routingContext, okapiHeaders, vertxContext);
    }
}
Also used : CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) FieldException(org.folio.cql2pgjson.exception.FieldException) CQLWrapper(org.folio.rest.persist.cql.CQLWrapper) Validate(org.folio.rest.annotations.Validate)

Example 10 with CQL2PgJSON

use of org.folio.cql2pgjson.CQL2PgJSON in project raml-module-builder by folio-org.

the class TestCLI method testCLIWithNoSchemaOrDBSchema.

@Test
public void testCLIWithNoSchemaOrDBSchema() throws FieldException, IOException, QueryValidationException, ParseException {
    String cql = "holdingsRecords.permanentLocationId=\"fcd64ce1-6995-48f0-840e-89ffa2\"";
    String[] args = new String[] { "-t", "instance", "-f", "jsonb", cql };
    String fullFieldName = "instance.jsonb";
    CQL2PgJSON cql2pgjson = new CQL2PgJSON(fullFieldName);
    String output = CQL2PGCLIMain.parseCQL(cql2pgjson, "instance", cql);
    String cli_output = CQL2PGCLIMain.handleOptions(args);
    logger.info(output);
    logger.info(cli_output);
    assertNotNull(output);
    assertEquals(output, cli_output);
}
Also used : CQL2PgJSON(org.folio.cql2pgjson.CQL2PgJSON) Test(org.junit.Test)

Aggregations

CQL2PgJSON (org.folio.cql2pgjson.CQL2PgJSON)25 CQLWrapper (org.folio.rest.persist.cql.CQLWrapper)22 JsonProcessingException (com.fasterxml.jackson.core.JsonProcessingException)16 Test (org.junit.Test)16 AsyncResult (io.vertx.core.AsyncResult)14 Future (io.vertx.core.Future)14 Handler (io.vertx.core.Handler)14 Vertx (io.vertx.core.Vertx)14 JsonArray (io.vertx.core.json.JsonArray)14 JsonObject (io.vertx.core.json.JsonObject)14 Async (io.vertx.ext.unit.Async)14 PgConnection (io.vertx.pgclient.PgConnection)14 Collections (java.util.Collections)14 UUID (java.util.UUID)14 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)14 Function (java.util.function.Function)14 PostgresTesterContainer (org.folio.postgres.testing.PostgresTesterContainer)14 ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)12 CompositeFuture (io.vertx.core.CompositeFuture)12 Promise (io.vertx.core.Promise)12