- PowerMockito.stub(PowerMockito.method(PolicyWriterUtil.class, "netconfRead")).toReturn(null);
- Assert.assertFalse(PolicyWriterUtil.writePolicyMap(policyMap, getLocation()));
+ PowerMockito.stub(PowerMockito.method(PolicyWriterUtil.class, NETCONF_READ)).toReturn(null);
+ assertFalse(PolicyWriterUtil.writePolicyMap(policyMap, getLocation()));
+ }
+
+ @Test
+ public void testRemovePolicyMap() throws Exception {
+ final PolicyMap policyMap = new PolicyMapBuilder().setName(UNIT_POLICY_MAP_NAME).build();
+ PowerMockito.stub(PowerMockito.method(NetconfTransactionCreator.class, NETCONF_WRITE_ONLY_TRANSACTION)).toReturn(wTxOptional);
+ when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
+
+ PowerMockito.stub(PowerMockito.method(NetconfTransactionCreator.class, NETCONF_READ_ONLY_TRANSACTION)).toReturn(rTxOptional);
+ when(rTx.read(eq(LogicalDatastoreType.CONFIGURATION), Matchers.<InstanceIdentifier<ClassMap>>any()))
+ .thenReturn(Futures.immediateCheckedFuture(Optional.of(
+ new ClassMapBuilder().setName(CLASS_MAP_NAME).build())));
+ PolicyWriterUtil.writePolicyMap(policyMap, getLocation());
+
+ LOG.debug("scenario: fail to remove");
+ PowerMockito.stub(PowerMockito.method(PolicyWriterUtil.class, NETCONF_READ)).toReturn(new ClassBuilder().build());
+
+ assertFalse(PolicyWriterUtil.removePolicyMap(getLocation()));
+
+ LOG.debug("scenario: remove succeed");
+ PowerMockito.stub(PowerMockito.method(PolicyWriterUtil.class, NETCONF_READ)).toReturn(null);
+
+ assertTrue(PolicyWriterUtil.removePolicyMap(getLocation()));