/**
 * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
 */
package org.opendaylight.yangtools.yang.parser.impl;

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

import java.io.IOException;
import java.net.URISyntaxException;
import java.util.Set;

import org.junit.Test;
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.util.YangParseException;

public class AugmentToExtensionTest {
    private Set<Module> modules;

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

    }

    @Test
    public void testCorrectPathIntoUnsupportedTarget() throws IOException, URISyntaxException {
        modules = TestUtils.loadModules(getClass().getResource(
                "/augment-to-extension-test/correct-path-into-unsupported-target").toURI());

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

        ContainerSchemaNode devicesContainer = (ContainerSchemaNode) devicesModul.getDataChildByName("my-container");
        Set<UsesNode> uses = devicesContainer.getUses();

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

        assertFalse(augmentationIsInContainer);
    }

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

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

        ContainerSchemaNode devicesContainer = (ContainerSchemaNode) devicesModul.getDataChildByName("my-container");
        Set<UsesNode> uses = devicesContainer.getUses();

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

        assertTrue(augmentationIsInContainer);
    }

}