URLReputation

import demistomock as demisto  # noqa: F401
import pytest
from CommonServerPython import *  # noqa: F401


@pytest.mark.parametrize("contents", ({"Error": "error"}, None))
def test_url_reputation(mocker, contents):
    """
    Given:
        - Script args: URL string.

    When:
        - Running url_reputation function.

    Then:
        - Validating the outputs as expected.
    """
    from URLReputation import url_reputation

    mocker.patch.object(demisto, "args", return_value={"url": "www.url.com"})
    execute_command_res = [{"Type": 4, "Contents": contents, "Brand": "brand"}]
    execute_mock = mocker.patch.object(demisto, "executeCommand", return_value=execute_command_res)
    results_mock = mocker.patch.object(demisto, "results")
    url_reputation()
    assert execute_mock.call_count == 1
    assert "returned an error" in results_mock.call_args[0][0][0]["Contents"]


def test_url_reputation_ignore_offset_error(mocker):
    """
    Given:
        - Script args: URL string.

    When:
        - Running url_reputation function using VT integration and an error entry (type 4) of "offset 1" is returned.

    Then:
        - Ensure the script will ignore the offset 1 error.
    """
    from URLReputation import url_reputation

    mocker.patch.object(demisto, "args", return_value={"url": "www.url.com"})
    execute_command_res = [{"Type": 4, "Contents": {"Offset": 1}, "Brand": "VirusTotal (API v3)"}]
    execute_mock = mocker.patch.object(demisto, "executeCommand", return_value=execute_command_res)
    results_mock = mocker.patch.object(demisto, "results")
    url_reputation()
    assert execute_mock.call_count == 1
    assert results_mock.call_args[0][0] == []