BUG-4688: Rework SchemaContext module lookups
[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / EffectiveModuleTest.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8
9 package org.opendaylight.yangtools.yang.stmt;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNotNull;
13 import static org.opendaylight.yangtools.yang.stmt.StmtTestUtils.sourceForResource;
14
15 import java.net.URI;
16 import java.util.Date;
17 import java.util.List;
18 import java.util.Set;
19 import org.junit.Test;
20 import org.opendaylight.yangtools.yang.common.QName;
21 import org.opendaylight.yangtools.yang.common.QNameModule;
22 import org.opendaylight.yangtools.yang.common.SimpleDateFormatUtil;
23 import org.opendaylight.yangtools.yang.common.YangVersion;
24 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
25 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
26 import org.opendaylight.yangtools.yang.model.api.DeviateKind;
27 import org.opendaylight.yangtools.yang.model.api.Deviation;
28 import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
29 import org.opendaylight.yangtools.yang.model.api.FeatureDefinition;
30 import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;
31 import org.opendaylight.yangtools.yang.model.api.Module;
32 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
33 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
34 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
35 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
36 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
37 import org.opendaylight.yangtools.yang.model.api.Status;
38 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
39 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
40 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
41 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
42 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
43
44 public class EffectiveModuleTest {
45
46     private static final StatementStreamSource ROOT_MODULE = sourceForResource(
47             "/semantic-statement-parser/effective-module/root.yang");
48     private static final StatementStreamSource IMPORTED_MODULE = sourceForResource(
49             "/semantic-statement-parser/effective-module/imported.yang");
50     private static final StatementStreamSource SUBMODULE = sourceForResource(
51             "/semantic-statement-parser/effective-module/submod.yang");
52
53     private static final QNameModule ROOT_MODULE_QNAME = QNameModule.create(URI.create("root-ns"),
54             SimpleDateFormatUtil.DEFAULT_DATE_REV);
55
56     private static final QName cont = QName.create(ROOT_MODULE_QNAME, "cont");
57     private static final QName feature1 = QName.create(ROOT_MODULE_QNAME, "feature1");
58
59     private static final SchemaPath contSchemaPath = SchemaPath.create(true, cont);
60     private static final SchemaPath feature1SchemaPath = SchemaPath.create(true, feature1);
61
62     private static final Date REVISION = QName.parseRevision("2000-01-01");
63
64     @Test
65     public void effectiveBuildTest() throws SourceException, ReactorException {
66         CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR.newBuild();
67         reactor.addSources(ROOT_MODULE, IMPORTED_MODULE, SUBMODULE);
68         SchemaContext result = reactor.buildEffective();
69
70         assertNotNull(result);
71
72         Module rootModule = result.findModules("root").iterator().next();
73         assertNotNull(rootModule);
74
75         assertEquals("root-pref", rootModule.getPrefix());
76         assertEquals(YangVersion.VERSION_1, rootModule.getYangVersion());
77         assertEquals("cisco", rootModule.getOrganization());
78         assertEquals("cisco email", rootModule.getContact());
79
80         final ContainerSchemaNode contSchemaNode = (ContainerSchemaNode) rootModule.getDataChildByName(cont);
81         assertNotNull(contSchemaNode);
82
83         final Set<AugmentationSchema> augmentations = rootModule.getAugmentations();
84         assertEquals(1, augmentations.size());
85         assertEquals(contSchemaPath, augmentations.iterator().next().getTargetPath());
86
87         final Set<ModuleImport> imports = rootModule.getImports();
88         assertEquals(1, imports.size());
89         final ModuleImport importStmt = imports.iterator().next();
90         assertNotNull(importStmt);
91         assertEquals("imported", importStmt.getModuleName());
92         assertEquals(REVISION, importStmt.getRevision());
93         assertEquals("imp-pref", importStmt.getPrefix());
94
95         final Set<Module> submodules = rootModule.getSubmodules();
96         assertEquals(1, submodules.size());
97         assertEquals("submod", submodules.iterator().next().getName());
98
99         final Set<NotificationDefinition> notifications = rootModule.getNotifications();
100         assertEquals(1, notifications.size());
101         assertEquals("notif1", notifications.iterator().next().getQName().getLocalName());
102
103         final Set<RpcDefinition> rpcs = rootModule.getRpcs();
104         assertEquals(1, rpcs.size());
105         assertEquals("rpc1", rpcs.iterator().next().getQName().getLocalName());
106
107         final Set<Deviation> deviations = rootModule.getDeviations();
108         assertEquals(1, deviations.size());
109         final Deviation deviationStmt = deviations.iterator().next();
110         assertNotNull(deviationStmt);
111         final QNameModule importedModuleQName = QNameModule.create(URI.create("imported"), REVISION);
112         final QName importedContQName = QName.create(importedModuleQName, "cont");
113         final SchemaPath importedContSchemaPath = SchemaPath.create(true, importedContQName);
114         assertEquals(importedContSchemaPath, deviationStmt.getTargetPath());
115         assertEquals(DeviateKind.ADD, deviationStmt.getDeviates().iterator().next().getDeviateType());
116         assertEquals("deviate reference", deviationStmt.getReference());
117
118         final Set<IdentitySchemaNode> identities = rootModule.getIdentities();
119         assertEquals(1, identities.size());
120         assertEquals("identity1", identities.iterator().next().getQName().getLocalName());
121
122         final Set<FeatureDefinition> features = rootModule.getFeatures();
123         assertEquals(1, features.size());
124         final FeatureDefinition featureStmt = features.iterator().next();
125         assertNotNull(featureStmt);
126         assertEquals(feature1, featureStmt.getQName());
127         assertEquals(feature1SchemaPath, featureStmt.getPath());
128         assertEquals("feature1 description", featureStmt.getDescription());
129         assertEquals("feature1 reference", featureStmt.getReference());
130         assertEquals(Status.CURRENT, featureStmt.getStatus());
131
132         final List<ExtensionDefinition> extensionSchemaNodes = rootModule.getExtensionSchemaNodes();
133         assertEquals(1, extensionSchemaNodes.size());
134         assertEquals("ext1", extensionSchemaNodes.iterator().next().getQName().getLocalName());
135     }
136 }