BUG-4688: Rework SchemaContext module lookups
[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / EffectiveSchemaContextTest.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.assertFalse;
13 import static org.junit.Assert.assertNotNull;
14 import static org.junit.Assert.assertNull;
15 import static org.junit.Assert.assertTrue;
16
17 import java.io.IOException;
18 import java.net.URI;
19 import java.net.URISyntaxException;
20 import java.text.ParseException;
21 import java.util.Collection;
22 import java.util.List;
23 import java.util.Set;
24 import org.junit.Test;
25 import org.opendaylight.yangtools.yang.common.QName;
26 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
27 import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
28 import org.opendaylight.yangtools.yang.model.api.Module;
29 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
30 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
31 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
32 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
33 import org.opendaylight.yangtools.yang.model.api.Status;
34 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
35 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
36 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
37 import org.opendaylight.yangtools.yang.parser.rfc6020.repo.YangStatementStreamSource;
38 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
39 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
40 import org.opendaylight.yangtools.yang.parser.stmt.reactor.EffectiveSchemaContext;
41 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
42
43 public class EffectiveSchemaContextTest {
44
45     @Test
46     public void testEffectiveSchemaContext() throws ReactorException, ParseException, URISyntaxException, IOException,
47             YangSyntaxErrorException {
48         final CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR.newBuild();
49
50         final YangStatementStreamSource source1 = YangStatementStreamSource.create(YangTextSchemaSource.forResource(
51             "/effective-schema-context-test/foo.yang"));
52         final YangStatementStreamSource source2 = YangStatementStreamSource.create(YangTextSchemaSource.forResource(
53             "/effective-schema-context-test/bar.yang"));
54         final YangStatementStreamSource source3 = YangStatementStreamSource.create(YangTextSchemaSource.forResource(
55             "/effective-schema-context-test/baz.yang"));
56
57         reactor.addSources(source1, source2, source3);
58         final SchemaContext schemaContext = reactor.buildEffective();
59         assertNotNull(schemaContext);
60
61         final Set<DataSchemaNode> dataDefinitions = schemaContext.getDataDefinitions();
62         assertEquals(3, dataDefinitions.size());
63
64         final Collection<DataSchemaNode> childNodes = schemaContext.getChildNodes();
65         assertEquals(3, childNodes.size());
66
67         final Set<NotificationDefinition> notifications = schemaContext.getNotifications();
68         assertEquals(3, notifications.size());
69
70         final Set<RpcDefinition> rpcs = schemaContext.getOperations();
71         assertEquals(3, rpcs.size());
72
73         final Set<ExtensionDefinition> extensions = schemaContext.getExtensions();
74         assertEquals(3, extensions.size());
75
76         final List<UnknownSchemaNode> unknownSchemaNodes = schemaContext.getUnknownSchemaNodes();
77         assertEquals(3, unknownSchemaNodes.size());
78
79         assertNull(schemaContext.getDataChildByName(QName.create("foo-namespace", "2016-09-21", "foo-cont")));
80
81         assertFalse(schemaContext.findModule("foo", QName.parseRevision("2016-08-21")).isPresent());
82         assertFalse(schemaContext.findModule(URI.create("foo-namespace"), QName.parseRevision("2016-08-21"))
83             .isPresent());
84
85         assertFalse(schemaContext.isAugmenting());
86         assertFalse(schemaContext.isAddedByUses());
87         assertFalse(schemaContext.isConfiguration());
88         assertFalse(schemaContext.isPresenceContainer());
89         assertNull(schemaContext.getConstraints());
90         assertNull(schemaContext.getDescription());
91         assertNull(schemaContext.getReference());
92         assertEquals(SchemaContext.NAME, schemaContext.getQName());
93         assertEquals(SchemaPath.ROOT, schemaContext.getPath());
94         assertEquals(Status.CURRENT, schemaContext.getStatus());
95         assertNotNull(schemaContext.getUses());
96         assertTrue(schemaContext.getUses().isEmpty());
97         assertNotNull(schemaContext.getAvailableAugmentations());
98         assertTrue(schemaContext.getAvailableAugmentations().isEmpty());
99
100         Module fooModule = schemaContext.findModule("foo", QName.parseRevision("2016-09-21")).get();
101         assertEquals(3, schemaContext.getModules().size());
102         assertEquals(3, ((EffectiveSchemaContext) schemaContext).getRootDeclaredStatements().size());
103         assertEquals(3,((EffectiveSchemaContext) schemaContext).getRootEffectiveStatements().size());
104
105         final Set<Module> modules = schemaContext.getModules();
106         final SchemaContext copiedSchemaContext = EffectiveSchemaContext.resolveSchemaContext(modules);
107         assertNotNull(copiedSchemaContext);
108         assertEquals(modules, copiedSchemaContext.getModules());
109     }
110 }