/*
 * 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.hamcrest.CoreMatchers.startsWith;
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.assertTrue;

import java.util.Collection;
import org.eclipse.jdt.annotation.NonNull;
import org.junit.jupiter.api.Test;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
import org.opendaylight.yangtools.yang.model.api.stmt.UniqueEffectiveStatement;

class Bug5946Test extends AbstractYangTest {
    private static final String NS = "foo";
    private static final String REV = "2016-05-26";
    private static final QName L1 = QName.create(NS, REV, "l1");
    private static final QName L2 = QName.create(NS, REV, "l2");
    private static final QName L3 = QName.create(NS, REV, "l3");
    private static final QName C = QName.create(NS, REV, "c");
    private static final QName WITHOUT_UNIQUE = QName.create(NS, REV, "without-unique");
    private static final QName SIMPLE_UNIQUE = QName.create(NS, REV, "simple-unique");
    private static final QName MULTIPLE_UNIQUE = QName.create(NS, REV, "multiple-unique");
    private static final SchemaNodeIdentifier L1_ID = SchemaNodeIdentifier.Descendant.of(L1);
    private static final SchemaNodeIdentifier L2_ID = SchemaNodeIdentifier.Descendant.of(L2);
    private static final SchemaNodeIdentifier C_L3_ID = SchemaNodeIdentifier.Descendant.of(C, L3);

    @Test
    void test() {
        final var context = assertEffectiveModel("/bugs/bug5946/foo.yang");

        var uniqueConstraints = getListConstraints(context, WITHOUT_UNIQUE);
        assertNotNull(uniqueConstraints);
        assertTrue(uniqueConstraints.isEmpty());

        var simpleUniqueConstraints = getListConstraints(context, SIMPLE_UNIQUE);
        assertNotNull(simpleUniqueConstraints);
        assertEquals(1, simpleUniqueConstraints.size());
        var simpleUniqueConstraintTag = simpleUniqueConstraints.iterator().next().argument();
        assertTrue(simpleUniqueConstraintTag.contains(L1_ID));
        assertTrue(simpleUniqueConstraintTag.contains(C_L3_ID));

        var multipleUniqueConstraints = getListConstraints(context, MULTIPLE_UNIQUE);
        assertNotNull(multipleUniqueConstraints);
        assertEquals(3, multipleUniqueConstraints.size());
        boolean l1l2 = false;
        boolean l1cl3 = false;
        boolean cl3l2 = false;
        for (var uniqueConstraint : multipleUniqueConstraints) {
            var uniqueConstraintTag = uniqueConstraint.argument();
            if (uniqueConstraintTag.contains(L1_ID) && uniqueConstraintTag.contains(L2_ID)) {
                l1l2 = true;
            } else if (uniqueConstraintTag.contains(L1_ID) && uniqueConstraintTag.contains(C_L3_ID)) {
                l1cl3 = true;
            } else if (uniqueConstraintTag.contains(C_L3_ID) && uniqueConstraintTag.contains(L2_ID)) {
                cl3l2 = true;
            }
        }
        assertTrue(l1l2 && l1cl3 && cl3l2);
    }

    @Test
    void testInvalid() {
        assertSourceException(
            startsWith("Unique statement argument '/simple-unique/l1' contains schema node identifier "
                + "'/simple-unique/l1' which is not in the descendant node identifier form."),
            "/bugs/bug5946/foo-invalid.yang");
    }

    private static  @NonNull Collection<? extends @NonNull UniqueEffectiveStatement> getListConstraints(
            final SchemaContext context, final QName listQName) {
        return assertInstanceOf(ListSchemaNode.class, context.getDataChildByName(listQName)).getUniqueConstraints();
    }
}