"""STANDALONE FUNCTION UNIT TESTS"""


def test_regex_result():
    """
    Given:
        - dig output
    When:
        - running regex_result function
    Then:
        - validate the output is as expected
    """
    from Dig import regex_result

    dig_output = "0.0.0.0, 1.1.1.1"
    resolved_addresses, dns_server = regex_result(dig_output, reverse_lookup=False)
    assert resolved_addresses, dns_server == (["0.0.0.0"], "1.1.1.1")

    resolved_addresses, dns_server = regex_result(dig_output, reverse_lookup=True)
    assert resolved_addresses, dns_server == (["0.0.0.0"], "1.1.1.1")


def test_dig_result(mocker):
    """
    Given:
        - server and name
    When:
        - running dig_result function
    Then:
        - validate the output is as expected
    """
    import subprocess

    import Dig
    from Dig import dig_result

    mocker.patch.object(Dig, "regex_result", return_value=(["0.0.0.0"], "1.1.1.1"))
    mocker.patch.object(subprocess, "check_output", return_value="OK")
    dig_result("server", "name") == {"name": "name", "resolvedaddresses": ["0.0.0.0"], "nameserver": "1.1.1.1"}  # noqa: B015
    dig_result("", "name") == {"name": "name", "resolvedaddresses": ["0.0.0.0"], "nameserver": "1.1.1.1"}  # noqa: B015


def test_reverse_dig_result(mocker):
    """
    Given:
        - server and name
    When:
        - running reverse_dig_result function
    Then:
        - validate the output is as expected
    """
    import subprocess

    import Dig
    from Dig import reverse_dig_result

    mocker.patch.object(Dig, "regex_result", return_value=(["0.0.0.0"], "1.1.1.1"))
    mocker.patch.object(subprocess, "check_output", return_value="OK")
    reverse_dig_result("server", "name") == {  # noqa: B015
        "name": "name",
        "resolvedaddresses": ["0.0.0.0"],
        "nameserver": "1.1.1.1",
    }  # noqa: B015
    reverse_dig_result("", "name") == {"name": "name", "resolvedaddresses": ["0.0.0.0"], "nameserver": "1.1.1.1"}  # noqa: B015


""" COMMAND FUNCTION UNIT TESTS"""


def test_dig_command(mocker):
    """
    Given:
        - server, name and reverseLookup as arguments
    When:
        - running dig_command function
    Then:
        - validate the returned XSOAR CommandResults is as expected
    """
    import Dig
    from Dig import dig_command

    dig_result_return_value = {"name": "name", "resolvedaddresses": ["0.0.0.0"], "nameserver": "1.1.1.1"}
    mocker.patch.object(Dig, "dig_result", return_value=dig_result_return_value)
    mocker.patch.object(Dig, "reverse_dig_result", return_value=dig_result_return_value)
    command_result = dig_command({"server": "server", "name": "name", "reverseLookup": False})
    command_result.outputs = dig_result_return_value

    command_result = dig_command({"server": "server", "name": "name", "reverseLookup": True})
    command_result.outputs = dig_result_return_value