Examples with FlowSerializer org.apache.nifi.controller.serialization.FlowSerializer used on opensource projects

Example 1 with FlowSerializer

use of org.apache.nifi.controller.serialization.FlowSerializer in project nifi by apache.

the class FingerprintFactoryTest method serializeElement.

private <T> Element serializeElement(final StringEncryptor encryptor, final Class<T> componentClass, final T component, final String serializerMethodName, ScheduledStateLookup scheduledStateLookup) throws Exception {
    final DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
    final DocumentBuilder docBuilder = docFactory.newDocumentBuilder();
    final Document doc = docBuilder.newDocument();
    final FlowSerializer flowSerializer = new StandardFlowSerializer(encryptor);
    final Method serializeMethod = StandardFlowSerializer.class.getDeclaredMethod(serializerMethodName, Element.class, componentClass, ScheduledStateLookup.class);
    serializeMethod.setAccessible(true);
    final Element rootElement = doc.createElement("root");
    serializeMethod.invoke(flowSerializer, rootElement, component, scheduledStateLookup);
    return rootElement;
}

Example 2 with FlowSerializer

use of org.apache.nifi.controller.serialization.FlowSerializer in project nifi by apache.

the class StandardFlowServiceTest method testLoadExistingFlow.

@Test
public void testLoadExistingFlow() throws IOException {
    byte[] flowBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow.xml"));
    flowService.load(new StandardDataFlow(flowBytes, null, null, new HashSet<>()));
    flowBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow-inheritable.xml"));
    flowService.load(new StandardDataFlow(flowBytes, null, null, new HashSet<>()));
    FlowSerializer serializer = new StandardFlowSerializer(mockEncryptor);
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    serializer.serialize(flowController, baos, ScheduledStateLookup.IDENTITY_LOOKUP);
    String expectedFlow = new String(flowBytes).trim();
    String actualFlow = new String(baos.toByteArray()).trim();
    Assert.assertEquals(expectedFlow, actualFlow);
}

Example 3 with FlowSerializer

use of org.apache.nifi.controller.serialization.FlowSerializer in project nifi by apache.

the class StandardFlowServiceTest method testLoadExistingFlowWithCorruptFlow.

@Test
public void testLoadExistingFlowWithCorruptFlow() throws IOException {
    byte[] originalBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow.xml"));
    flowService.load(new StandardDataFlow(originalBytes, null, null, new HashSet<>()));
    try {
        byte[] updatedBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow-corrupt.xml"));
        flowService.load(new StandardDataFlow(updatedBytes, null, null, new HashSet<>()));
        fail("should have thrown " + FlowSerializationException.class);
    } catch (FlowSerializationException ufe) {
        FlowSerializer serializer = new StandardFlowSerializer(mockEncryptor);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        serializer.serialize(flowController, baos, ScheduledStateLookup.IDENTITY_LOOKUP);
        String expectedFlow = new String(originalBytes).trim();
        String actualFlow = new String(baos.toByteArray()).trim();
        Assert.assertEquals(expectedFlow, actualFlow);
    }
}

Example 4 with FlowSerializer

use of org.apache.nifi.controller.serialization.FlowSerializer in project nifi by apache.

the class StandardFlowServiceTest method testLoadWithFlow.

@Test
public void testLoadWithFlow() throws IOException {
    byte[] flowBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow.xml"));
    flowService.load(new StandardDataFlow(flowBytes, null, null, new HashSet<>()));
    FlowSerializer serializer = new StandardFlowSerializer(mockEncryptor);
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    serializer.serialize(flowController, baos, ScheduledStateLookup.IDENTITY_LOOKUP);
    String expectedFlow = new String(flowBytes).trim();
    String actualFlow = new String(baos.toByteArray()).trim();
    Assert.assertEquals(expectedFlow, actualFlow);
}

Example 5 with FlowSerializer

use of org.apache.nifi.controller.serialization.FlowSerializer in project nifi by apache.

the class StandardFlowServiceTest method testLoadExistingFlowWithUninheritableFlow.

@Test
public void testLoadExistingFlowWithUninheritableFlow() throws IOException {
    byte[] originalBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow.xml"));
    flowService.load(new StandardDataFlow(originalBytes, null, null, new HashSet<>()));
    try {
        byte[] updatedBytes = IOUtils.toByteArray(StandardFlowServiceTest.class.getResourceAsStream("/conf/all-flow-uninheritable.xml"));
        flowService.load(new StandardDataFlow(updatedBytes, null, null, new HashSet<>()));
        fail("should have thrown " + UninheritableFlowException.class);
    } catch (UninheritableFlowException ufe) {
        FlowSerializer serializer = new StandardFlowSerializer(mockEncryptor);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        serializer.serialize(flowController, baos, ScheduledStateLookup.IDENTITY_LOOKUP);
        String expectedFlow = new String(originalBytes).trim();
        String actualFlow = new String(baos.toByteArray()).trim();
        Assert.assertEquals(expectedFlow, actualFlow);
    }
}