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