/*
 * 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.assertTrue;

import java.util.Set;
import org.junit.Test;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.UsesNode;
import org.opendaylight.yangtools.yang.parser.spi.meta.SomeModifiersUnresolvedException;

public class AugmentToExtensionTest {
    private Set<Module> modules;

    @Test(expected = SomeModifiersUnresolvedException.class)
    public void testIncorrectPath() throws Exception {
        modules = TestUtils.loadModules(getClass().getResource("/augment-to-extension-test/incorrect-path").toURI());
    }

    /*
     * FIXME: Figure way to determine use case of tail-f:input without hacks
     *
     */
    @Test
    public void testCorrectPathIntoUnsupportedTarget() throws Exception {

        try {
        modules = TestUtils.loadModules(getClass().getResource(
                "/augment-to-extension-test/correct-path-into-unsupported-target").toURI());
        } catch (final Exception e) {
            StmtTestUtils.log(e, "    ");
            throw e;
        }

        final Module devicesModule = TestUtils.findModule(modules, "augment-module");

        final ContainerSchemaNode devicesContainer = (ContainerSchemaNode) devicesModule.getDataChildByName(QName
                .create(devicesModule.getQNameModule(), "my-container"));
        final Set<UsesNode> uses = devicesContainer.getUses();

        for (final UsesNode usesNode : uses) {
            assertTrue(usesNode.getAugmentations().isEmpty());
        }
    }


    @Test
    public void testCorrectAugment() throws Exception {
        modules = TestUtils.loadModules(getClass().getResource("/augment-to-extension-test/correct-augment").toURI());

        final Module devicesModule = TestUtils.findModule(modules, "augment-module");

        final ContainerSchemaNode devicesContainer = (ContainerSchemaNode) devicesModule.getDataChildByName(QName
                .create(devicesModule.getQNameModule(), "my-container"));
        final Set<UsesNode> uses = devicesContainer.getUses();

        boolean augmentationIsInContainer = false;
        for (final UsesNode usesNode : uses) {
            final Set<AugmentationSchema> augmentations = usesNode.getAugmentations();
            for (final AugmentationSchema augmentationSchema : augmentations) {
                augmentationIsInContainer = true;
            }
        }

        assertTrue(augmentationIsInContainer);
    }

}