### Question:
BridgeOperationalState { public Optional<OvsdbBridgeAugmentation> getOvsdbBridgeAugmentation(InstanceIdentifier<?> iid) { Optional<Node> nodeOptional = getBridgeNode(iid); if (nodeOptional.isPresent()) { return Optional.ofNullable(nodeOptional.get().augmentation(OvsdbBridgeAugmentation.class)); } return Optional.empty(); } BridgeOperationalState(DataBroker db, DataChangeEvent changes);  BridgeOperationalState(DataBroker db, Collection<DataTreeModification<Node>> changes); @SuppressWarnings("IllegalCatch") Optional<Node> getBridgeNode(InstanceIdentifier<?> iid); Optional<OvsdbBridgeAugmentation> getOvsdbBridgeAugmentation(InstanceIdentifier<?> iid); Optional<TerminationPoint> getBridgeTerminationPoint(InstanceIdentifier<?> iid); Optional<OvsdbTerminationPointAugmentation> getOvsdbTerminationPointAugmentation(InstanceIdentifier<?> iid); Optional<ControllerEntry> getControllerEntry(InstanceIdentifier<?> iid); Optional<ProtocolEntry> getProtocolEntry(InstanceIdentifier<ProtocolEntry> iid);  }### Answer:
@Test public void testGetOvsdbBridgeAugmentation() throws Exception { Optional<OvsdbBridgeAugmentation> optOvsdbBri = briOperationState.getOvsdbBridgeAugmentation(nodeIid); verify(briOperationState, times(1)).getBridgeNode(any(InstanceIdentifier.class)); assertNotNull(optOvsdbBri); assertTrue(optOvsdbBri.equals(Optional.empty())); Node node = mock(Node.class); Optional<Node> optNode = Optional.of(node); doReturn(optNode).when(briOperationState).getBridgeNode(any(InstanceIdentifier.class)); OvsdbBridgeAugmentation ovsdbBriAug = mock(OvsdbBridgeAugmentation.class); doReturn(ovsdbBriAug).when(node).augmentation(OvsdbBridgeAugmentation.class); Optional<OvsdbBridgeAugmentation> ovsdbBriAugOptional = briOperationState.getOvsdbBridgeAugmentation(iid); assertNotNull(ovsdbBriAugOptional); assertTrue(ovsdbBriAugOptional.get() instanceof OvsdbBridgeAugmentation); }