/*
 * 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.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.junit.Test;
import org.opendaylight.mdsal.binding.model.api.GeneratedType;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;

public class GenTypesSubSetTest {

    @Test
    public void genTypesFromSubsetOfTwoModulesTest() {
        final EffectiveModelContext context = YangParserTestUtils.parseYangResources(GenTypesSubSetTest.class,
                "/leafref-test-models/abstract-topology@2013-02-08.yang", "/ietf-models/ietf-interfaces.yang",
                "/ietf-models/ietf-inet-types.yang", "/ietf-models/ietf-yang-types.yang");
        final Set<Module> toGenModules = new HashSet<>();
        for (final Module module : context.getModules()) {
            if (module.getName().equals("abstract-topology") || module.getName().equals("ietf-interfaces")) {
                toGenModules.add(module);
            }
        }

        assertEquals("Set of to Generate Modules must contain 2 modules", 2, toGenModules.size());
        final List<GeneratedType> genTypes = DefaultBindingGenerator.generateFor(context, toGenModules);
        assertNotNull("genTypes is null", genTypes);
        assertEquals(25, genTypes.size());
    }

    @Test
    public void genTypesFromSubsetOfThreeModulesTest() {
        final EffectiveModelContext context = YangParserTestUtils.parseYangResources(GenTypesSubSetTest.class,
                "/leafref-test-models/abstract-topology@2013-02-08.yang", "/ietf-models/ietf-interfaces.yang",
                "/ietf-models/ietf-inet-types.yang", "/ietf-models/ietf-yang-types.yang",
                "/ietf-models/iana-if-type.yang");
        assertNotNull(context);

        final Set<Module> toGenModules = new HashSet<>();
        for (final Module module : context.getModules()) {
            if (module.getName().equals("abstract-topology") || module.getName().equals("ietf-interfaces")
                || module.getName().equals("iana-if-type")) {
                toGenModules.add(module);
            }
        }
        assertEquals("Set of to Generate Modules must contain 3 modules", 3, toGenModules.size());

        final List<GeneratedType> genTypes = DefaultBindingGenerator.generateFor(context, toGenModules);
        assertNotNull("genTypes is null", genTypes);
        assertEquals(27, genTypes.size());
    }
}