{% macro save_test_history(results) -%}

{{ adapter.dispatch('save_test_history_impl', 'f2dq') (results) }}

{%- endmacro %}

{% macro default__save_test_history_impl(results) %}
    -- depends_on: {{ ref('f2dq_test_history') }}
    {% set command = flags.WHICH %}
    {% if execute and results and command in ('test', 'build') %}
        {% set history_rel = ref('f2dq_test_history') %}
        {% set history_exists = adapter.get_relation(
            database=history_rel.database,
            schema=history_rel.schema,
            identifier=history_rel.identifier
        ) is not none %}
        {% if not history_exists %}
            {% do log(
                "f2dq_test_history relation missing (" ~ history_rel ~ "); skipping save_test_history",
                info=True
            ) %}
        {% else %}
            {% set tests = [] %}
            {% for el in results %}
                {% if el.node.resource_type.value == 'test' %}
                    {% do tests.append(f2dq.test_data_dict(el)) %}
                {% endif %}
            {% endfor %}

            {% if tests %}
                {% do f2dq.insert_list_to_table(
                    history_rel,
                    tests,
                    ['table_name', 'column_name', 'test_name', 'status', 'execution_time', 'message', 'failures_count', 'failures_json', 'failures_table', 'severity', 'compiled_sql', 'run_at'],
                    { 'run_at': timestamp_type() }
                ) %}
            {% endif %}
        {% endif %}

    {% endif %}
    {{ return ('') }}

{% endmacro %}

{% macro test_data_dict(el) %}

    {% set run_started_at_str = run_started_at.strftime('%Y-%m-%d %H:%M:%S') %}

    {% if el.node.to_dict().get('test_metadata') %}
        {% set any_refs = modules.re.findall("ref\(\'(?P<name>.*)\'\)", el.node.test_metadata.kwargs['model']) %}
        {% set any_source = modules.re.findall("source\(\'(?P<one>.*)\'\,\s+\'(?P<two>.*)\'\)", el.node.test_metadata.kwargs['model']) %}

        {% if any_refs %}
            {% set name = any_refs[0] %}
            {% set node_name = f2dq.priv_full_name_from_depends(el.node, name) %}
            {% set schema = graph.nodes.get(node_name)['schema'] %}
            {% set database = graph.nodes.get(node_name)['database'] %}
            {% set table_name = (database + '.' + schema + '.' + name) | lower %} 
            
        {% elif any_source %}
            {% set package_name = any_source[0][0] %}
            {% set name = any_source[0][1] %}
            {% set node_name = f2dq.priv_full_name_from_depends(el.node, name) %}
            {% set schema = graph.sources.get(node_name)['schema'] %}
            {% set database = graph.sources.get(node_name)['database'] %}
            {% set table_name = (database + '.' + schema + '.' + name) | lower %}
        {% else %}
            {% set table_name = none %}
        {% endif %}
    {% else %}
        {% set table_name = none %}
    {% endif %}

    {% if var.has_var('f2dq:query_test_failures') %}
        {% set query_failures = var('f2dq:query_test_failures') %}
    {% else %}
        {% set query_failures = true %}
    {% endif %}

    {% if el.failures and el.failures > 0 and el.node.relation_name and query_failures %}
        {% if var.has_var('f2dq:test_history_failures_limit') %}
            {% set limit_count = var('f2dq:test_history_failures_limit')%}
        {% else %}
            {% set limit_count = 10 %}
        {% endif %}

        {% set failures_query %}
            select * from {{ el.node.relation_name}} limit {{ limit_count }}
        {% endset %}
        {% set failures_list = f2dq.agate_to_list(run_query(failures_query)) %}
    {% endif %}

    {% set failures_json = none %}

    {{ return ({
        'table_name': table_name,
        'column_name': el.node.column_name or none,
        'test_name': el.node.name,
        'status': el.status.name,
        'execution_time': el.execution_time,
        'message': el.message,
        'failures_count': el.failures,
        'failures_json': '' ~ failures_list,
        'failures_table': el.node.relation_name or none,
        'severity': el.node.config.severity,
        'compiled_sql': el.node.compiled_sql or el.node.compiled_code or none,
        'run_at': run_started_at_str,
        })
    }}

{% endmacro %}

{% macro priv_full_name_from_depends(node, name) %}

    {% for full_name in node.depends_on.nodes %}
        {% set node_name = full_name.split('.')[-1] %}
        {% if node_name == name %}
            {{ return(full_name) }}
        {% endif %}
    {% endfor %}

    {{ return(none) }}

{% endmacro %}