import json

"""CONSTANTS"""

BASE_URL = "http://test.com/"
ACCESS_ID = "access_id"
SECRET_KEY = "secret_key"


def util_load_json(path):
    with open(path, encoding="utf-8") as f:
        return json.loads(f.read())


def test_ip(requests_mock):
    from CommonServerPython import Common
    from CTIX import Client, ip_details_command

    ip_to_check = "6.7.8.9"
    mock_response = util_load_json("test_data/ip_details.json")
    requests_mock.get(f"http://test.com/objects/indicator/?q={ip_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"ip": ip_to_check, "enhanced": False}

    response = ip_details_command(client, args)

    assert response[0].outputs == mock_response["results"][0]
    assert response[0].outputs_prefix == "CTIX.IP"
    assert response[0].outputs_key_field == "name2"

    assert isinstance(response, list)
    assert len(response) == 1
    assert isinstance(response[0].indicator, Common.IP)
    assert response[0].indicator.ip == ip_to_check


def test_ip_not_found(requests_mock):
    from CTIX import Client, ip_details_command

    ip_to_check = "1.1.1.1"
    mock_response = {"results": []}
    requests_mock.get(f"http://test.com/objects/indicator/?q={ip_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"ip": ip_to_check, "enhanced": False}

    response = ip_details_command(client, args)

    assert response[0].outputs == []
    assert response[0].readable_output == f"No matches found for IP {ip_to_check}"


def test_domain(requests_mock):
    from CommonServerPython import Common
    from CTIX import Client, domain_details_command

    domain_to_check = "testing.com"
    mock_response = util_load_json("test_data/domain_details.json")
    requests_mock.get(f"http://test.com/objects/indicator/?q={domain_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"domain": domain_to_check, "enhanced": False}

    response = domain_details_command(client, args)

    assert response[0].outputs == mock_response["results"][0]
    assert response[0].outputs_prefix == "CTIX.Domain"
    assert response[0].outputs_key_field == "name2"

    assert isinstance(response, list)
    assert len(response) == 1
    assert isinstance(response[0].indicator, Common.Domain)
    assert response[0].indicator.domain == domain_to_check


def test_domain_not_found(requests_mock):
    from CTIX import Client, domain_details_command

    domain_to_check = "abc.com"
    mock_response = {"results": []}
    requests_mock.get(f"http://test.com/objects/indicator/?q={domain_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"domain": domain_to_check, "enhanced": False}

    response = domain_details_command(client, args)

    assert response[0].outputs == []
    assert response[0].readable_output == f"No matches found for Domain {domain_to_check}"


def test_url(requests_mock):
    from CommonServerPython import Common
    from CTIX import Client, url_details_command

    url_to_check = "https://www.ibm.com/support/mynotifications/"
    mock_response = util_load_json("test_data/url_details.json")
    requests_mock.get(f"http://test.com/objects/indicator/?q={url_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"url": url_to_check, "enhanced": False}

    response = url_details_command(client, args)

    assert response[0].outputs == mock_response["results"][0]
    assert response[0].outputs_prefix == "CTIX.URL"
    assert response[0].outputs_key_field == "name2"

    assert isinstance(response, list)
    assert len(response) == 1
    assert isinstance(response[0].indicator, Common.URL)
    assert response[0].indicator.url == url_to_check


def test_url_not_found(requests_mock):
    from CTIX import Client, url_details_command

    url_to_check = "https://abc.com/path"
    mock_response = {"results": []}
    requests_mock.get(f"http://test.com/objects/indicator/?q={url_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"url": url_to_check, "enhanced": False}

    response = url_details_command(client, args)

    assert response[0].outputs == []
    assert response[0].readable_output == f"No matches found for URL {url_to_check}"


def test_file(requests_mock):
    from CommonServerPython import Common
    from CTIX import Client, file_details_command

    file_to_check = "4d552241543b8176a3189864a16b6052f9d163a124291ec9552e1b77"
    mock_response = util_load_json("test_data/file_details.json")
    requests_mock.get(f"http://test.com/objects/indicator/?q={file_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"file": file_to_check, "enhanced": False}

    response = file_details_command(client, args)

    assert response[0].outputs == mock_response["results"][0]
    assert response[0].outputs_prefix == "CTIX.File"
    assert response[0].outputs_key_field == "name2"

    assert isinstance(response, list)
    assert len(response) == 1
    assert isinstance(response[0].indicator, Common.File)
    assert response[0].indicator.name == file_to_check


def test_file_not_found(requests_mock):
    from CTIX import Client, file_details_command

    file_to_check = "6AD8334857B3F054A9F93BA380B5555B"
    mock_response = {"results": []}
    requests_mock.get(f"http://test.com/objects/indicator/?q={file_to_check}", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})

    args = {"file": file_to_check, "enhanced": False}

    response = file_details_command(client, args)

    assert response[0].outputs == []
    assert response[0].readable_output == f"No matches found for FILE {file_to_check}"


def test_create_intel(requests_mock):
    from CTIX import Client, create_intel_command

    mock_response = util_load_json("test_data/create_intel.json")
    requests_mock.post("http://test.com/create-intel/", json=mock_response)

    client = Client(base_url=BASE_URL, access_id=ACCESS_ID, secret_key=SECRET_KEY, verify=False, proxies={})
    post_data = {"ips": "1.2.3.4,3.45.56.78", "urls": "https://abc_test.com,https://test_abc.com"}
    response = create_intel_command(client, post_data)

    assert all(k in response["CTIX"]["Intel"]["response"] for k in ("data", "status"))
    assert "status" in response["CTIX"]["Intel"]
    assert response["CTIX"]["Intel"]["response"]["status"] == 201