Examples with SchemaNotFoundException org.apache.nifi.schema.access.SchemaNotFoundException used on opensource projects

Example 26 with SchemaNotFoundException

use of org.apache.nifi.schema.access.SchemaNotFoundException in project nifi by apache.

the class TestCSVHeaderSchemaStrategy method testSimple.

@Test
public void testSimple() throws SchemaNotFoundException, IOException {
    final String headerLine = "a, b, c, d, e\\,z, f";
    final byte[] headerBytes = headerLine.getBytes();
    final Map<PropertyDescriptor, String> properties = new HashMap<>();
    properties.put(CSVUtils.CSV_FORMAT, CSVUtils.CUSTOM.getValue());
    properties.put(CSVUtils.COMMENT_MARKER, "#");
    properties.put(CSVUtils.VALUE_SEPARATOR, ",");
    properties.put(CSVUtils.TRIM_FIELDS, "true");
    properties.put(CSVUtils.QUOTE_CHAR, "\"");
    properties.put(CSVUtils.ESCAPE_CHAR, "\\");
    final ConfigurationContext context = new MockConfigurationContext(properties, null);
    final CSVHeaderSchemaStrategy strategy = new CSVHeaderSchemaStrategy(context);
    final RecordSchema schema;
    try (final InputStream bais = new ByteArrayInputStream(headerBytes)) {
        schema = strategy.getSchema(null, bais, null);
    }
    final List<String> expectedFieldNames = Arrays.asList("a", "b", "c", "d", "e,z", "f");
    assertEquals(expectedFieldNames, schema.getFieldNames());
    assertTrue(schema.getFields().stream().allMatch(field -> field.getDataType().equals(RecordFieldType.STRING.getDataType())));
}