/*
 * Copyright (c) 2017 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.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import com.google.common.collect.ImmutableSet;
import java.util.Optional;
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.stmt.FeatureEffectiveStatement;

public class Bug8922Test {
    private static final QName MY_CON = QName.create("foo", "my-con");
    private static final QName TARGET = QName.create("foo", "target");

    @Test
    public void testAllFeaturesSupported() throws Exception {
        final var context = StmtTestUtils.parseYangSource("/bugs/bug8922/foo.yang");
        assertNotNull(context);
        final var findNode = context.findDataTreeChild(TARGET, MY_CON).get();
        assertThat(findNode, instanceOf(ContainerSchemaNode.class));
        assertEquals(Optional.of("New description"), findNode.getDescription());

        assertEquals(1, context.findModuleStatements("foo").iterator().next()
            .streamEffectiveSubstatements(FeatureEffectiveStatement.class).count());
    }

    @Test
    public void testNoFeatureSupported() throws Exception {
        final var context = StmtTestUtils.parseYangSource("/bugs/bug8922/foo.yang", ImmutableSet.of());
        assertNotNull(context);
        assertEquals(Optional.empty(), context.findDataTreeChild(TARGET, MY_CON));
        assertTrue(context.getAvailableAugmentations().isEmpty());

        assertEquals(0, context.findModuleStatements("foo").iterator().next()
            .streamEffectiveSubstatements(FeatureEffectiveStatement.class).count());
    }
}