/*
 * 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.mdsal.binding.generator.impl;

import static org.junit.Assert.assertTrue;

import java.util.List;
import org.junit.Test;
import org.opendaylight.mdsal.binding.generator.api.BindingGenerator;
import org.opendaylight.mdsal.binding.model.api.GeneratedProperty;
import org.opendaylight.mdsal.binding.model.api.GeneratedTransferObject;
import org.opendaylight.mdsal.binding.model.api.GeneratedType;
import org.opendaylight.mdsal.binding.model.api.Type;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;

public class GenerateInnerClassForBitsAndUnionInLeavesTest {

    @Test
    public void testInnerClassCreationForBitsAndUnionsInLeafes() {
        final SchemaContext context = YangParserTestUtils.parseYangResource("/bit_and_union_in_leaf.yang");
        assertTrue(context != null);

        final BindingGenerator bindingGen = new BindingGeneratorImpl();
        final List<Type> genTypes = bindingGen.generateTypes(context);
        assertTrue(genTypes != null);

        boolean parentContainerFound = false;
        boolean bitLeafTOFound = false;
        boolean unionLeafTOFound = false;

        boolean firstBitPropertyFound = false;
        boolean secondBitPropertyFound = false;
        boolean thirdBitPropertyFound = false;

        boolean firstBitPropertyTypeOK = false;
        boolean secondBitPropertyTypeOK = false;
        boolean thirdBitPropertyTypeOK = false;

        boolean int32UnionPropertyFound = false;
        boolean int32UnionPropertyTypeOK = false;
        boolean stringUnionPropertyFound = false;
        boolean stringUnionPropertyTypeOK = false;
        boolean uint8UnionPropertyFound = false;
        boolean uint8UnionPropertyTypeOK = false;

        for (Type type : genTypes) {
            if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
                if (type.getName().equals("ParentContainer")) {
                    parentContainerFound = true;
                    GeneratedType parentContainer = (GeneratedType) type;
                    List<GeneratedType> enclosedTypes = parentContainer.getEnclosedTypes();
                    for (GeneratedType genType : enclosedTypes) {
                        if (genType instanceof GeneratedTransferObject) {
                            if (genType.getName().equals("BitLeaf")) {
                                bitLeafTOFound = true;
                                GeneratedTransferObject bitLeafTO = (GeneratedTransferObject) genType;

                                List<GeneratedProperty> bitLeafProperties = bitLeafTO.getProperties();
                                for (GeneratedProperty bitLeafProperty : bitLeafProperties) {
                                    String bitLeafPropertyType = bitLeafProperty.getReturnType().getName();
                                    if (bitLeafProperty.getName().equals("firstBit")) {
                                        firstBitPropertyFound = true;
                                        if (bitLeafPropertyType.equals("Boolean")) {
                                            firstBitPropertyTypeOK = true;
                                        }
                                    } else if (bitLeafProperty.getName().equals("secondBit")) {
                                        secondBitPropertyFound = true;
                                        if (bitLeafPropertyType.equals("Boolean")) {
                                            secondBitPropertyTypeOK = true;
                                        }
                                    } else if (bitLeafProperty.getName().equals("thirdBit")) {
                                        thirdBitPropertyFound = true;
                                        if (bitLeafPropertyType.equals("Boolean")) {
                                            thirdBitPropertyTypeOK = true;
                                        }
                                    }

                                }

                            } else if (genType.getName().equals("UnionLeaf")) {
                                unionLeafTOFound = true;
                                GeneratedTransferObject unionLeafTO = (GeneratedTransferObject) genType;

                                List<GeneratedProperty> unionLeafProperties = unionLeafTO.getProperties();
                                for (GeneratedProperty unionLeafProperty : unionLeafProperties) {
                                    String unionLeafPropertyType = unionLeafProperty.getReturnType().getName();
                                    if (unionLeafProperty.getName().equals("int32")) {
                                        int32UnionPropertyFound = true;
                                        if (unionLeafPropertyType.equals("Integer")) {
                                            int32UnionPropertyTypeOK = true;
                                        }
                                    } else if (unionLeafProperty.getName().equals("string")) {
                                        stringUnionPropertyFound = true;
                                        if (unionLeafPropertyType.equals("String")) {
                                            stringUnionPropertyTypeOK = true;
                                        }
                                    } else if (unionLeafProperty.getName().equals("uint8")) {
                                        uint8UnionPropertyFound = true;
                                        if (unionLeafPropertyType.equals("Short")) {
                                            uint8UnionPropertyTypeOK = true;
                                        }
                                    }

                                }

                            }
                        }
                    }
                }
            }
        }
        assertTrue(parentContainerFound);

        assertTrue(bitLeafTOFound);
        assertTrue(firstBitPropertyFound);
        assertTrue(secondBitPropertyFound);
        assertTrue(thirdBitPropertyFound);

        assertTrue(firstBitPropertyTypeOK);
        assertTrue(secondBitPropertyTypeOK);
        assertTrue(thirdBitPropertyTypeOK);

        assertTrue(unionLeafTOFound);
        assertTrue(int32UnionPropertyFound);
        assertTrue(int32UnionPropertyTypeOK);
        assertTrue(stringUnionPropertyFound);
        assertTrue(stringUnionPropertyTypeOK);
        assertTrue(uint8UnionPropertyFound);
        assertTrue(uint8UnionPropertyTypeOK);

    }
}