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 @Deprecated(forRemoval = true)
  46 public class YangSchemaUtilsTest {
  47
  48     private static final AugmentationSchemaNode AUGMENTATION_SCHEMA = mock(AugmentationSchemaNode.class);
  49     private static final UnknownSchemaNode UNKNOWN_SCHEMA_NODE = mock(UnknownSchemaNode.class);
  50     private static final QName Q_NAME = QName.create(URI.create("testUri"), "foo_augment");
  51
  52     @Before
  53     public void setUp() throws Exception {
  54         doReturn(ImmutableList.of(UNKNOWN_SCHEMA_NODE)).when(AUGMENTATION_SCHEMA).getUnknownSchemaNodes();
  55         doReturn(QName.create("", YangSchemaUtils.AUGMENT_IDENTIFIER)).when(UNKNOWN_SCHEMA_NODE).getNodeType();
  56         doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getQName();
  57     }
  58
  59     @Test
  60     public void getAugmentationQName() throws Exception {
  61         assertEquals(Q_NAME, YangSchemaUtils.getAugmentationQName(AUGMENTATION_SCHEMA));
  62         final DataSchemaNode dataSchemaNode = mock(DataSchemaNode.class);
  63         doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getNodeType();
  64         doReturn(ImmutableList.of(dataSchemaNode)).when(AUGMENTATION_SCHEMA).getChildNodes();
  65         doReturn(Boolean.FALSE).when(dataSchemaNode).isAugmenting();
  66         doReturn(Q_NAME).when(dataSchemaNode).getQName();
  67         doReturn(ImmutableList.of(dataSchemaNode)).when(AUGMENTATION_SCHEMA).getChildNodes();
  68         assertEquals(Q_NAME, YangSchemaUtils.getAugmentationQName(AUGMENTATION_SCHEMA));
  69     }
  70
  71     @Test
  72     public void getAugmentationIdentifier() throws Exception {
  73         assertEquals(Q_NAME, YangSchemaUtils.getAugmentationIdentifier(AUGMENTATION_SCHEMA));
  74         doReturn(Q_NAME).when(UNKNOWN_SCHEMA_NODE).getNodeType();
  75         assertNull(YangSchemaUtils.getAugmentationIdentifier(AUGMENTATION_SCHEMA));
  76     }
  77
  78     @Test
  79     public void findTypeDefinition() throws Exception {
  80         SchemaContext context = createTestContext();
  81         assertNotNull(context);
  82         final QName qName = QName.create(context.getModules().iterator().next().getNamespace(),
  83                 context.getModules().iterator().next().getRevision(), context.getModules().iterator().next().getName());
  84         assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(ImmutableList.of(qName), false)));
  85         final List<QName> qNames = new ArrayList<>();
  86         context.getTypeDefinitions().forEach(typeDefinition -> qNames.add(typeDefinition.getQName()));
  87         assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
  88
  89         context = mock(SchemaContext.class);
  90         final Module container = mock(Module.class);
  91         doReturn(Optional.empty()).when(context).findModule(any(QNameModule.class));
  92         assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
  93         doReturn(Optional.of(container)).when(context).findModule(any(QNameModule.class));
  94
  95         final DataSchemaNode node = mock(DataSchemaNode.class);
  96         doReturn(node).when(container).getDataChildByName(any());
  97         final TypeDefinition<?> typeDefinition = mock(TypeDefinition.class);
  98         doReturn(Q_NAME).when(typeDefinition).getQName();
  99         doReturn(ImmutableSet.of(typeDefinition)).when(container).getTypeDefinitions();
 100         assertEquals(typeDefinition, YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(false, Q_NAME)));
 101
 102         final GroupingDefinition grouping = mock(GroupingDefinition.class);
 103         doReturn(Q_NAME).when(grouping).getQName();
 104         doReturn(ImmutableSet.of(grouping)).when(container).getGroupings();
 105         doReturn(ImmutableSet.of(typeDefinition)).when(grouping).getTypeDefinitions();
 106         assertEquals(typeDefinition, YangSchemaUtils.findTypeDefinition(context,
 107             SchemaPath.create(false, Q_NAME, Q_NAME)));
 108
 109         final DataNodeContainer dataNode =
 110                 mock(DataNodeContainer.class, withSettings().extraInterfaces(DataSchemaNode.class));
 111         doReturn(dataNode).when(container).getDataChildByName(any());
 112         doReturn(ImmutableSet.of(typeDefinition)).when(dataNode).getTypeDefinitions();
 113         assertEquals(typeDefinition, YangSchemaUtils.findTypeDefinition(context,
 114             SchemaPath.create(false, Q_NAME, Q_NAME)));
 115
 116         final ChoiceSchemaNode choiceNode =
 117                 mock(ChoiceSchemaNode.class, withSettings().extraInterfaces(DataSchemaNode.class));
 118         doReturn(choiceNode).when(container).getDataChildByName(any());
 119         final CaseSchemaNode caseNode = mock(CaseSchemaNode.class);
 120         doReturn(caseNode).when(choiceNode).getCaseNodeByName(any());
 121         doReturn(ImmutableSet.of(typeDefinition)).when(caseNode).getTypeDefinitions();
 122         assertEquals(typeDefinition, YangSchemaUtils.findTypeDefinition(context,
 123             SchemaPath.create(false, Q_NAME, Q_NAME, Q_NAME)));
 124     }
 125
 126     @Test
 127     public void privateConstructTest() throws ReflectiveOperationException {
 128         final Constructor<YangSchemaUtils> constructor = YangSchemaUtils.class.getDeclaredConstructor();
 129         constructor.setAccessible(true);
 130         try {
 131             constructor.newInstance();
 132             fail();
 133         } catch (InvocationTargetException e) {
 134             assertTrue(e.getCause() instanceof UnsupportedOperationException);
 135         }
 136     }
 137 }