/*
 * Copyright 2016 Open Networking Laboratory
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.onosproject.yangutils.parser.impl.listeners;

import java.io.IOException;
import org.junit.Test;
import org.onosproject.yangutils.datamodel.YangContainer;
import org.onosproject.yangutils.datamodel.YangList;
import org.onosproject.yangutils.datamodel.YangModule;
import org.onosproject.yangutils.datamodel.YangNode;
import org.onosproject.yangutils.datamodel.YangNodeType;
import org.onosproject.yangutils.datamodel.YangStatusType;
import org.onosproject.yangutils.datamodel.YangUses;
import org.onosproject.yangutils.parser.exceptions.ParserException;
import org.onosproject.yangutils.parser.impl.YangUtilsParserManager;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.core.Is.is;

/**
 * Test cases for testing uses listener.
 */
public class UsesListenerTest {

    private final YangUtilsParserManager manager = new YangUtilsParserManager();

    /**
     * Checks uses statement inside module.
     */
    @Test
    public void processUsesInModule() throws IOException, ParserException {

        YangNode node = manager.getDataModel("src/test/resources/UsesInModule.yang");

        // Check whether the data model tree returned is of type module.
        assertThat((node instanceof YangModule), is(true));

        // Check whether the node type is set properly to module.
        assertThat(node.getNodeType(), is(YangNodeType.MODULE_NODE));

        // Check whether the module name is set correctly.
        YangModule yangNode = (YangModule) node;
        assertThat(yangNode.getName(), is("Test"));

        YangUses yangUses = (YangUses) yangNode.getChild();
        assertThat(yangUses.getName(), is("endpoint"));
    }

    /**
     * Checks uses statement inside container.
     */
    @Test
    public void processUsesInContainer() throws IOException, ParserException {

        YangNode node = manager.getDataModel("src/test/resources/UsesInContainer.yang");

        // Check whether the data model tree returned is of type module.
        assertThat((node instanceof YangModule), is(true));

        // Check whether the node type is set properly to module.
        assertThat(node.getNodeType(), is(YangNodeType.MODULE_NODE));

        // Check whether the module name is set correctly.
        YangModule yangNode = (YangModule) node;
        assertThat(yangNode.getName(), is("Test"));

        YangContainer yangContainer = (YangContainer) yangNode.getChild();
        assertThat(yangContainer.getName(), is("valid"));

        YangUses yangUses = (YangUses) yangContainer.getChild();
        assertThat(yangUses.getName(), is("endpoint"));

        // Check attributes associated with uses.
        assertThat(yangUses.getStatus(), is(YangStatusType.CURRENT));
        assertThat(yangUses.getReference(), is("\"RFC 6020\""));
        assertThat(yangUses.getDescription(), is("\"grouping under test\""));
    }

    /**
     * Checks uses statement inside list.
     */
    @Test
    public void processUsesInList() throws IOException, ParserException {

        YangNode node = manager.getDataModel("src/test/resources/UsesInList.yang");

        // Check whether the data model tree returned is of type module.
        assertThat((node instanceof YangModule), is(true));

        // Check whether the node type is set properly to module.
        assertThat(node.getNodeType(), is(YangNodeType.MODULE_NODE));

        // Check whether the module name is set correctly.
        YangModule yangNode = (YangModule) node;
        assertThat(yangNode.getName(), is("Test"));

        YangList yangList = (YangList) yangNode.getChild();
        assertThat(yangList.getName(), is("valid"));

        YangUses yangUses = (YangUses) yangList.getChild();
        assertThat(yangUses.getName(), is("endpoint"));

        // Check attributes associated with uses.
        assertThat(yangUses.getStatus(), is(YangStatusType.CURRENT));
        assertThat(yangUses.getReference(), is("\"RFC 6020\""));
        assertThat(yangUses.getDescription(), is("\"grouping under test\""));
    }
}
