// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.application;

import org.junit.Test;
import org.w3c.dom.Document;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.stream.XMLStreamException;
import javax.xml.transform.*;
import java.io.*;
import java.nio.file.NoSuchFileException;

/**
 * @author lulf
 * @since 5.22
 */
public class IncludeProcessorTest {

    @Test
    public void testInclude() throws IOException, SAXException, XMLStreamException, ParserConfigurationException, TransformerException {
        File app = new File("src/test/resources/multienvapp");
        DocumentBuilder docBuilder = Xml.getPreprocessDocumentBuilder();

        String expected = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><services xmlns:deploy=\"vespa\" xmlns:preprocess=\"properties\" version=\"1.0\">\n" +
                "    <preprocess:properties>\n" +
                "        <qrs.port>4099</qrs.port>\n" +
                "        <qrs.port>5000</qrs.port>\n" +
                "    </preprocess:properties>\n" +
                "    <preprocess:properties deploy:environment='prod'>\n" +
                "        <qrs.port deploy:region='us-west'>5001</qrs.port>" +
                "        <qrs.port deploy:region='us-east'>5002</qrs.port>" +
                "    </preprocess:properties>\n" +
                "    <admin version=\"2.0\">\n" +
                "        <adminserver hostalias=\"node0\"/>\n" +
                "    </admin>\n" +
                "    <admin deploy:environment=\"prod\" version=\"2.0\">\n" +
                "        <adminserver hostalias=\"node1\"/>\n" +
                "    </admin>\n" +
                "    <content id=\"foo\" version=\"1.0\">\n" +
                "    <redundancy>1</redundancy><documents>\n" +
                "    <document mode=\"index\" type=\"music.sd\"/>\n" +
                "</documents><nodes>\n" +
                "    <node distribution-key=\"0\" hostalias=\"node0\"/>\n" +
                "</nodes><nodes deploy:environment=\"prod\">\n" +
                "    <node distribution-key=\"0\" hostalias=\"node0\"/>\n" +
                "    <node distribution-key=\"1\" hostalias=\"node1\"/>\n" +
                "</nodes><nodes deploy:environment=\"prod\" deploy:region=\"us-west\">\n" +
                "    <node distribution-key=\"0\" hostalias=\"node0\"/>\n" +
                "    <node distribution-key=\"1\" hostalias=\"node1\"/>\n" +
                "    <node distribution-key=\"2\" hostalias=\"node2\"/>\n" +
                "</nodes></content>\n" +
                "<jdisc id=\"stateless\" version=\"1.0\">\n" +
                "    <search deploy:environment=\"prod\">\n" +
                "      <chain id=\"common\">\n" +
                "        <searcher id=\"MySearcher1\" />\n" +
                "        <searcher deploy:environment=\"prod\" id=\"MySearcher2\" />\n" +
                "      </chain>\n" +
                "    </search>\n" +
                "    <search/>\n" +
                "    <component id=\"foo\" class=\"MyFoo\" bundle=\"foobundle\" />\n" +
                "    <component id=\"bar\" class=\"TestBar\" bundle=\"foobundle\" deploy:environment=\"dev\" />\n" +
                "    <component id=\"bar\" class=\"ProdBar\" bundle=\"foobundle\" deploy:environment=\"prod\" />\n" +
                "    <component id=\"baz\" class=\"ProdBaz\" bundle=\"foobundle\" deploy:environment=\"prod\" />\n" +
                "    <nodes>\n" +
                "        <node baseport=\"${qrs.port}\" hostalias=\"node0\"/>\n" +
                "    </nodes>\n" +
                "</jdisc></services>";

        Document doc = (new IncludeProcessor(app)).process(docBuilder.parse(Xml.getServices(app)));
        // System.out.println(Xml.documentAsString(doc));
        TestBase.assertDocument(expected, doc);
    }

    @Test(expected = NoSuchFileException.class)
    public void testRequiredIncludeIsDefault() throws ParserConfigurationException, IOException, SAXException, TransformerException {
        File app = new File("src/test/resources/multienvapp_failrequired");
        DocumentBuilder docBuilder = Xml.getPreprocessDocumentBuilder();
        (new IncludeProcessor(app)).process(docBuilder.parse(Xml.getServices(app)));
    }

}