+++ /dev/null
-/*
- * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
- *
- * This program and the accompanying materials are made available under the
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,
- * and is available at http://www.eclipse.org/legal/epl-v10.html
- */
-package org.opendaylight.yangtools.yang.parser.repo;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertNull;
-import static org.junit.Assert.assertTrue;
-
-import com.google.common.collect.ImmutableSet;
-import com.google.common.util.concurrent.ListenableFuture;
-import java.util.Set;
-import org.junit.Test;
-import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
-import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
-import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.repo.api.EffectiveModelContextFactory;
-import org.opendaylight.yangtools.yang.model.repo.api.SchemaContextFactoryConfiguration;
-import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
-import org.opendaylight.yangtools.yang.parser.rfc7950.ir.IRSchemaSource;
-import org.opendaylight.yangtools.yang.parser.rfc7950.repo.TextToIRTransformer;
-
-public class SharedSchemaRepositoryWithFeaturesTest {
-
- @Test
- public void testSharedSchemaRepositoryWithSomeFeaturesSupported() throws Exception {
- final Set<QName> supportedFeatures = ImmutableSet.of(QName.create("foobar-namespace", "test-feature-1"));
-
- final SharedSchemaRepository sharedSchemaRepository = new SharedSchemaRepository(
- "shared-schema-repo-with-features-test");
-
- final SettableSchemaProvider<IRSchemaSource> foobar = getImmediateYangSourceProviderFromResource(
- "/if-feature-resolution-test/shared-schema-repository/foobar.yang");
- foobar.register(sharedSchemaRepository);
- foobar.setResult();
-
- final ListenableFuture<EffectiveModelContext> testSchemaContextFuture =
- sharedSchemaRepository.createEffectiveModelContextFactory(
- SchemaContextFactoryConfiguration.builder().setSupportedFeatures(supportedFeatures).build())
- .createEffectiveModelContext(foobar.getId());
- assertTrue(testSchemaContextFuture.isDone());
- assertSchemaContext(testSchemaContextFuture.get(), 1);
-
- final Module module = testSchemaContextFuture.get().findModules("foobar").iterator().next();
- assertNotNull(module);
- assertEquals(2, module.getChildNodes().size());
-
- final ContainerSchemaNode testContainerA = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-a"));
- final LeafSchemaNode testLeafA = (LeafSchemaNode) testContainerA.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-a"));
-
- final ContainerSchemaNode testContainerB = (ContainerSchemaNode) module.dataChildByName(
- QName.create(module.getQNameModule(), "test-container-b"));
- assertNull(testContainerB);
-
- final ContainerSchemaNode testContainerC = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-c"));
- final LeafSchemaNode testLeafC = (LeafSchemaNode) testContainerC.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-c"));
- }
-
- @Test
- public void testSharedSchemaRepositoryWithAllFeaturesSupported() throws Exception {
- final SharedSchemaRepository sharedSchemaRepository = new SharedSchemaRepository(
- "shared-schema-repo-with-features-test");
-
- final SettableSchemaProvider<IRSchemaSource> foobar = getImmediateYangSourceProviderFromResource(
- "/if-feature-resolution-test/shared-schema-repository/foobar.yang");
- foobar.register(sharedSchemaRepository);
- foobar.setResult();
-
- final EffectiveModelContextFactory fact = sharedSchemaRepository.createEffectiveModelContextFactory();
- final ListenableFuture<EffectiveModelContext> testSchemaContextFuture =
- fact.createEffectiveModelContext(foobar.getId());
- assertTrue(testSchemaContextFuture.isDone());
- assertSchemaContext(testSchemaContextFuture.get(), 1);
-
- final Module module = testSchemaContextFuture.get().findModules("foobar").iterator().next();
- assertNotNull(module);
- assertEquals(3, module.getChildNodes().size());
-
- final ContainerSchemaNode testContainerA = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-a"));
- final LeafSchemaNode testLeafA = (LeafSchemaNode) testContainerA.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-a"));
-
- final ContainerSchemaNode testContainerB = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-b"));
- final LeafSchemaNode testLeafB = (LeafSchemaNode) testContainerB.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-b"));
-
- final ContainerSchemaNode testContainerC = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-c"));
- final LeafSchemaNode testLeafC = (LeafSchemaNode) testContainerC.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-c"));
- }
-
- @Test
- public void testSharedSchemaRepositoryWithNoFeaturesSupported() throws Exception {
- final SharedSchemaRepository sharedSchemaRepository = new SharedSchemaRepository(
- "shared-schema-repo-with-features-test");
-
- final SettableSchemaProvider<IRSchemaSource> foobar = getImmediateYangSourceProviderFromResource(
- "/if-feature-resolution-test/shared-schema-repository/foobar.yang");
- foobar.register(sharedSchemaRepository);
- foobar.setResult();
-
- final ListenableFuture<EffectiveModelContext> testSchemaContextFuture =
- sharedSchemaRepository.createEffectiveModelContextFactory(
- SchemaContextFactoryConfiguration.builder().setSupportedFeatures(ImmutableSet.of()).build())
- .createEffectiveModelContext(foobar.getId());
- assertTrue(testSchemaContextFuture.isDone());
- assertSchemaContext(testSchemaContextFuture.get(), 1);
-
- final Module module = testSchemaContextFuture.get().findModules("foobar").iterator().next();
- assertNotNull(module);
- assertEquals(1, module.getChildNodes().size());
-
- final ContainerSchemaNode testContainerC = (ContainerSchemaNode) module.getDataChildByName(
- QName.create(module.getQNameModule(), "test-container-c"));
- final LeafSchemaNode testLeafC = (LeafSchemaNode) testContainerC.getDataChildByName(
- QName.create(module.getQNameModule(), "test-leaf-c"));
- }
-
- private static SettableSchemaProvider<IRSchemaSource> getImmediateYangSourceProviderFromResource(
- final String resourceName) throws Exception {
- final YangTextSchemaSource yangSource = YangTextSchemaSource.forResource(resourceName);
- return SettableSchemaProvider.createImmediate(TextToIRTransformer.transformText(yangSource),
- IRSchemaSource.class);
- }
-
- private static void assertSchemaContext(final SchemaContext schemaContext, final int moduleSize) {
- assertNotNull(schemaContext);
- assertEquals(moduleSize, schemaContext.getModules().size());
- }
-}