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.rev211210.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;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.service.rev230526.OrgOpenroadmServiceService;
@ExtendWith(MockitoExtension.class)
public class TapiProviderTest {
@Mock
private ServiceDataStoreOperations serviceDataStoreOperations;
@Mock
- private TransportpceTapinetworkutilsService tapiNetworkUtils;
- @Mock
- private TapiNotificationListener tapiNetworkModelListenerImpl;
+ private TapiNetworkModelNotificationHandler tapiNetworkModelNotificationHandler;
@Mock
private TapiNetworkModelService tapiNetworkModelServiceImpl;
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);
+ networkTransactionService, serviceHandler, 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(rpcProviderRegistry, times(2)).registerRpcImplementations(any());
verify(dataBroker, times(4)).registerDataTreeChangeListener(any(), any());
}
-}
\ No newline at end of file
+}