2e1b4c7eacc2345be9929a9a16155b7c4f79a665
[mdsal.git] / binding / mdsal-binding-generator / src / test / java / org / opendaylight / mdsal / binding / generator / impl / UnionTypeDefTest.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.impl;
9
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12
13 import java.util.List;
14 import org.junit.Test;
15 import org.opendaylight.mdsal.binding.model.api.GeneratedProperty;
16 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
17 import org.opendaylight.mdsal.binding.model.ri.Types;
18 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
19
20 public class UnionTypeDefTest {
21     @Test
22     public void unionTypeResolvingTest() {
23         final List<GeneratedType> genTypes = DefaultBindingGenerator.generateFor(YangParserTestUtils.parseYangResources(
24             UnionTypeDefTest.class, "/union-test-models/abstract-topology.yang", "/ietf-models/ietf-inet-types.yang"));
25
26         assertNotNull("genTypes is null", genTypes);
27         assertEquals(34, genTypes.size());
28
29         // TODO: implement test
30     }
31
32     @Test
33     public void unionTypedefLeafrefTest() {
34         final List<GeneratedType> generateTypes = DefaultBindingGenerator.generateFor(
35             YangParserTestUtils.parseYangResource("/bug8449.yang"));
36         assertNotNull(generateTypes);
37         assertEquals(6, generateTypes.size());
38
39         final GeneratedType cont = generateTypes.stream()
40             .filter(type -> type.getName().equals("Cont"))
41             .findFirst()
42             .orElseThrow();
43
44         final List<GeneratedType> enclosedTypes = cont.getEnclosedTypes();
45         assertEquals(1, enclosedTypes.size());
46
47         final GeneratedType refType = enclosedTypes.get(0);
48         final List<GeneratedProperty> properties = refType.getProperties();
49         assertEquals(1, properties.size());
50
51         final GeneratedProperty property = properties.get(0);
52         assertEquals("stringRefValue", property.getName());
53         assertEquals(Types.STRING, property.getReturnType());
54     }
55 }