/*
 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.startsWith;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertInstanceOf;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.opendaylight.yangtools.yang.stmt.StmtTestUtils.sourceForResource;

import com.google.common.collect.ImmutableSetMultimap;
import java.util.Optional;
import org.junit.jupiter.api.Test;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;

class Bug8307Test {
    private static final StatementStreamSource FOO_MODULE = sourceForResource("/bugs/bug8307/foo.yang");
    private static final StatementStreamSource BAR_MODULE = sourceForResource("/bugs/bug8307/bar.yang");
    private static final StatementStreamSource BAZ_MODULE = sourceForResource("/bugs/bug8307/baz.yang");
    private static final StatementStreamSource FOOBAR_MODULE = sourceForResource("/bugs/bug8307/foobar.yang");
    private static final StatementStreamSource FOO_INVALID_MODULE = sourceForResource("/bugs/bug8307/foo-invalid.yang");
    private static final StatementStreamSource BAR_INVALID_MODULE = sourceForResource("/bugs/bug8307/bar-invalid.yang");
    private static final StatementStreamSource BAZ_INVALID_MODULE = sourceForResource("/bugs/bug8307/baz-invalid.yang");

    private static final QNameModule FOO = QNameModule.of("foo-ns", "2017-05-16");
    private static final QName MY_FOO_CONT_A = QName.create(FOO, "my-foo-cont-a");
    private static final QName MY_FOO_CONT_B = QName.create(FOO, "my-foo-cont-b");
    private static final QName MY_FOO_CONT_C = QName.create(FOO, "my-foo-cont-c");
    private static final QNameModule BAR = QNameModule.of("bar-ns", "2017-05-16");
    private static final QName MY_BAR_CONT_A = QName.create(BAR, "my-bar-cont-a");
    private static final QName MY_BAR_CONT_B = QName.create(BAR, "my-bar-cont-b");
    private static final QNameModule BAZ = QNameModule.of("baz-ns", "2017-05-16");
    private static final QName MY_BAZ_CONT = QName.create(BAZ, "my-baz-cont");

    @Test
    void testDeviationsSupportedInSomeModules() throws Exception {
        final var schemaContext = RFC7950Reactors.defaultReactor().newBuild()
            .addSources(FOO_MODULE, BAR_MODULE, BAZ_MODULE, FOOBAR_MODULE)
            .setModulesWithSupportedDeviations(ImmutableSetMultimap.<QNameModule, QNameModule>builder()
                .put(FOO, BAR)
                .put(FOO, BAZ)
                .put(BAR, BAZ)
                .build())
            .buildEffective();
        assertNotNull(schemaContext);

        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_FOO_CONT_A));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_FOO_CONT_B));
        assertNotNull(schemaContext.findDataTreeChild(MY_FOO_CONT_C).orElse(null));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_BAR_CONT_A));
        assertNotNull(schemaContext.findDataTreeChild(MY_BAR_CONT_B).orElse(null));
    }

    @Test
    void testDeviationsSupportedInAllModules() throws Exception {
        final var schemaContext = RFC7950Reactors.defaultReactor().newBuild()
            .addSources(FOO_MODULE, BAR_MODULE, BAZ_MODULE, FOOBAR_MODULE)
            .buildEffective();
        assertNotNull(schemaContext);

        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_FOO_CONT_A));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_FOO_CONT_B));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_FOO_CONT_C));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_BAR_CONT_A));
        assertEquals(Optional.empty(), schemaContext.findDataTreeChild(MY_BAR_CONT_B));
    }

    @Test
    void testDeviationsSupportedInNoModule() throws Exception {
        final var schemaContext = RFC7950Reactors.defaultReactor().newBuild()
            .addSources(FOO_MODULE, BAR_MODULE, BAZ_MODULE, FOOBAR_MODULE)
            .setModulesWithSupportedDeviations(ImmutableSetMultimap.of())
            .buildEffective();
        assertNotNull(schemaContext);

        assertNotNull(schemaContext.findDataTreeChild(MY_FOO_CONT_A).orElse(null));
        assertNotNull(schemaContext.findDataTreeChild(MY_FOO_CONT_B).orElse(null));
        assertNotNull(schemaContext.findDataTreeChild(MY_FOO_CONT_C).orElse(null));
        assertNotNull(schemaContext.findDataTreeChild(MY_BAR_CONT_A).orElse(null));
        assertNotNull(schemaContext.findDataTreeChild(MY_BAR_CONT_B).orElse(null));
    }

    @Test
    void shouldFailOnAttemptToDeviateTheSameModule() {
        final var reactor = RFC7950Reactors.defaultReactor().newBuild().addSources(FOO_INVALID_MODULE);

        final var cause = assertInstanceOf(InferenceException.class,
            assertThrows(ReactorException.class, reactor::buildEffective).getCause());
        assertThat(cause.getMessage(),
            startsWith("Deviation must not target the same module as the one it is defined in"));
    }

    @Test
    void shouldFailOnAttemptToDeviateTheSameModule2() {
        final var reactor = RFC7950Reactors.defaultReactor().newBuild()
            .addSources(BAR_INVALID_MODULE, BAZ_INVALID_MODULE);

        final var cause = assertInstanceOf(InferenceException.class,
            assertThrows(ReactorException.class, reactor::buildEffective).getCause());
        assertThat(cause.getMessage(),
            startsWith("Deviation must not target the same module as the one it is defined in"));
    }
}