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.mockito.Matchers.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;
  18
  19 import com.google.common.collect.ImmutableList;
  20 import com.google.common.collect.ImmutableSet;
  21 import java.lang.reflect.Constructor;
  22 import java.net.URI;
  23 import java.util.ArrayList;
  24 import java.util.Date;
  25 import java.util.List;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.opendaylight.mdsal.binding.generator.util.YangSchemaUtils;
  29 import org.opendaylight.yangtools.yang.common.QName;
  30 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
  31 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
  32 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
  33 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
  34 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  35 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
  36 import org.opendaylight.yangtools.yang.model.api.Module;
  37 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  38 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  39 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
  40 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
  41
  42 public class YangSchemaUtilsTest {
  43
  44     private static final AugmentationSchema AUGMENTATION_SCHEMA = mock(AugmentationSchema.class);
  45     private static final UnknownSchemaNode UNKNOWN_SCHEMA_NODE = mock(UnknownSchemaNode.class);
  46     private static final QName Q_NAME =
  47             QName.create(URI.create("testUri"), new Date(System.currentTimeMillis()), "foo_augment");
  48
  49     @Before
  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();
  54     }
  55
  56     @Test
  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(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));
  66     }
  67
  68     @Test
  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));
  73     }
  74
  75     @Test
  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 qNames = new ArrayList();
  83         context.getTypeDefinitions().forEach(typeDefinition -> qNames.add(typeDefinition.getQName()));
  84         assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
  85
  86         context = mock(SchemaContext.class);
  87         final Module container = mock(Module.class);
  88         doReturn(null).when(context).findModuleByNamespaceAndRevision(any(), any());
  89         assertNull(YangSchemaUtils.findTypeDefinition(context, SchemaPath.create(qNames, false)));
  90         doReturn(container).when(context).findModuleByNamespaceAndRevision(any(), any());
  91
  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)));
  99
 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)));
 106
 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)));
 113
 114         final ChoiceSchemaNode choiceNode =
 115                 mock(ChoiceSchemaNode.class, withSettings().extraInterfaces(DataSchemaNode.class));
 116         doReturn(choiceNode).when(container).getDataChildByName((QName) any());
 117         final ChoiceCaseNode caseNode = mock(ChoiceCaseNode.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)));
 123     }
 124
 125     @Test(expected = UnsupportedOperationException.class)
 126     public void privateConstructTest() throws Throwable {
 127         final Constructor constructor = YangSchemaUtils.class.getDeclaredConstructor();
 128         constructor.setAccessible(true);
 129         try {
 130             constructor.newInstance();
 131         } catch (Exception e) {
 132             throw e.getCause();
 133         }
 134     }
 135 }