Search in sources :

Example 1 with NormalSubstitutionNode

use of org.apache.nifi.processors.evtx.parser.bxml.NormalSubstitutionNode in project nifi by apache.

the class XmlBxmlNodeVisitorTest method testVisitAttributeNormalSubstitutionNode.

@Test
public void testVisitAttributeNormalSubstitutionNode() throws IOException, XMLStreamException {
    String attributeName = "attributeName";
    String attributeValue = "attributeValue";
    VariantTypeNode sub = mock(VariantTypeNode.class);
    when(sub.getValue()).thenReturn(attributeValue);
    substitutions.add(sub);
    AttributeNode attributeNode = mock(AttributeNode.class);
    NormalSubstitutionNode normalSubstitutionNode = mock(NormalSubstitutionNode.class);
    when(attributeNode.getAttributeName()).thenReturn(attributeName);
    when(attributeNode.getValue()).thenReturn(normalSubstitutionNode);
    doAnswer(invocation -> {
        ((BxmlNodeVisitor) invocation.getArguments()[0]).visit(normalSubstitutionNode);
        return null;
    }).when(normalSubstitutionNode).accept(any(BxmlNodeVisitor.class));
    when(normalSubstitutionNode.getIndex()).thenReturn(0);
    xmlBxmlNodeVisitor.visit(attributeNode);
    verify(xmlStreamWriter).writeAttribute(attributeName, attributeValue);
}
Also used : BxmlNodeVisitor(org.apache.nifi.processors.evtx.parser.BxmlNodeVisitor) AttributeNode(org.apache.nifi.processors.evtx.parser.bxml.AttributeNode) VariantTypeNode(org.apache.nifi.processors.evtx.parser.bxml.value.VariantTypeNode) NormalSubstitutionNode(org.apache.nifi.processors.evtx.parser.bxml.NormalSubstitutionNode) Test(org.junit.Test)

Example 2 with NormalSubstitutionNode

use of org.apache.nifi.processors.evtx.parser.bxml.NormalSubstitutionNode in project nifi by apache.

the class XmlBxmlNodeVisitorTest method testVisitNormalSubstitutionNode.

@Test
public void testVisitNormalSubstitutionNode() throws IOException {
    NormalSubstitutionNode normalSubstitutionNode = mock(NormalSubstitutionNode.class);
    VariantTypeNode sub = mock(VariantTypeNode.class);
    substitutions.add(sub);
    when(normalSubstitutionNode.getIndex()).thenReturn(0);
    xmlBxmlNodeVisitor.visit(normalSubstitutionNode);
    verify(sub).accept(xmlBxmlNodeVisitor);
}
Also used : NormalSubstitutionNode(org.apache.nifi.processors.evtx.parser.bxml.NormalSubstitutionNode) VariantTypeNode(org.apache.nifi.processors.evtx.parser.bxml.value.VariantTypeNode) Test(org.junit.Test)

Aggregations

NormalSubstitutionNode (org.apache.nifi.processors.evtx.parser.bxml.NormalSubstitutionNode)2 VariantTypeNode (org.apache.nifi.processors.evtx.parser.bxml.value.VariantTypeNode)2 Test (org.junit.Test)2 BxmlNodeVisitor (org.apache.nifi.processors.evtx.parser.BxmlNodeVisitor)1 AttributeNode (org.apache.nifi.processors.evtx.parser.bxml.AttributeNode)1