2 * Copyright (c) 2015 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.model.util;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertTrue;
12 import static org.mockito.ArgumentMatchers.any;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
16 import com.google.common.collect.ImmutableSet;
18 import java.util.Arrays;
19 import java.util.Collection;
20 import java.util.HashSet;
21 import java.util.List;
22 import java.util.Optional;
24 import org.junit.Test;
25 import org.opendaylight.yangtools.yang.common.QName;
26 import org.opendaylight.yangtools.yang.common.QNameModule;
27 import org.opendaylight.yangtools.yang.common.Revision;
28 import org.opendaylight.yangtools.yang.common.UnresolvedQName.Unqualified;
29 import org.opendaylight.yangtools.yang.common.XMLNamespace;
30 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
31 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
32 import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
33 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
34 import org.opendaylight.yangtools.yang.model.api.Module;
35 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
36 import org.opendaylight.yangtools.yang.model.api.ModuleLike;
37 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
38 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
39 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
40 import org.opendaylight.yangtools.yang.model.api.Submodule;
41 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
42 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
43 import org.opendaylight.yangtools.yang.model.api.stmt.ImportEffectiveStatement;
44 import org.opendaylight.yangtools.yang.model.spi.AbstractSchemaContext;
45 import org.opendaylight.yangtools.yang.model.util.FilteringSchemaContextProxy.ModuleId;
47 public class SchemaContextProxyTest {
49 private static final URI NAMESPACE = URI.create("urn:opendaylight:params:xml:ns:yang:controller:config");
50 private static final Revision REVISION = Revision.of("2015-01-01");
51 private static final Revision REVISION2 = Revision.of("2015-01-15");
53 private static final String CONFIG_NAME = "config";
54 private static final String ROOT_NAME = "root";
55 private static final String MODULE2_NAME = "module2";
56 private static final String MODULE3_NAME = "module3";
57 private static final String MODULE4_NAME = "module4";
58 private static final String MODULE41_NAME = "module41";
59 private static final String MODULE5_NAME = "module5";
61 private static SchemaContext mockSchema(final Module... modules) {
62 final List<Module> sortedModules = Arrays.asList(modules);
63 sortedModules.sort(AbstractSchemaContext.NAME_REVISION_COMPARATOR);
64 SchemaContext mock = mock(SchemaContext.class);
65 doReturn(ImmutableSet.copyOf(sortedModules)).when(mock).getModules();
76 public void testBasic() {
77 Module moduleConfig = mockModule(CONFIG_NAME);
78 Module module2 = mockModule(MODULE2_NAME);
79 Module module3 = mockModule(MODULE3_NAME);
81 mockModuleImport(module2, moduleConfig);
82 mockModuleImport(module3, module2, moduleConfig);
84 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3);
86 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
88 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3);
92 * No root or additional modules
98 public void testNull() {
99 Module moduleConfig = mockModule(CONFIG_NAME);
100 Module module2 = mockModule(MODULE2_NAME);
101 Module module3 = mockModule(MODULE3_NAME);
103 mockModuleImport(module2, moduleConfig);
104 mockModuleImport(module3, module2, moduleConfig);
106 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3);
108 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null, null);
109 assertProxyContext(filteringSchemaContextProxy, null);
119 public void testConfigDifferentRevisions() {
120 Module moduleConfigNullRevision = mockModule(CONFIG_NAME, null);
121 Module moduleConfig = mockModule(CONFIG_NAME, REVISION);
122 Module moduleConfig2 = mockModule(CONFIG_NAME, REVISION2);
123 Module module2 = mockModule(MODULE2_NAME);
124 Module module3 = mockModule(MODULE3_NAME);
126 mockModuleImport(module2, moduleConfig);
127 mockModuleImport(module3, module2, moduleConfigNullRevision);
129 SchemaContext schemaContext = mockSchema(moduleConfig, moduleConfig2, module2, module3);
131 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
133 assertProxyContext(filteringSchemaContextProxy, moduleConfig, moduleConfig2, module2, module3);
143 public void testBasicNullRevision() throws Exception {
144 final Module moduleConfig = mockModule(CONFIG_NAME, Revision.of("2013-04-05"));
145 final Module module2 = mockModule(MODULE2_NAME, Revision.of("2014-06-17"));
146 final Module module20 = mockModule(MODULE2_NAME, null);
147 final Module module3 = mockModule(MODULE3_NAME, Revision.of("2014-06-12"));
148 final Module module30 = mockModule(MODULE3_NAME, null);
150 mockModuleImport(module20, moduleConfig);
151 mockModuleImport(module2, moduleConfig);
152 mockModuleImport(module3, module20, moduleConfig);
153 mockModuleImport(module30, module20, moduleConfig);
155 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3);
157 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
160 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3);
170 public void testBasicMoreRootModules() {
171 final Module moduleConfig = mockModule(CONFIG_NAME);
172 final Module moduleRoot = mockModule(ROOT_NAME);
173 final Module module2 = mockModule(MODULE2_NAME);
174 final Module module3 = mockModule(MODULE3_NAME);
176 mockModuleImport(module2, moduleConfig);
177 mockModuleImport(module3, moduleRoot);
179 SchemaContext schemaContext = mockSchema(moduleConfig, moduleRoot, module2, module3);
181 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null, moduleRoot,
183 assertProxyContext(filteringSchemaContextProxy, moduleRoot, module3, moduleConfig, module2);
193 public void testChainNotDepend() {
194 Module moduleConfig = mockModule(CONFIG_NAME);
195 Module module2 = mockModule(MODULE2_NAME);
196 Module module3 = mockModule(MODULE3_NAME);
198 mockModuleImport(module2, moduleConfig);
199 mockModuleImport(module3, module2);
201 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3);
203 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
205 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2);
212 * M2 -> M3 -> M4 -> M5
215 public void testChainDependMulti() {
216 Module moduleConfig = mockModule(CONFIG_NAME);
217 Module module2 = mockModule(MODULE2_NAME);
218 Module module3 = mockModule(MODULE3_NAME);
219 Module module4 = mockModule(MODULE4_NAME);
220 Module module5 = mockModule(MODULE5_NAME);
222 mockModuleImport(module2, moduleConfig, module3);
223 mockModuleImport(module3, module4);
224 mockModuleImport(module4, module5);
226 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4, module5);
228 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
230 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3, module4, module5);
240 public void testChainNotDependMulti() {
241 Module moduleConfig = mockModule(CONFIG_NAME);
242 Module module2 = mockModule(MODULE2_NAME);
243 Module module3 = mockModule(MODULE3_NAME);
244 Module module4 = mockModule(MODULE4_NAME);
246 mockModuleImport(module2, moduleConfig, module3);
247 mockModuleImport(module4, module3);
249 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4);
251 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
253 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3);
263 public void testChainNotMulti() {
264 final Module moduleConfig = mockModule(CONFIG_NAME);
265 final Module module2 = mockModule(MODULE2_NAME);
266 final Module module3 = mockModule(MODULE3_NAME);
267 final Module module4 = mockModule(MODULE4_NAME);
268 final Module module5 = mockModule(MODULE5_NAME);
270 mockModuleImport(module2, moduleConfig);
271 mockModuleImport(module3, moduleConfig);
272 mockModuleImport(module4, moduleConfig);
273 mockModuleImport(module5, moduleConfig);
275 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4, module5);
277 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
279 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3, module4, module5);
286 * M2 <- M3 M4=M3(Different revision)
289 public void testBasicRevisionChange() throws Exception {
290 Module moduleConfig = mockModule(CONFIG_NAME);
291 Module module2 = mockModule(MODULE2_NAME);
292 Module module3 = mockModule(MODULE3_NAME);
293 Module module4 = mockModule(MODULE3_NAME, Revision.of("2015-10-10"));
295 mockModuleImport(module2, moduleConfig);
296 mockModuleImport(module3, module2, moduleConfig);
298 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4);
300 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
302 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3);
308 * M2 -(no revision)-> M3(R2) ... M3(R1)
311 public void testImportNoRevision() {
312 Module moduleConfig = mockModule(CONFIG_NAME, REVISION);
313 Module module2 = mockModule(MODULE2_NAME, REVISION);
315 Module module3 = mockModule(MODULE3_NAME, null);
316 Module module30 = mockModule(MODULE3_NAME, REVISION);
317 Module module31 = mockModule(MODULE3_NAME, REVISION2);
318 mockModuleImport(module2, moduleConfig, module3);
320 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module30, module31);
322 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
325 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module31);
337 public void testBasicSubmodule() {
338 Module moduleConfig = mockModule(CONFIG_NAME);
339 Module module2 = mockModule(MODULE2_NAME);
340 Module module3 = mockModule(MODULE3_NAME);
341 Module module4 = mockModule(MODULE4_NAME);
342 Submodule module41 = mockSubmodule(MODULE41_NAME);
344 mockSubmodules(module4, module41);
345 mockModuleImport(module2, moduleConfig, module3);
346 mockModuleImport(module41, moduleConfig);
348 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4);
350 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, null,
352 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3, module4);
356 * M2 -> M3 -> M4 -> M5
359 public void testChainAdditionalModules() {
360 Module module2 = mockModule(MODULE2_NAME);
361 Module module3 = mockModule(MODULE3_NAME);
362 Module module4 = mockModule(MODULE4_NAME);
363 Module module5 = mockModule(MODULE5_NAME);
365 mockModuleImport(module2, module3);
366 mockModuleImport(module3, module4);
367 mockModuleImport(module4, module5);
369 SchemaContext schemaContext = mockSchema(module2, module3, module4, module5);
371 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(module2),
373 assertProxyContext(filteringSchemaContextProxy, module2, module3, module4, module5);
386 public void testChainAdditionalModulesConfig() {
387 Module moduleConfig = mockModule(CONFIG_NAME);
388 Module module2 = mockModule(MODULE2_NAME);
390 Module module3 = mockModule(MODULE3_NAME);
391 Module module4 = mockModule(MODULE4_NAME);
392 Module module5 = mockModule(MODULE5_NAME);
394 mockModuleImport(module2, moduleConfig);
395 mockModuleImport(module3, module4);
397 SchemaContext schemaContext = mockSchema(moduleConfig, module2, module3, module4, module5);
399 FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(module3),
401 assertProxyContext(filteringSchemaContextProxy, moduleConfig, module2, module3, module4);
405 public void testGetDataDefinitions() {
406 final Module moduleConfig = mockModule(CONFIG_NAME);
407 final SchemaContext schemaContext = mockSchema(moduleConfig);
408 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
411 final ContainerSchemaNode mockedContainer = mock(ContainerSchemaNode.class);
412 doReturn(Set.of(mockedContainer)).when(moduleConfig).getChildNodes();
414 final Collection<? extends DataSchemaNode> dataDefinitions = filteringSchemaContextProxy.getDataDefinitions();
415 assertTrue(dataDefinitions.contains(mockedContainer));
419 public void testGetNotifications() {
420 final Module moduleConfig = mockModule(CONFIG_NAME);
421 final SchemaContext schemaContext = mockSchema(moduleConfig);
422 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
425 final NotificationDefinition mockedNotification = mock(NotificationDefinition.class);
426 doReturn(Set.of(mockedNotification)).when(moduleConfig).getNotifications();
428 final Collection<? extends NotificationDefinition> schemaContextProxyNotifications =
429 filteringSchemaContextProxy.getNotifications();
430 assertTrue(schemaContextProxyNotifications.contains(mockedNotification));
434 public void testGetOperations() {
435 final Module moduleConfig = mockModule(CONFIG_NAME);
436 final SchemaContext schemaContext = mockSchema(moduleConfig);
437 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
440 final RpcDefinition mockedRpc = mock(RpcDefinition.class);
441 doReturn(Set.of(mockedRpc)).when(moduleConfig).getRpcs();
443 final Collection<? extends RpcDefinition> operations = filteringSchemaContextProxy.getOperations();
444 assertTrue(operations.contains(mockedRpc));
448 public void testGetExtensions() {
449 final Module moduleConfig = mockModule(CONFIG_NAME);
450 final SchemaContext schemaContext = mockSchema(moduleConfig);
451 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
454 final ExtensionDefinition mockedExtension = mock(ExtensionDefinition.class);
455 doReturn(List.of(mockedExtension)).when(moduleConfig).getExtensionSchemaNodes();
457 final Collection<? extends ExtensionDefinition> schemaContextProxyExtensions =
458 filteringSchemaContextProxy.getExtensions();
459 assertTrue(schemaContextProxyExtensions.contains(mockedExtension));
463 public void testGetUnknownSchemaNodes() {
464 final Module moduleConfig = mockModule(CONFIG_NAME);
465 final SchemaContext schemaContext = mockSchema(moduleConfig);
466 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
469 final UnknownSchemaNode mockedUnknownSchemaNode = mock(UnknownSchemaNode.class);
470 doReturn(List.of(mockedUnknownSchemaNode)).when(moduleConfig).getUnknownSchemaNodes();
472 final Collection<? extends UnknownSchemaNode> schemaContextProxyUnknownSchemaNodes =
473 filteringSchemaContextProxy.getUnknownSchemaNodes();
474 assertTrue(schemaContextProxyUnknownSchemaNodes.contains(mockedUnknownSchemaNode));
478 public void testGetTypeDefinitions() {
479 final Module moduleConfig = mockModule(CONFIG_NAME);
480 final SchemaContext schemaContext = mockSchema(moduleConfig);
481 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
484 final TypeDefinition<?> mockedTypeDefinition = mock(TypeDefinition.class);
485 doReturn(Set.of(mockedTypeDefinition)).when(moduleConfig).getTypeDefinitions();
487 final Collection<? extends TypeDefinition<?>> schemaContextProxyTypeDefinitions = filteringSchemaContextProxy
488 .getTypeDefinitions();
489 assertTrue(schemaContextProxyTypeDefinitions.contains(mockedTypeDefinition));
493 public void testGetChildNodes() {
494 final Module moduleConfig = mockModule(CONFIG_NAME);
495 final SchemaContext schemaContext = mockSchema(moduleConfig);
496 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
499 final ContainerSchemaNode mockedContainer = mock(ContainerSchemaNode.class);
500 doReturn(Set.of(mockedContainer)).when(moduleConfig).getChildNodes();
502 final Collection<? extends DataSchemaNode> schemaContextProxyChildNodes =
503 filteringSchemaContextProxy.getChildNodes();
504 assertTrue(schemaContextProxyChildNodes.contains(mockedContainer));
508 public void testGetGroupings() {
509 final Module moduleConfig = mockModule(CONFIG_NAME);
510 final SchemaContext schemaContext = mockSchema(moduleConfig);
511 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
514 final GroupingDefinition mockedGrouping = mock(GroupingDefinition.class);
515 doReturn(Set.of(mockedGrouping)).when(moduleConfig).getGroupings();
517 final Collection<? extends GroupingDefinition> schemaContextProxyGroupings =
518 filteringSchemaContextProxy.getGroupings();
519 assertTrue(schemaContextProxyGroupings.contains(mockedGrouping));
523 public void testGetDataChildByName() {
524 final Module moduleConfig = mockModule(CONFIG_NAME);
525 final SchemaContext schemaContext = mockSchema(moduleConfig);
526 final FilteringSchemaContextProxy filteringSchemaContextProxy = createProxySchemaCtx(schemaContext, Set.of(),
529 final QName qname = QName.create("config-namespace", "2016-08-11", "cont");
530 final ContainerSchemaNode mockedContainer = mock(ContainerSchemaNode.class);
531 doReturn(mockedContainer).when(moduleConfig).dataChildByName(any(QName.class));
533 final DataSchemaNode dataSchemaNode = filteringSchemaContextProxy.getDataChildByName(qname);
534 assertTrue(dataSchemaNode instanceof ContainerSchemaNode);
537 private static void assertProxyContext(final FilteringSchemaContextProxy filteringSchemaContextProxy,
538 final Module... expected) {
540 final Set<Module> modSet = expected != null ? ImmutableSet.copyOf(expected) : Set.of();
541 Set<Module> modSetFiltering = filteringSchemaContextProxy.getModules();
543 assertEquals(modSet, modSetFiltering);
545 //asserting collections
546 if (expected != null) {
547 for (final Module module : expected) {
548 assertEquals(module, filteringSchemaContextProxy.findModule(module.getName(), module.getRevision())
551 Collection<? extends Module> mod = filteringSchemaContextProxy.findModules(module.getNamespace());
552 assertTrue(mod.contains(module));
553 assertEquals(module, filteringSchemaContextProxy.findModule(module.getNamespace(),
554 module.getRevision().orElse(null)).get());
559 private static FilteringSchemaContextProxy createProxySchemaCtx(final SchemaContext schemaContext,
560 final Set<Module> additionalModules, final Module... modules) {
561 Set<Module> modulesSet = modules != null ? ImmutableSet.copyOf(modules) : Set.of();
562 return new FilteringSchemaContextProxy(schemaContext, createModuleIds(modulesSet),
563 createModuleIds(additionalModules));
566 private static Set<ModuleId> createModuleIds(final Set<Module> modules) {
567 Set<ModuleId> moduleIds = new HashSet<>();
568 if (modules != null) {
569 for (Module module : modules) {
570 moduleIds.add(new ModuleId(module.getName(), module.getRevision()));
577 private static void mockSubmodules(final Module mainModule, final Submodule... submodules) {
578 doReturn(Set.of(submodules)).when(mainModule).getSubmodules();
581 private static void mockModuleImport(final ModuleLike importer, final Module... imports) {
582 Set<ModuleImport> mockedImports = new HashSet<>();
583 for (final Module module : imports) {
584 mockedImports.add(new ModuleImport() {
586 public Unqualified getModuleName() {
587 return Unqualified.of(module.getName());
591 public Optional<Revision> getRevision() {
592 return module.getRevision();
596 public String getPrefix() {
597 return module.getName();
601 public Optional<String> getDescription() {
602 return module.getDescription();
606 public Optional<String> getReference() {
607 return module.getReference();
611 public String toString() {
612 return String.format("Module: %s, revision:%s", module.getName(), module.getRevision());
616 public ImportEffectiveStatement asEffectiveStatement() {
617 throw new UnsupportedOperationException();
621 doReturn(mockedImports).when(importer).getImports();
624 //mock module with revision
625 private static Module mockModule(final String name, final Revision rev) {
627 final Module mod = mockModule(name);
629 doReturn(QNameModule.create(mod.getNamespace(), rev)).when(mod).getQNameModule();
630 doReturn(Optional.ofNullable(rev)).when(mod).getRevision();
631 doReturn(mod.getQNameModule().toString()).when(mod).toString();
636 //mock module with default revision
637 private static Module mockModule(final String name) {
638 Module mockedModule = mock(Module.class);
639 mockModuleLike(mockedModule, name);
643 private static Submodule mockSubmodule(final String name) {
644 Submodule mockedModule = mock(Submodule.class);
645 mockModuleLike(mockedModule, name);
649 private static void mockModuleLike(final ModuleLike mockedModule, final String name) {
650 doReturn(name).when(mockedModule).getName();
651 doReturn(Optional.of(REVISION)).when(mockedModule).getRevision();
652 final XMLNamespace newNamespace = XMLNamespace.of(NAMESPACE.toString() + ":" + name);
653 doReturn(newNamespace).when(mockedModule).getNamespace();
654 doReturn(QNameModule.create(newNamespace, REVISION)).when(mockedModule).getQNameModule();
655 doReturn(Set.of()).when(mockedModule).getSubmodules();
656 doReturn(mockedModule.getQNameModule().toString()).when(mockedModule).toString();
657 mockModuleImport(mockedModule);