+ private SalFlowService flowService;
+ @Mock
+ private DataTreeModification<FlowCapableNode> dataTreeModification;
+ @Captor
+ private ArgumentCaptor<AddFlowInput> addFlowInputCaptor;
+
+ @Before
+ public void setUp() {
+ lldpPacketPuntEnforcer = new LLDPPacketPuntEnforcer(flowService, Mockito.mock(DataBroker.class));
+ final DataTreeIdentifier<FlowCapableNode> identifier = new DataTreeIdentifier(LogicalDatastoreType.OPERATIONAL, nodeIID);
+ Mockito.when(dataTreeModification.getRootPath()).thenReturn(identifier);
+ Mockito.when(dataTreeModification.getRootNode()).thenReturn(Mockito.mock(DataObjectModification.class));
+ Mockito.when(dataTreeModification.getRootNode().getModificationType()).thenReturn(ModificationType.WRITE);
+ }