BUG-6858: adapt to ise api, change lookup from ise
[groupbasedpolicy.git] / sxp-integration / sxp-ep-provider / src / test / java / org / opendaylight / groupbasedpolicy / sxp / ep / provider / impl / dao / EPPolicyTemplateDaoFacadeImplTest.java
index 532da14ddb16a9140a3f22bb6020c28e5a2c3b67..9150b9f646acde71815eb18e5c7b4da5aa44584e 100644 (file)
@@ -101,7 +101,7 @@ public class EPPolicyTemplateDaoFacadeImplTest {
                 .build();
 
         Mockito.when(delegateDao.read(SGT)).thenReturn(Futures.immediateFuture(Optional.absent()));
-        Mockito.when(provider.provideTemplate(SGT)).thenReturn(java.util.Optional.of(template));
+        Mockito.when(provider.provideTemplate(SGT)).thenReturn(Futures.immediateFuture(java.util.Optional.of(template)));
         Mockito.when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
         facade.setTemplateProvider(provider);
 
@@ -124,7 +124,7 @@ public class EPPolicyTemplateDaoFacadeImplTest {
     @Test
     public void testRead_failProviderMiss() throws Exception {
         Mockito.when(delegateDao.read(SGT)).thenReturn(Futures.immediateFuture(Optional.absent()));
-        Mockito.when(provider.provideTemplate(SGT)).thenReturn(java.util.Optional.empty());
+        Mockito.when(provider.provideTemplate(SGT)).thenReturn(Futures.immediateFuture(java.util.Optional.empty()));
         facade.setTemplateProvider(provider);
 
 
@@ -141,8 +141,7 @@ public class EPPolicyTemplateDaoFacadeImplTest {
                 .build();
 
         Mockito.when(delegateDao.read(SGT)).thenReturn(Futures.immediateFuture(Optional.absent()));
-        Mockito.when(provider.provideTemplate(SGT)).thenReturn(java.util.Optional.of(template));
-        Mockito.when(wTx.submit()).thenReturn(Futures.immediateFailedCheckedFuture(TX_EXCEPTION));
+        Mockito.when(provider.provideTemplate(SGT)).thenReturn(Futures.immediateFailedCheckedFuture(TX_EXCEPTION));
         facade.setTemplateProvider(provider);
 
         final ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> actual = facade.read(SGT);