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

import java.util.Collection;
import java.util.Iterator;
import java.util.Set;
import java.util.stream.Collectors;
import org.junit.Test;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.ExtensionDefinition;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.stmt.UnrecognizedStatement;

/**
 * Test antlr grammar capability to parse nested unknown nodes.
 */
public class Bug394Test extends AbstractYangTest {
    @Test
    public void testParseList() throws Exception {
        final EffectiveModelContext context = assertEffectiveModelDir("/bugs/bug394-retest");
        final Module bug394 = context.findModules("bug394").iterator().next();
        final Module bug394_ext = context.findModules("bug394-ext").iterator().next();

        final ContainerSchemaNode logrecords = (ContainerSchemaNode) bug394.getDataChildByName(QName.create(
                bug394.getQNameModule(), "logrecords"));
        assertNotNull(logrecords);

        final Collection<? extends UnrecognizedStatement> nodes = logrecords.asEffectiveStatement().getDeclared()
            .declaredSubstatements(UnrecognizedStatement.class);
        assertEquals(2, nodes.size());

        final Set<QName> extensions = bug394_ext.getExtensionSchemaNodes().stream()
            .map(ExtensionDefinition::getQName)
            .collect(Collectors.toUnmodifiableSet());
        assertEquals(3, extensions.size());

        final Iterator<? extends UnrecognizedStatement> it = nodes.iterator();
        assertTrue(extensions.contains(it.next().statementDefinition().getStatementName()));
        assertTrue(extensions.contains(it.next().statementDefinition().getStatementName()));
    }
}