import pytest
import requests_mock
from AnsibleTower import Client, delete_host, job_template_launch, create_ad_hoc_command, list_job_events_by_id, output_content
from test_data.test_responses import (
    JOB_TEMPLATE_LAUNCH_RES,
    ADHOC_COMMAND_LAUNCH_RES,
    JOB_TEMPLATE_EXPECTED,
    ADHOC_COMMAND_LAUNCH_EXPECTED,
    ANSIBLE_TOWER_JOB_EVENTS_LIST_BY_ID_RES,
    ANSIBLE_TOWER_JOB_EVENTS_LIST_BY_ID_EXPECTED,
)

API_URL = "https://example"

test_data = [
    (delete_host, {"host_id": "1"}, {}, {"id": "1", "Deleted": True}, "AnsibleAWX.Host(val.id && val.id == obj.id)"),
    (
        job_template_launch,
        {"job_template_id": "1"},
        JOB_TEMPLATE_LAUNCH_RES,
        JOB_TEMPLATE_EXPECTED,
        "AnsibleAWX.Job(val.id && val.id == obj.id)",
    ),
    (
        create_ad_hoc_command,
        {"inventory_id": "1", "credential_id": "1", "module_name": "ping"},
        ADHOC_COMMAND_LAUNCH_RES,
        ADHOC_COMMAND_LAUNCH_EXPECTED,
        "AnsibleAWX.AdhocCommand(val.id && val.id == obj.id)",
    ),
    (
        list_job_events_by_id,
        {"job_id": "39"},
        ANSIBLE_TOWER_JOB_EVENTS_LIST_BY_ID_RES,
        ANSIBLE_TOWER_JOB_EVENTS_LIST_BY_ID_EXPECTED,
        "AnsibleAWX.JobEvents(val.id && val.id == obj.id)",
    ),
]

remove_fields_test_responses = [
    (
        {
            "results": [
                {"related": {"all_groups": "/api/v2/hosts/4/all_groups/"}, "summary_fields": {}, "name": "inventory", "id": 1}
            ]
        }
    ),
    (
        {
            "related": {"all_groups": "/api/v2/hosts/4/all_groups/"},
            "summary_fields": {"groups": {"count": 0, "results": []}},
            "name": "new name",
            "id": 1,
        }
    ),
    ({"name": "new name", "id": 1}),
]


def test_api_request_remove_fields():
    """
    Given:
        - an api endpoint

    When:
        - call api_request

    Then:
        - validating that irrelevant fields - related and summary_fields - removed

    """
    client = Client(API_URL, "username", "password", True, False)
    url = "https://example/api/v2/inventories/"
    for response_mock in remove_fields_test_responses:
        with requests_mock.Mocker() as m:
            m.get(url, status_code=200, json=response_mock)
            response = client.api_request(method="GET", url_suffix="inventories/", params={})
            assert response.get("related", None) is None
            assert response.get("summary_fields", None) is None


def test_client_base_url_legacy():
    """
    Given:
        - is_aap_gateway is False (default)
    When:
        - Client is initialized
    Then:
        - base_url should use the legacy /api/v2/ endpoint
    """
    client = Client(API_URL, "username", "password", True, False)
    assert client._base_url == "https://example/api/v2/"


def test_client_base_url_aap_gateway():
    """
    Given:
        - is_aap_gateway is True
    When:
        - Client is initialized
    Then:
        - base_url should use the new /api/controller/v2/ endpoint
    """
    client = Client(API_URL, "username", "password", True, False, is_aap_gateway=True)
    assert client._base_url == "https://example/api/controller/v2/"


@pytest.mark.parametrize("command, args, response, expected_result, output_prefix", test_data)
def test_check_command_result_output(command, args, response, expected_result, output_prefix, mocker):
    """
    Given:
        - parameters to launch

    When:
        - call api_request and use the response for command results

    Then:
        - validating that irrelevant fields removed and that the added fields was added correctly

    """

    client = Client(API_URL, "username", "password", True, False)
    mocker.patch.object(client, "api_request", return_value=response)
    results = command(client, args)
    if isinstance(results, list):
        output = results[0].to_context().get("EntryContext", {})
    else:
        output = results.to_context().get("EntryContext", {})
    assert output.get(output_prefix, "") == expected_result


def test_filtered_data():
    """
    Given:
        - a string to filter lines from the stdout

    When:
        - print output is True and the user provide a string to filter

    Then:
        - validating that the returned text is filtered and arranged correctly

    """
    content = "UNREACHABLE! => \n line 1 \n line 2"
    actual_output = output_content(content, True, "reachable", "Test headline\n")
    expected_output = "Test headline\nFiltered text: reachable\n\nUNREACHABLE! => \n"
    assert actual_output == expected_output