Bug5437Test.java example

Explorer
yangtools-master
/*
 * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.yangtools.yang.stmt;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import org.junit.Test;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.model.api.SchemaNode;
import org.opendaylight.yangtools.yang.model.api.SchemaPath;
import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.BinaryTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.IntegerTypeDefinition;
import org.opendaylight.yangtools.yang.model.api.type.LeafrefTypeDefinition;
import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;

public class Bug5437Test {
    private static final String NS = "foo";
    private static final String REV = "2016-03-01";

    @Test
    public void test() throws Exception {
        SchemaContext context = StmtTestUtils.parseYangSources("/bugs/bug5437");
        assertNotNull(context);

        QName root = QName.create(NS, REV, "root");
        QName leafRef2 = QName.create(NS, REV, "leaf-ref-2");
        QName conGrp = QName.create(NS, REV, "con-grp");
        QName leafRef = QName.create(NS, REV, "leaf-ref");

        SchemaPath leafRefPath = SchemaPath.create(true, root, conGrp, leafRef);
        SchemaPath leafRef2Path = SchemaPath.create(true, root, leafRef2);
        SchemaNode findDataSchemaNode = SchemaContextUtil.findDataSchemaNode(context, leafRefPath);
        SchemaNode findDataSchemaNode2 = SchemaContextUtil.findDataSchemaNode(context, leafRef2Path);
        assertTrue(findDataSchemaNode instanceof LeafSchemaNode);
        assertTrue(findDataSchemaNode2 instanceof LeafSchemaNode);
        LeafSchemaNode leafRefNode = (LeafSchemaNode) findDataSchemaNode;
        LeafSchemaNode leafRefNode2 = (LeafSchemaNode) findDataSchemaNode2;

        assertTrue(leafRefNode.getType() instanceof LeafrefTypeDefinition);
        assertTrue(leafRefNode2.getType() instanceof LeafrefTypeDefinition);

        TypeDefinition<?> baseTypeForLeafRef = SchemaContextUtil.getBaseTypeForLeafRef(
                (LeafrefTypeDefinition) leafRefNode.getType(), context, leafRefNode);
        TypeDefinition<?> baseTypeForLeafRef2 = SchemaContextUtil.getBaseTypeForLeafRef(
                (LeafrefTypeDefinition) leafRefNode2.getType(), context, leafRefNode2);

        assertTrue(baseTypeForLeafRef instanceof BinaryTypeDefinition);
        assertTrue(baseTypeForLeafRef2 instanceof IntegerTypeDefinition);
    }
}