Examples with NetconfDeviceSchemasResolver org.opendaylight.netconf.sal.connect.api.NetconfDeviceSchemasResolver used on opensource projects

Example 1 with NetconfDeviceSchemasResolver

use of org.opendaylight.netconf.sal.connect.api.NetconfDeviceSchemasResolver in project netconf by opendaylight.

the class NetconfDeviceTest method testNetconfDeviceMissingSource.

@Test
public void testNetconfDeviceMissingSource() throws Exception {
    final RemoteDeviceHandler<NetconfSessionPreferences> facade = getFacade();
    final NetconfDeviceCommunicator listener = getListener();
    final EffectiveModelContextFactory schemaFactory = getSchemaFactory();
    final SchemaRepository schemaRepository = getSchemaRepository();
    // Make fallback attempt to fail due to empty resolved sources
    final MissingSchemaSourceException schemaResolutionException = new MissingSchemaSourceException("fail first", TEST_SID);
    doReturn(Futures.immediateFailedFuture(schemaResolutionException)).when(schemaRepository).getSchemaSource(eq(TEST_SID), eq(YangTextSchemaSource.class));
    doAnswer(invocation -> {
        if (((Collection<?>) invocation.getArguments()[0]).size() == 2) {
            return Futures.immediateFailedFuture(schemaResolutionException);
        } else {
            return Futures.immediateFuture(SCHEMA_CONTEXT);
        }
    }).when(schemaFactory).createEffectiveModelContext(anyCollection());
    final NetconfDeviceSchemasResolver stateSchemasResolver = (deviceRpc, remoteSessionCapabilities, id, schemaContext) -> {
        final Module first = Iterables.getFirst(SCHEMA_CONTEXT.getModules(), null);
        final QName qName = QName.create(first.getQNameModule(), first.getName());
        final NetconfStateSchemas.RemoteYangSchema source1 = new NetconfStateSchemas.RemoteYangSchema(qName);
        final NetconfStateSchemas.RemoteYangSchema source2 = new NetconfStateSchemas.RemoteYangSchema(QName.create(first.getQNameModule(), "test-module2"));
        return new NetconfStateSchemas(Sets.newHashSet(source1, source2));
    };
    final NetconfDevice.SchemaResourcesDTO schemaResourcesDTO = new NetconfDevice.SchemaResourcesDTO(getSchemaRegistry(), schemaRepository, schemaFactory, stateSchemasResolver);
    final NetconfDevice device = new NetconfDeviceBuilder().setReconnectOnSchemasChange(true).setSchemaResourcesDTO(schemaResourcesDTO).setGlobalProcessingExecutor(getExecutor()).setBaseSchemas(BASE_SCHEMAS).setId(getId()).setSalFacade(facade).build();
    // Monitoring supported
    final NetconfSessionPreferences sessionCaps = getSessionCaps(true, Lists.newArrayList(TEST_CAPABILITY, TEST_CAPABILITY2));
    device.onRemoteSessionUp(sessionCaps, listener);
    verify(facade, timeout(5000)).onDeviceConnected(any(MountPointContext.class), any(NetconfSessionPreferences.class), any(NetconfDeviceRpc.class), isNull());
    verify(schemaFactory, times(1)).createEffectiveModelContext(anyCollection());
}

Example 2 with NetconfDeviceSchemasResolver

use of org.opendaylight.netconf.sal.connect.api.NetconfDeviceSchemasResolver in project netconf by opendaylight.

the class NetconfDeviceTest method testNetconfDeviceFlawedModelFailedResolution.

@Test
public void testNetconfDeviceFlawedModelFailedResolution() throws Exception {
    final RemoteDeviceHandler<NetconfSessionPreferences> facade = getFacade();
    final NetconfDeviceCommunicator listener = getListener();
    final EffectiveModelContextFactory schemaFactory = getSchemaFactory();
    final SchemaRepository schemaRepository = getSchemaRepository();
    final SchemaResolutionException schemaResolutionException = new SchemaResolutionException("fail first", TEST_SID, new Throwable("YangTools parser fail"));
    doAnswer(invocation -> {
        if (((Collection<?>) invocation.getArguments()[0]).size() == 2) {
            return Futures.immediateFailedFuture(schemaResolutionException);
        } else {
            return Futures.immediateFuture(SCHEMA_CONTEXT);
        }
    }).when(schemaFactory).createEffectiveModelContext(anyCollection());
    final NetconfDeviceSchemasResolver stateSchemasResolver = (deviceRpc, remoteSessionCapabilities, id, schemaContext) -> {
        final Module first = Iterables.getFirst(SCHEMA_CONTEXT.getModules(), null);
        final QName qName = QName.create(first.getQNameModule(), first.getName());
        final NetconfStateSchemas.RemoteYangSchema source1 = new NetconfStateSchemas.RemoteYangSchema(qName);
        final NetconfStateSchemas.RemoteYangSchema source2 = new NetconfStateSchemas.RemoteYangSchema(QName.create(first.getQNameModule(), "test-module2"));
        return new NetconfStateSchemas(Sets.newHashSet(source1, source2));
    };
    final NetconfDevice.SchemaResourcesDTO schemaResourcesDTO = new NetconfDevice.SchemaResourcesDTO(getSchemaRegistry(), schemaRepository, schemaFactory, stateSchemasResolver);
    final NetconfDevice device = new NetconfDeviceBuilder().setReconnectOnSchemasChange(true).setSchemaResourcesDTO(schemaResourcesDTO).setGlobalProcessingExecutor(getExecutor()).setId(getId()).setSalFacade(facade).setBaseSchemas(BASE_SCHEMAS).build();
    // Monitoring supported
    final NetconfSessionPreferences sessionCaps = getSessionCaps(true, Lists.newArrayList(TEST_CAPABILITY, TEST_CAPABILITY2));
    device.onRemoteSessionUp(sessionCaps, listener);
    verify(facade, timeout(5000)).onDeviceConnected(any(MountPointContext.class), any(NetconfSessionPreferences.class), any(NetconfDeviceRpc.class), isNull());
    verify(schemaFactory, times(2)).createEffectiveModelContext(anyCollection());
}