"""Test xml helper utilities."""
from dbdeploy.base import config
from dbdeploy.util.xml_helpers import get_schema_file_path
from dbdeploy.util.xml_helpers import validate


class TestXMLValidation:
    def test_xml_is_invalid(self):
        config.FLOW_NAME = 'snowflake-neo4j'
        xml_schema = get_schema_file_path(config.FLOW_NAME)
        xml = """<?xml version="1.0" encoding="UTF-8"?>
        <test></test>
        """
        valid, _ = validate(xml_schema_file_path=xml_schema, xmlfile=xml)
        assert not valid

    def test_xml_is_valid(self):

        config.FLOW_NAME = 'snowflake-neo4j'
        xml_schema = get_schema_file_path(config.FLOW_NAME)
        xml = """<?xml version="1.0" encoding="UTF-8"?>
        <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
            <changeset
                id="TEST:1"
                author="ibolshakov"
                neo4j-server="music-graph"
                run-always="true"
            >
                <precondition>
                    <sqlquery>
                        <![CDATA[
                        SELECT test_col FROM test_table
                        ]]>
                    </sqlquery>
                </precondition>
                <cypherquery>
                    <![CDATA[
                        MERGE (t:TestNode {test: event.test})
                    ]]>
                </cypherquery>
            </changeset>
        </changelog>
        """
        valid, _ = validate(xml_schema_file_path=xml_schema, xmlfile=xml)
        assert valid

    def test_neo4j_xml_is_valid(self):
        config.FLOW_NAME = 'neo4j'
        xml_schema = get_schema_file_path(config.FLOW_NAME)
        xml = """<?xml version="1.0" encoding="UTF-8"?>
        <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
            <changeset id="TEST:1" author="ibolshakov" run-on-change="true">
                <cypherquery>
                    <![CDATA[
                        MERGE (t:TestNode {test: event.test})
                    ]]>
                </cypherquery>
            </changeset>
        </changelog>
        """
        valid, _ = validate(xml_schema_file_path=xml_schema, xmlfile=xml)
        assert valid