import static org.mockito.Mockito.when;
import static org.opendaylight.mdsal.common.api.CommitInfo.emptyFluentFuture;
+import com.google.common.collect.ClassToInstanceMap;
import com.google.common.util.concurrent.Futures;
import java.util.Optional;
import org.junit.jupiter.api.Test;
import org.opendaylight.mdsal.binding.api.NotificationPublishService;
import org.opendaylight.mdsal.binding.api.NotificationService;
import org.opendaylight.mdsal.binding.api.RpcProviderService;
+import org.opendaylight.mdsal.binding.api.RpcService;
import org.opendaylight.transportpce.common.network.NetworkTransactionService;
import org.opendaylight.transportpce.servicehandler.service.ServiceDataStoreOperations;
import org.opendaylight.transportpce.tapi.impl.TapiProvider;
+import org.opendaylight.transportpce.tapi.listeners.TapiNetworkModelNotificationHandler;
import org.opendaylight.transportpce.tapi.topology.TapiNetworkModelService;
-import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.tapinetworkutils.rev210408.TransportpceTapinetworkutilsService;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.service.rev230526.OrgOpenroadmServiceService;
-import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.common.rev181210.TapiCommonService;
-import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev181210.TapiConnectivityService;
-import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.notification.rev181210.TapiNotificationListener;
-import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.topology.rev181210.TapiTopologyService;
@ExtendWith(MockitoExtension.class)
public class TapiProviderTest {
@Mock
private DataBroker dataBroker;
@Mock
- private RpcProviderService rpcProviderRegistry;
+ private RpcProviderService rpcProviderService;
@Mock
private NotificationService notificationService;
@Mock
+ private RpcService rpcService;
+ @Mock
private NotificationPublishService notificationPublishService;
@Mock
private NetworkTransactionService networkTransactionService;
@Mock
- private OrgOpenroadmServiceService serviceHandler;
- @Mock
private ServiceDataStoreOperations serviceDataStoreOperations;
@Mock
- private TransportpceTapinetworkutilsService tapiNetworkUtils;
- @Mock
- private TapiNotificationListener tapiNetworkModelListenerImpl;
+ private TapiNetworkModelNotificationHandler tapiNetworkModelNotificationHandler;
@Mock
private TapiNetworkModelService tapiNetworkModelServiceImpl;
void testInitRegisterTapiToRpcRegistry() {
when(networkTransactionService.read(any(), any())).thenReturn(Futures.immediateFuture(Optional.empty()));
doReturn(emptyFluentFuture()).when(networkTransactionService).commit();
- new TapiProvider(dataBroker, rpcProviderRegistry, notificationService, notificationPublishService,
- networkTransactionService, serviceHandler, serviceDataStoreOperations, tapiNetworkUtils,
- tapiNetworkModelListenerImpl, tapiNetworkModelServiceImpl);
+ new TapiProvider(dataBroker, rpcProviderService, rpcService, notificationService, notificationPublishService,
+ networkTransactionService, serviceDataStoreOperations,
+ tapiNetworkModelNotificationHandler, tapiNetworkModelServiceImpl);
- verify(rpcProviderRegistry, times(1)).registerRpcImplementation(any(), any(TapiConnectivityService.class));
- verify(rpcProviderRegistry, times(2)).registerRpcImplementation(any(), any(TapiTopologyService.class));
- verify(rpcProviderRegistry, times(2)).registerRpcImplementation(any(), any(TapiCommonService.class));
- verify(dataBroker, times(4)).registerDataTreeChangeListener(any(), any());
+ verify(rpcProviderService, times(2)).registerRpcImplementations(any(ClassToInstanceMap.class));
+ verify(dataBroker, times(4)).registerTreeChangeListener(any(), any());
}
-}
\ No newline at end of file
+}