package org.opendaylight.netconf.mdsal.connector.ops.get;

import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.model.InitializationError;
import org.opendaylight.controller.config.util.xml.XmlElement;
import org.opendaylight.controller.config.util.xml.XmlUtil;
import org.opendaylight.netconf.mdsal.connector.CurrentSchemaContext;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
import org.w3c.dom.Document;
import org.xml.sax.SAXException;

@RunWith(value = Parameterized.class)
public class FilterContentValidatorTest {

    private static final int TEST_CASE_COUNT = 8;
    private final XmlElement filterContent;
    private final String expected;
    private FilterContentValidator validator;

    @Parameterized.Parameters
    public static Collection<Object[]> data() throws IOException, SAXException, URISyntaxException, InitializationError {
        List<Object[]> result = new ArrayList<>();
        final Path path = Paths.get(FilterContentValidatorTest.class.getResource("/filter/expected.txt").toURI());
        final List<String> expected = Files.readAllLines(path);
        if (expected.size() != TEST_CASE_COUNT) {
            throw new InitializationError("Number of lines in results file must be same as test case count");
        }
        for (int i = 1; i <= TEST_CASE_COUNT; i++) {
            final Document document = XmlUtil.readXmlToDocument(FilterContentValidatorTest.class.getResourceAsStream("/filter/f" + i + ".xml"));
            result.add(new Object[]{document, expected.get(i-1)});
        }
        return result;
    }

    public FilterContentValidatorTest(Document filterContent, String expected) {
        this.filterContent = XmlElement.fromDomDocument(filterContent);
        this.expected = expected;
    }

    @Before
    public void setUp() throws Exception {
        List<InputStream> sources = new ArrayList<>();
        sources.add(getClass().getResourceAsStream("/yang/filter-validator-test-mod-0.yang"));
        sources.add(getClass().getResourceAsStream("/yang/filter-validator-test-augment.yang"));
        SchemaContext context = parseYangSources(sources);
        CurrentSchemaContext currentContext = mock(CurrentSchemaContext.class);
        doReturn(context).when(currentContext).getCurrentContext();
        validator = new FilterContentValidator(currentContext);
    }

    @Test
    public void testValidate() throws Exception {
        if (expected.startsWith("success")) {
            final String expId = expected.replace("success=", "");
            Assert.assertEquals(expId, validator.validate(filterContent).toString());
        } else if (expected.startsWith("error")) {
            try {
                validator.validate(filterContent);
                Assert.fail(XmlUtil.toString(filterContent) + " is not valid and should throw exception.");
            } catch (Exception e) {
                final String expectedExceptionClass = expected.replace("error=", "");
                Assert.assertEquals(expectedExceptionClass, e.getClass().getName());
            }
        }

    }

    public static SchemaContext parseYangSources(Collection<InputStream> testFiles)
            throws SourceException, ReactorException, FileNotFoundException {
        CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR
                .newBuild();
        for (InputStream testFile : testFiles) {
            reactor.addSource(new YangStatementSourceImpl(testFile));
        }
        return reactor.buildEffective();
    }
}