from kafka.admin import NewTopic
import pytest

from dbdeploy.dtos import ExecutionJob


@pytest.fixture
def mock_xml():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         run-always="true"
         neo4j-server='music-graph'
        >
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <cypherquery>
                <![CDATA[
                    MERGE (t:TestNode {test: event.test})
                ]]>
            </cypherquery>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_multi_changeset_xml():
    return """<?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-on-change="true"
        >
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table1
                    ]]>
                </sqlquery>
            </precondition>
            <cypherquery>
                <![CDATA[
                    MERGE (t:TestNode1 {test: event.test})
                ]]>
            </cypherquery>
        </changeset>
        <changeset
            id="TEST:2"
            author="ibolshakov"
            neo4j-server='music-graph'
            run-on-change="true"
        >
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table2
                    ]]>
                </sqlquery>
            </precondition>
            <cypherquery>
                <![CDATA[
                    MERGE (t:TestNode2 {test: event.test})
                ]]>
            </cypherquery>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_bad_xml():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1" author="ibolshakov">
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_default_neo4j_server_xml():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         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>
    """


@pytest.fixture
def mock_unknown_neo4j_server_xml():
    return """<?xml version="1.0" encoding="UTF-8"?>
        <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
            <changeset id="TEST:1"
             author="ssavva"
             neo4j-server='UNKNOWN-NEO4J-SERVER'
            >
                <cypherquery>
                    <![CDATA[
                        MERGE (t:TestNode {test: event.test})
                    ]]>
                </cypherquery>
            </changeset>
        </changelog>
        """


@pytest.fixture
def mock_skip_cypher():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         run-always="true"
         topic="testTopic.name"
         kafka-cluster-name="cdc-destination">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_skip_cypher_kafka_cluster_not_set():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         run-always="true"
         topic="testTopic.name">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_skip_cypher_missing_topic_name():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_skip_snowflake_to_mysql():
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="ibolshakov"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
                <column>
                    <name>
                        test_col2
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_joblist():
    return [
        ExecutionJob(
            changeset_id='test_changeset:1',
            id='f007ba11-1111-0000-0000-000000000000',
            topic=NewTopic(name=None, num_partitions=1, replication_factor=3),  # noqa: E501
            dlq_topic=NewTopic(name=None, num_partitions=1, replication_factor=3),  # noqa: E501
            precondition=True,
            snowflake_account='orchard',
            sql_query='SELECT test_col FROM test_table',
            cypher_query='MERGE (t:TestNode {test: event.test_col})',
            offsets={},
            is_done=False,
            topics_created=False,
            connector_started=False),
        ExecutionJob(
            changeset_id='test_changeset:2',
            id='f007ba11-2222-0000-0000-000000000000',
            topic=NewTopic(name=None, num_partitions=1, replication_factor=3),  # noqa: E501
            dlq_topic=NewTopic(name=None, num_partitions=1, replication_factor=3),  # noqa: E501
            precondition=True,
            snowflake_account='orchard',
            sql_query='SELECT test_col FROM test_table',
            cypher_query='MERGE (t:TestNode {test: event.test_col})',
            offsets={},
            is_done=False,
            topics_created=False,
            connector_started=False)]


@pytest.fixture
def mock_job():
    return ExecutionJob(
        changeset_id='test_changeset:1',
        id='f007ba11-1111-0000-0000-000000000000',
        topic=NewTopic(name='topic_1', num_partitions=1, replication_factor=3),  # noqa: E501
        dlq_topic=NewTopic(name='dlqtopic_1', num_partitions=1, replication_factor=3),  # noqa: E501
        precondition=True,
        snowflake_account='orchard',
        sql_query='SELECT test_col FROM test_table',
        cypher_query='MERGE (t:TestNode {test: event.test_col})',
        offsets={},
        is_done=False,
        topics_created=False,
        neo4j_server='music-graph',
        connector_started=False)


@pytest.fixture
def mock_job_nosql():
    return ExecutionJob(
        changeset_id='test_changeset:1',
        id='f007ba11-1111-0000-0000-000000000000',
        topic=NewTopic(name='topic_1', num_partitions=1, replication_factor=3),  # noqa: E501
        dlq_topic=NewTopic(name='dlqtopic_1', num_partitions=1, replication_factor=3),  # noqa: E501
        precondition=False,
        snowflake_account='orchard',
        sql_query=None,
        cypher_query='MERGE (t:TestNode {test: event.test_col})',
        offsets={},
        is_done=False,
        topics_created=False,
        connector_started=False)


@pytest.fixture
def mock_job_nocypher():
    return ExecutionJob(
        changeset_id='test_changeset:1',
        id='f007ba11-1111-0000-0000-000000000000',
        precondition=True,
        sql_query='SELECT test_col FROM test_table',
        cypher_query=None,
        snowflake_account='orchard',
        offsets={},
        is_done=False,
        skip_sink_connector=True,
        topic_name='test.TopicName',
        topics_created=False,
        connector_started=False)


@pytest.fixture
def mock_snowflake_to_mysql_with_upsert():
    """XML fixture with insert-mode="upsert" specified."""
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="test"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1"
             insert-mode="upsert">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
                <column>
                    <name>
                        test_col2
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_snowflake_to_mysql_with_insert():
    """XML fixture with insert-mode="insert" specified."""
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="test"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1"
             insert-mode="insert">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_snowflake_to_mysql_with_delete():
    """XML fixture with insert-mode="delete" specified."""
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="test"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1"
             insert-mode="delete">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_snowflake_to_mysql_with_invalid_insert_mode():
    """XML fixture with invalid insert-mode specified."""
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="test"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1"
             insert-mode="invalid_mode">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """


@pytest.fixture
def mock_snowflake_to_mysql_default_insert_mode():
    """XML fixture with no insert-mode specified (should default to 'update')."""
    return """<?xml version="1.0" encoding="UTF-8"?>
    <changelog xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
        <changeset id="TEST:1"
         author="test"
         run-always="true">
            <precondition>
                <sqlquery>
                    <![CDATA[
                    SELECT test_col FROM test_table
                    ]]>
                </sqlquery>
            </precondition>
            <tableschema
             table-name="test_table"
             primary-key="test_col1">
                <column>
                    <name>
                        test_col1
                    </name>
                    <type>
                        string
                    </type>
                </column>
            </tableschema>
        </changeset>
    </changelog>
    """