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.mdsal.binding.generator.util;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.junit.Assert.assertNull;
13 import static org.mockito.ArgumentMatchers.any;
14 import static org.mockito.Mockito.doReturn;
15 import static org.mockito.Mockito.mock;
16 import static org.mockito.Mockito.withSettings;
17 import static org.opendaylight.mdsal.binding.yang.types.TypeProviderModel.createTestContext;
19 import com.google.common.collect.ImmutableList;
20 import com.google.common.collect.ImmutableSet;
21 import java.lang.reflect.Constructor;
22 import java.lang.reflect.InvocationTargetException;
24 import java.util.ArrayList;
25 import java.util.List;
26 import java.util.Optional;
27 import org.junit.Before;
28 import org.junit.Test;
29 import org.opendaylight.yangtools.yang.common.QName;
30 import org.opendaylight.yangtools.yang.common.QNameModule;
31 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
32 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
33 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
34 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
35 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
36 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
37 import org.opendaylight.yangtools.yang.model.api.Module;
38 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
39 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
40 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
41 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
43 public class YangSchemaUtilsTest {
45 private static final AugmentationSchemaNode AUGMENTATION_SCHEMA = mock(AugmentationSchemaNode.class);
46 private static final UnknownSchemaNode UNKNOWN_SCHEMA_NODE = mock(UnknownSchemaNode.class);
47 private static final QName Q_NAME = QName.create(URI.create("testUri"), "foo_augment");
50 public void setUp() throws Exception {
51 doReturn(ImmutableList.of(UNKNOWN_SCHEMA_NODE)).when(AUGMENTATION_SCHEMA).getUnknownSchemaNodes();
52 doReturn(QName.create("", YangSchemaUtils.AUGMENT_IDENTIFIER)).when(UNKNOWN_SCHEMA_NODE).getNodeType();
53 doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getQName();
57 public void getAugmentationQName() throws Exception {
58 assertEquals(Q_NAME, YangSchemaUtils.getAugmentationQName(AUGMENTATION_SCHEMA));
59 final DataSchemaNode dataSchemaNode = mock(DataSchemaNode.class);
60 doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getNodeType();
61 doReturn(ImmutableList.of(dataSchemaNode)).when(AUGMENTATION_SCHEMA).getChildNodes();
62 doReturn(Boolean.FALSE).when(dataSchemaNode).isAugmenting();
63 doReturn(Q_NAME).when(dataSchemaNode).getQName();
64 doReturn(ImmutableList.of(dataSchemaNode)).when(AUGMENTATION_SCHEMA).getChildNodes();
65 assertEquals(Q_NAME, YangSchemaUtils.getAugmentationQName(AUGMENTATION_SCHEMA));
69 public void getAugmentationIdentifier() throws Exception {
70 assertEquals(Q_NAME, YangSchemaUtils.getAugmentationIdentifier(AUGMENTATION_SCHEMA));
71 doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getNodeType();
72 assertNull(YangSchemaUtils.getAugmentationIdentifier(AUGMENTATION_SCHEMA));
76 public void findTypeDefinition() throws Exception {
77 SchemaContext context = createTestContext();
78 assertNotNull(context);
79 final QName qName = QName.create(context.getModules().iterator().next().getNamespace(),
80 context.getModules().iterator().next().getRevision(), context.getModules().iterator().next().getName());
81 assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(ImmutableList.of(qName), false)));
82 final List<QName> qNames = new ArrayList<>();
83 context.getTypeDefinitions().forEach(typeDefinition -> qNames.add(typeDefinition.getQName()));
84 assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
86 context = mock(SchemaContext.class);
87 final Module container = mock(Module.class);
88 doReturn(Optional.empty()).when(context).findModule(any(QNameModule.class));
89 assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
90 doReturn(Optional.of(container)).when(context).findModule(any(QNameModule.class));
92 final DataSchemaNode node = mock(DataSchemaNode.class);
93 doReturn(node).when(container).getDataChildByName((QName) any());
94 final TypeDefinition<?> typeDefinition = mock(TypeDefinition.class);
95 doReturn(Q_NAME).when(typeDefinition).getQName();
96 doReturn(ImmutableSet.of(typeDefinition)).when(container).getTypeDefinitions();
97 assertEquals(typeDefinition,
98 YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(ImmutableList.of(Q_NAME), false)));
100 final GroupingDefinition grouping = mock(GroupingDefinition.class);
101 doReturn(Q_NAME).when(grouping).getQName();
102 doReturn(ImmutableSet.of(grouping)).when(container).getGroupings();
103 doReturn(ImmutableSet.of(typeDefinition)).when(grouping).getTypeDefinitions();
104 assertEquals(typeDefinition,
105 YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(ImmutableList.of(Q_NAME, Q_NAME), false)));
107 final DataNodeContainer dataNode =
108 mock(DataNodeContainer.class, withSettings().extraInterfaces(DataSchemaNode.class));
109 doReturn(dataNode).when(container).getDataChildByName((QName) any());
110 doReturn(ImmutableSet.of(typeDefinition)).when(dataNode).getTypeDefinitions();
111 assertEquals(typeDefinition,
112 YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(ImmutableList.of(Q_NAME, Q_NAME), false)));
114 final ChoiceSchemaNode choiceNode =
115 mock(ChoiceSchemaNode.class, withSettings().extraInterfaces(DataSchemaNode.class));
116 doReturn(choiceNode).when(container).getDataChildByName((QName) any());
117 final CaseSchemaNode caseNode = mock(CaseSchemaNode.class);
118 doReturn(caseNode).when(choiceNode).getCaseNodeByName((QName) any());
119 doReturn(ImmutableSet.of(typeDefinition)).when(caseNode).getTypeDefinitions();
120 assertEquals(typeDefinition,
121 YangSchemaUtils.findTypeDefinition(context,
122 SchemaPath.create(ImmutableList.of(Q_NAME, Q_NAME, Q_NAME), false)));
125 @Test(expected = UnsupportedOperationException.class)
126 public void privateConstructTest() throws Throwable {
127 final Constructor<YangSchemaUtils> constructor = YangSchemaUtils.class.getDeclaredConstructor();
128 constructor.setAccessible(true);
130 constructor.newInstance();
131 } catch (InvocationTargetException e) {