2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.yangtools.yang.stmt;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertTrue;
13 import static org.opendaylight.yangtools.yang.stmt.StmtTestUtils.sourceForResource;
15 import java.util.Collection;
16 import org.junit.Test;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
19 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
20 import org.opendaylight.yangtools.yang.model.api.Module;
21 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
22 import org.opendaylight.yangtools.yang.model.api.stmt.UnrecognizedStatement;
23 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
26 public class ExtensionStmtTest {
28 public void testExtensionDefinition() throws ReactorException {
29 final SchemaContext result = RFC7950Reactors.defaultReactor().newBuild()
30 .addSource(sourceForResource("/model/bar.yang"))
32 assertNotNull(result);
34 final Module testModule = result.findModules("bar").iterator().next();
35 assertNotNull(testModule);
37 assertEquals(1, testModule.getExtensionSchemaNodes().size());
39 final Collection<? extends ExtensionDefinition> extensions = testModule.getExtensionSchemaNodes();
40 final ExtensionDefinition extension = extensions.iterator().next();
41 assertEquals("opendaylight", extension.getQName().getLocalName());
42 assertEquals("name", extension.getArgument());
43 assertTrue(extension.isYinElement());
47 public void testExtensionUsage() throws ReactorException {
48 final SchemaContext result = RFC7950Reactors.defaultReactor().newBuild()
49 .addSource(sourceForResource("/semantic-statement-parser/ext-typedef.yang"))
50 .addSource(sourceForResource("/semantic-statement-parser/ext-use.yang"))
52 assertNotNull(result);
54 final Module testModule1 = result.findModules("ext-typedef").iterator().next();
55 assertNotNull(testModule1);
57 assertEquals(1, testModule1.getExtensionSchemaNodes().size());
59 final Collection<? extends ExtensionDefinition> extensions = testModule1.getExtensionSchemaNodes();
60 final ExtensionDefinition extensionDefinition = extensions.iterator().next();
62 final Module testModule2 = result.findModules("ext-use").iterator().next();
63 assertNotNull(testModule2);
65 final LeafSchemaNode leaf = (LeafSchemaNode) testModule2.getDataChildByName(
66 QName.create(testModule2.getQNameModule(), "value"));
69 final Collection<? extends UnrecognizedStatement> unknownNodes = leaf.asEffectiveStatement().getDeclared()
70 .declaredSubstatements(UnrecognizedStatement.class);
71 assertEquals(1, unknownNodes.size());
72 final UnrecognizedStatement extensionUse = unknownNodes.iterator().next();
73 assertEquals(extensionDefinition.getQName(), extensionUse.statementDefinition().getStatementName());
74 assertEquals(extensionDefinition.getArgument(), extensionUse.statementDefinition().getArgumentDefinition()
75 .orElseThrow().getArgumentName().getLocalName());
77 assertEquals("key:value", extensionUse.argument());