Examples with EntityBuilder com.haulmont.cuba.core.sys.jpql.model.EntityBuilder used on opensource projects

Example 41 with EntityBuilder

use of com.haulmont.cuba.core.sys.jpql.model.EntityBuilder in project cuba by cuba-platform.

the class HintProviderTest method requestHint_with_templateParam.

@Test
public void requestHint_with_templateParam() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    JpqlEntityModel driver = builder.produceImmediately("Driver", "name", "signal");
    builder.startNewEntity("Car");
    builder.addStringAttribute("model");
    builder.addCollectionReferenceAttribute("drivers", "Driver");
    JpqlEntityModel car = builder.produce();
    DomainModel model = new DomainModel(car, driver);
    HintProvider hintProvider = createTestHintProvider(model);
    HintResponse response = hintProvider.requestHint("select a.~ from Car a where a.model = ${param}");
    List<String> options = response.getOptions();
    assertEquals(2, options.size());
    assertEquals("drivers", options.get(0));
    assertEquals("model", options.get(1));
}

Example 42 with EntityBuilder

use of com.haulmont.cuba.core.sys.jpql.model.EntityBuilder in project cuba by cuba-platform.

the class HintProviderTest method requestHint_fieldNameHint_join_withCollections.

@Test
public void requestHint_fieldNameHint_join_withCollections() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    JpqlEntityModel personEntity = builder.produceImmediately("Person", "name");
    builder.startNewEntity("Team");
    builder.addStringAttribute("name");
    builder.addStringAttribute("owner");
    builder.addReferenceAttribute("manager", "Person");
    JpqlEntityModel teamEntity = builder.produce();
    builder.startNewEntity("Player");
    builder.addStringAttribute("name");
    builder.addStringAttribute("nickname");
    builder.addReferenceAttribute("team", "Team");
    builder.addReferenceAttribute("agent", "Person");
    JpqlEntityModel playerEntity = builder.produce();
    builder.startNewEntity("League");
    builder.addStringAttribute("name");
    builder.addCollectionReferenceAttribute("teams", "Team");
    JpqlEntityModel leagueEntity = builder.produce();
    DomainModel model = new DomainModel();
    model.add(teamEntity);
    model.add(playerEntity);
    model.add(leagueEntity);
    model.add(personEntity);
    HintProvider hintProvider = createTestHintProvider(model);
    HintResponse response = hintProvider.requestHint("select m.~ from League l " + "left join l.teams as t " + "left join t.manager as m");
    List<String> options = response.getOptions();
    assertEquals(1, options.size());
    assertEquals("name", options.get(0));
    hintProvider = createTestHintProvider(model);
    response = hintProvider.requestHint("select l from League l " + "join l.teams as t " + "join t.~");
    options = response.getOptions();
    assertEquals(1, options.size());
    assertEquals("manager", options.get(0));
}

Example 43 with EntityBuilder

use of com.haulmont.cuba.core.sys.jpql.model.EntityBuilder in project cuba by cuba-platform.

the class HintProviderTest method requestHint_with_variableRebinding.

@Test
public void requestHint_with_variableRebinding() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    JpqlEntityModel driver = builder.produceImmediately("Driver", "name", "signal");
    builder.startNewEntity("Car");
    builder.addStringAttribute("model");
    builder.addCollectionReferenceAttribute("drivers", "Driver");
    JpqlEntityModel car = builder.produce();
    DomainModel model = new DomainModel(car, driver);
    HintProvider hintProvider = createTestHintProvider(model);
    try {
        hintProvider.requestHint("select a.~ from Car a, in(a.drivers) a where a.model = ?1");
        Assert.fail();
    } catch (IllegalArgumentException ignored) {
    }
}

Example 44 with EntityBuilder

use of com.haulmont.cuba.core.sys.jpql.model.EntityBuilder in project cuba by cuba-platform.

the class HintProviderTest method requestHint_fieldNameHint_simple_referencedEntity_collections.

@Test
public void requestHint_fieldNameHint_simple_referencedEntity_collections() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    JpqlEntityModel teamEntity = builder.produceImmediately("Team", "name");
    builder.startNewEntity("Player");
    builder.addStringAttribute("name");
    builder.addStringAttribute("nickname");
    builder.addReferenceAttribute("team", "Team");
    JpqlEntityModel playerEntity = builder.produce();
    builder.startNewEntity("League");
    builder.addStringAttribute("name");
    builder.addCollectionReferenceAttribute("teams", "Team");
    JpqlEntityModel leagueEntity = builder.produce();
    DomainModel model = new DomainModel();
    model.add(teamEntity);
    model.add(playerEntity);
    model.add(leagueEntity);
    HintProvider hintProvider = createTestHintProvider(model);
    HintResponse response = hintProvider.requestHint("SELECT l.teams.~ FROM League l");
    List<String> options = response.getOptions();
    assertEquals(0, options.size());
    assertEquals("Query error", response.getErrorMessage());
    assertEquals(1, response.getCauseErrorMessages().size());
    assertEquals("Cannot get attribute of collection [l.teams.]", response.getCauseErrorMessages().get(0));
    response = hintProvider.requestHint("SELECT l.name FROM League l where l.teams.~");
    options = response.getOptions();
    assertEquals(0, options.size());
    assertEquals("Query error", response.getErrorMessage());
    assertEquals(1, response.getCauseErrorMessages().size());
    assertEquals("Cannot get attribute of collection [l.teams.]", response.getCauseErrorMessages().get(0));
    response = hintProvider.requestHint("SELECT l.name FROM League l where exists (select 1 from Team t where t.name = l.teams.n~");
    options = response.getOptions();
    assertEquals(0, options.size());
    assertEquals("Query error", response.getErrorMessage());
    assertEquals(1, response.getCauseErrorMessages().size());
    assertEquals("Cannot get attribute of collection [l.teams.n]", response.getCauseErrorMessages().get(0));
}

Example 45 with EntityBuilder

use of com.haulmont.cuba.core.sys.jpql.model.EntityBuilder in project cuba by cuba-platform.

the class HintProviderTest method requestHint_entityNameHint_subquery.

// The following functions are not supported by JPA2 JPQL (see jpql21.bnf)
// @Test
// public void requestHint_fieldNameHint_subqueryEntity() throws RecognitionException {
// EntityBuilder builder = new EntityBuilder();
// JpqlEntityModel teamEntity = builder.produceImmediately("Team", "name", "owner");
// 
// builder.startNewEntity("Player");
// builder.addStringAttribute("name");
// builder.addStringAttribute("nickname");
// builder.addReferenceAttribute("team", "Team");
// JpqlEntityModel playerEntity = builder.produce();
// 
// DomainModel model = new DomainModel();
// model.add(teamEntity);
// model.add(playerEntity);
// 
// HintProvider hintProvider = createTestHintProvider(model);
// 
// HintResponse response = hintProvider.requestHint("select p.~ from (select t from Team t) p");
// List<String> options = response.getOptions();
// assertEquals(2, options.size());
// assertEquals("name", options.get(0));
// assertEquals("owner", options.get(1));
// 
// response = hintProvider.requestHint("select p.~ from (select t.name from Team t) p");
// options = response.getOptions();
// assertEquals(1, options.size());
// assertEquals("name", options.get(0));
// 
// response = hintProvider.requestHint("select p.t~ from (select t.name from Team t) p");
// options = response.getOptions();
// assertEquals(0, options.size());
// 
// response = hintProvider.requestHint("select p.o~ from (select t.name, t.owner from Team t) p");
// options = response.getOptions();
// assertEquals(1, options.size());
// assertEquals("owner", options.get(0));
// 
// response = hintProvider.requestHint("select g.o~ from (select t.name, t.owner from Team t) p, (select t.name from Team t) g");
// options = response.getOptions();
// assertEquals(0, options.size());
// }
// The following functions are not supported by JPA2 JPQL (see jpql21.bnf)
// @Test
// public void requestHint_fieldNameHint_severalLevelsOfSubquery() throws RecognitionException {
// EntityBuilder builder = new EntityBuilder();
// JpqlEntityModel teamEntity = builder.produceImmediately("Team", "name", "owner");
// 
// builder.startNewEntity("Player");
// builder.addStringAttribute("name");
// builder.addStringAttribute("nickname");
// builder.addReferenceAttribute("team", "Team");
// JpqlEntityModel playerEntity = builder.produce();
// 
// DomainModel model = new DomainModel();
// model.add(teamEntity);
// model.add(playerEntity);
// 
// HintProvider hintProvider = createTestHintProvider(model);
// 
// HintResponse response = hintProvider.requestHint(
// "select p.~ from " +
// "(select t from Team t " +
// "where t.name in " +
// "   (select a.name from Player a)" +
// ") p");
// List<String> options = response.getOptions();
// assertEquals(2, options.size());
// assertEquals("name", options.get(0));
// assertEquals("owner", options.get(1));
// 
// response = hintProvider.requestHint("select p.owner from (select t.o~ from Team t where t.name in (select a.name from Player a)) p");
// options = response.getOptions();
// assertEquals(1, options.size());
// assertEquals("owner", options.get(0));
// 
// response = hintProvider.requestHint("select p.owner from (select t from Team t where t.name in (select a.~ from Player a)) p");
// options = response.getOptions();
// assertEquals(3, options.size());
// assertEquals("name", options.get(0));
// assertEquals("nickname", options.get(1));
// assertEquals("team", options.get(2));
// 
// response = hintProvider.requestHint("select p.owner from (select t from Team t where t.~ in (select a.name from Player a)) p");
// options = response.getOptions();
// assertEquals(2, options.size());
// assertEquals("name", options.get(0));
// assertEquals("owner", options.get(1));
// 
// response = hintProvider.requestHint("select p.owner from (select a.ni~ from Team t where t.name in (select a.name from Player a)) p");
// options = response.getOptions();
// assertEquals(0, options.size());
// assertEquals("Query error", response.getErrorMessage());
// assertEquals(1, response.getCauseErrorMessages().size());
// assertEquals("Cannot parse [a.ni]", response.getCauseErrorMessages().get(0));
// }
// The following functions are not supported by JPA2 JPQL (see jpql21.bnf)
// @Test
// public void requestHint_fieldNameHint_subqueries_using_AS() throws RecognitionException {
// EntityBuilder builder = new EntityBuilder();
// JpqlEntityModel teamEntity = builder.produceImmediately("Team", "name", "owner");
// 
// builder.startNewEntity("Player");
// builder.addStringAttribute("name");
// builder.addStringAttribute("nickname");
// builder.addReferenceAttribute("team", "Team");
// JpqlEntityModel playerEntity = builder.produce();
// 
// DomainModel model = new DomainModel();
// model.add(teamEntity);
// model.add(playerEntity);
// 
// HintProvider hintProvider = createTestHintProvider(model);
// 
// HintResponse response = hintProvider.requestHint(
// "select p.~ from (select t.owner from Team as t where t.name) p");
// List<String> options = response.getOptions();
// assertEquals(1, options.size());
// assertEquals("owner", options.get(0));
// }
@Test
public void requestHint_entityNameHint_subquery() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    JpqlEntityModel teamEntity = builder.produceImmediately("Team", "name", "owner");
    builder.startNewEntity("Player");
    builder.addStringAttribute("name");
    builder.addStringAttribute("nickname");
    builder.addReferenceAttribute("team", "Team");
    JpqlEntityModel playerEntity = builder.produce();
    DomainModel model = new DomainModel();
    model.add(teamEntity);
    model.add(playerEntity);
    HintProvider hintProvider = createTestHintProvider(model);
    HintResponse response = hintProvider.requestHint("select p from (select t from T~) p");
    List<String> options = response.getOptions();
    assertEquals(1, options.size());
    assertEquals("Team", options.get(0));
}