Search in sources :

Example 6 with VariantTypeNode

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

the class XmlBxmlNodeVisitorTest method testVisitAttributeConditionalSubstitutionNode.

@Test
public void testVisitAttributeConditionalSubstitutionNode() 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);
    ConditionalSubstitutionNode conditionalSubstitutionNode = mock(ConditionalSubstitutionNode.class);
    when(attributeNode.getAttributeName()).thenReturn(attributeName);
    when(attributeNode.getValue()).thenReturn(conditionalSubstitutionNode);
    doAnswer(invocation -> {
        ((BxmlNodeVisitor) invocation.getArguments()[0]).visit(conditionalSubstitutionNode);
        return null;
    }).when(conditionalSubstitutionNode).accept(any(BxmlNodeVisitor.class));
    when(conditionalSubstitutionNode.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) ConditionalSubstitutionNode(org.apache.nifi.processors.evtx.parser.bxml.ConditionalSubstitutionNode) Test(org.junit.Test)

Aggregations

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