use of example.PlayerStatsView in project elide by yahoo.
the class QueryEngineTest method testAllArgumentQuery.
/**
* Test group by, having, dimension, metric at the same time.
*
* @throws Exception exception
*/
@Test
public void testAllArgumentQuery() throws Exception {
Map<String, Sorting.SortOrder> sortMap = new TreeMap<>();
sortMap.put("countryName", Sorting.SortOrder.asc);
Query query = Query.builder().source(playerStatsViewTable).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).dimensionProjection(playerStatsViewTable.getDimensionProjection("countryName")).whereFilter(filterParser.parseFilterExpression("countryName=='United States'", playerStatsViewType, false)).havingFilter(filterParser.parseFilterExpression("highScore > 300", playerStatsViewType, false)).sorting(new SortingImpl(sortMap, PlayerStatsView.class, dictionary)).arguments(playerStatsViewTableArgs).build();
List<Object> results = toList(engine.executeQuery(query, transaction).getData());
PlayerStatsView stats2 = new PlayerStatsView();
stats2.setId("0");
stats2.setHighScore(2412);
stats2.setCountryName("United States");
assertEquals(ImmutableList.of(stats2), results);
}
use of example.PlayerStatsView in project elide by yahoo.
the class QueryEngineTest method testFromSubQuery.
/**
* Test loading records using {@link FromSubquery}.
*
* @throws Exception exception
*/
@Test
public void testFromSubQuery() throws Exception {
Query query = Query.builder().source(playerStatsViewTable.toQueryable()).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).arguments(playerStatsViewTableArgs).build();
List<Object> results = toList(engine.executeQuery(query, transaction).getData());
PlayerStatsView stats2 = new PlayerStatsView();
stats2.setId("0");
stats2.setHighScore(2412);
assertEquals(ImmutableList.of(stats2), results);
}
use of example.PlayerStatsView in project elide by yahoo.
the class SQLUnitTest method init.
public static void init(SQLDialect sqlDialect, Set<Optimizer> optimizers, MetaDataStore metaDataStore) {
Properties properties = new Properties();
properties.put("driverClassName", "org.h2.Driver");
String jdbcUrl = "jdbc:h2:mem:db1;DB_CLOSE_DELAY=-1" + ";NON_KEYWORDS=VALUE,USER" + ";DATABASE_TO_UPPER=FALSE" + getCompatabilityMode(sqlDialect.getDialectType());
properties.put("jdbcUrl", jdbcUrl);
HikariConfig config = new HikariConfig(properties);
DataSource dataSource = new HikariDataSource(config);
try (Connection h2Conn = dataSource.getConnection()) {
h2Conn.createStatement().execute("RUNSCRIPT FROM 'classpath:prepare_tables.sql'");
} catch (SQLException e) {
((HikariDataSource) dataSource).close();
throw new IllegalStateException(e);
}
SQLUnitTest.metaDataStore = metaDataStore;
dictionary = EntityDictionary.builder().build();
dictionary.bindEntity(PlayerStatsWithView.class);
dictionary.bindEntity(PlayerStatsView.class);
dictionary.bindEntity(PlayerStats.class);
dictionary.bindEntity(Country.class);
dictionary.bindEntity(SubCountry.class);
dictionary.bindEntity(Player.class);
dictionary.bindEntity(CountryView.class);
dictionary.bindEntity(CountryViewNested.class);
dictionary.bindEntity(Continent.class);
dictionary.bindEntity(GameRevenue.class);
filterParser = RSQLFilterDialect.builder().dictionary(dictionary).build();
// Manually register the serdes because we are not running a complete Elide service.
CoerceUtil.register(Day.class, new Day.DaySerde());
CoerceUtil.register(Hour.class, new Hour.HourSerde());
CoerceUtil.register(ISOWeek.class, new ISOWeek.ISOWeekSerde());
CoerceUtil.register(Minute.class, new Minute.MinuteSerde());
CoerceUtil.register(Month.class, new Month.MonthSerde());
CoerceUtil.register(Quarter.class, new Quarter.QuarterSerde());
CoerceUtil.register(Second.class, new Second.SecondSerde());
CoerceUtil.register(Week.class, new Week.WeekSerde());
CoerceUtil.register(Year.class, new Year.YearSerde());
metaDataStore.populateEntityDictionary(dictionary);
// Need to provide details for connections used by all available models.
Map<String, ConnectionDetails> connectionDetailsMap = new HashMap<>();
connectionDetailsMap.put("mycon", new ConnectionDetails(dataSource, sqlDialect));
connectionDetailsMap.put("SalesDBConnection", new ConnectionDetails(DUMMY_DATASOURCE, sqlDialect));
Function<String, ConnectionDetails> connectionLookup = (name) -> connectionDetailsMap.getOrDefault(name, new ConnectionDetails(dataSource, sqlDialect));
engine = new SQLQueryEngine(metaDataStore, connectionLookup, optimizers, new DefaultQueryPlanMerger(metaDataStore), new DefaultQueryValidator(metaDataStore.getMetadataDictionary()));
playerStatsTable = (SQLTable) metaDataStore.getTable("playerStats", NO_VERSION);
videoGameTable = (SQLTable) metaDataStore.getTable("videoGame", NO_VERSION);
playerStatsViewTable = (SQLTable) metaDataStore.getTable("playerStatsView", NO_VERSION);
playerStatsViewTableArgs = new HashMap<>();
playerStatsViewTableArgs.put("rating", Argument.builder().name("overallRating").value("Great").build());
playerStatsViewTableArgs.put("minScore", Argument.builder().name("minScore").value("0").build());
}
use of example.PlayerStatsView in project elide by yahoo.
the class QueryEngineTest method testNotProjectedFilter.
/**
* Test filtering on an attribute that's not present in the query.
*
* @throws Exception exception
*/
@Test
public void testNotProjectedFilter() throws Exception {
Query query = Query.builder().source(playerStatsViewTable).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).whereFilter(filterParser.parseFilterExpression("countryName=='United States'", playerStatsViewType, false)).arguments(playerStatsViewTableArgs).build();
List<Object> results = toList(engine.executeQuery(query, transaction).getData());
PlayerStatsView stats2 = new PlayerStatsView();
stats2.setId("0");
stats2.setHighScore(2412);
assertEquals(ImmutableList.of(stats2), results);
}
Aggregations