Examples with DataModel org.apache.phoenix.pherf.configuration.DataModel used on opensource projects

Example 6 with DataModel

use of org.apache.phoenix.pherf.configuration.DataModel in project phoenix by apache.

the class SchemaReaderIT method assertApplySchemaTest.

private void assertApplySchemaTest() {
    try {
        util.setZookeeper("localhost");
        SchemaReader reader = new SchemaReader(util, ".*datamodel/.*test.*sql");
        List<Path> resources = new ArrayList<>(reader.getResourceList());
        assertTrue("Could not pull list of schema files.", resources.size() > 0);
        assertNotNull("Could not read schema file.", this.getClass().getResourceAsStream(PherfConstants.RESOURCE_DATAMODEL + "/" + resources.get(0).getFileName().toString()));
        assertNotNull("Could not read schema file.", reader.resourceToString(resources.get(0)));
        try {
            reader.applySchema();
        } catch (SQLException e) {
            fail("Failed to apply schema " + e.getMessage());
        }
        Connection connection = null;
        URL resourceUrl = getClass().getResource("/scenario/test_scenario.xml");
        assertNotNull("Test data XML file is missing", resourceUrl);
        connection = util.getConnection();
        Path resourcePath = Paths.get(resourceUrl.toURI());
        DataModel data = XMLConfigParser.readDataModel(resourcePath);
        List<Scenario> scenarioList = data.getScenarios();
        Scenario scenario = scenarioList.get(0);
        List<Column> columnList = util.getColumnsFromPhoenix(scenario.getSchemaName(), scenario.getTableNameWithoutSchemaName(), connection);
        assertTrue("Could not retrieve Metadata from Phoenix", columnList.size() > 0);
    } catch (Exception e) {
        fail("Could not initialize SchemaReader");
        e.printStackTrace();
    }
}