+ DataBroker dataBroker = mock(DataBroker.class);
+ UniAugmentation uni = PowerMockito.mock(UniAugmentation.class);
+ InstanceIdentifier<?> uniKey = PowerMockito.mock(InstanceIdentifier.class);
+ InstanceIdentifier<Node> ovsdbNodeIid = mock(InstanceIdentifier.class);
+ Optional<Node> optionalNode = mock(Optional.class, Mockito.RETURNS_MOCKS);
+ Node nd = mock(Node.class, Mockito.RETURNS_MOCKS);
+ when(optionalNode.isPresent()).thenReturn(true);
+ when(optionalNode.get()).thenReturn(nd);
+ WriteTransaction transaction = mock(WriteTransaction.class);
+ when(dataBroker.newWriteOnlyTransaction()).thenReturn(transaction);
+ PowerMockito.suppress(MemberMatcher.method(UnimgrUtils.class, "readNode", DataBroker.class, LogicalDatastoreType.class, InstanceIdentifier.class));
+ when(UnimgrUtils.readNode(any(DataBroker.class), any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(optionalNode);
+ UnimgrUtils.updateUniNode(LogicalDatastoreType.OPERATIONAL, uniKey, uni, ovsdbNodeIid, dataBroker);
+ verify(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(Node.class));
+ verify(transaction).submit();
+ }
+
+ @Test
+ public void testUpdateUniNode2() {
+ DataBroker dataBroker = mock(DataBroker.class);
+ UniAugmentation uni = PowerMockito.mock(UniAugmentation.class);
+ InstanceIdentifier<?> uniKey = PowerMockito.mock(InstanceIdentifier.class);
+ Node ovsdbNode = mock(Node.class);
+ InstanceIdentifier<Node> ovsdbNodeIid = mock(InstanceIdentifier.class);
+ Optional<Node> optionalNode = mock(Optional.class, Mockito.RETURNS_MOCKS);
+ Node nd = mock(Node.class, Mockito.RETURNS_MOCKS);
+ when(optionalNode.isPresent()).thenReturn(true);
+ when(optionalNode.get()).thenReturn(nd);
+ WriteTransaction transaction = mock(WriteTransaction.class);
+ when(dataBroker.newWriteOnlyTransaction()).thenReturn(transaction);
+ PowerMockito.suppress(MemberMatcher.method(UnimgrMapper.class, "getOvsdbNodeIid", NodeId.class));
+ when(UnimgrMapper.getOvsdbNodeIid(any(NodeId.class))).thenReturn(ovsdbNodeIid);
+ PowerMockito.suppress(MemberMatcher.method(UnimgrUtils.class, "readNode", DataBroker.class, LogicalDatastoreType.class, InstanceIdentifier.class));
+ when(UnimgrUtils.readNode(any(DataBroker.class), any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(optionalNode);
+ UnimgrUtils.updateUniNode(LogicalDatastoreType.OPERATIONAL, uniKey, uni, ovsdbNode, dataBroker);
+ verify(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(Node.class));
+ verify(transaction).submit();