import demistomock as demisto
from QualysCreateIncidentFromReport import main, get_asset_id_for_ip


def test_main(mocker):
    """
    Tests the full flow of the script

    Given: A valid report and successful responses
    When: Running the QualysCreateIncidentReport script
    Then: Return a successful response

    """
    with open("test_data/qualys_host_list_rawresponse.xml") as f:
        raw_response = f.read()
    mocker.patch.object(demisto, "args", return_value={})
    mocker.patch.object(
        demisto, "getFilePath", return_value={"id": id, "path": "test_data/test_report.xml", "name": "test_report.xml"}
    )
    mocker.patch.object(
        demisto,
        "executeCommand",
        side_effect=[[{"Contents": raw_response, "Type": "notes"}], [{"Contents": {"total": 1}, "Type": "notes"}]],
    )
    demisto_results = mocker.spy(demisto, "results")
    main()
    demisto_results.assert_called_once_with("Done.")


def test_get_asset_id_for_ip(mocker):
    """
    Tests parsing the data returned by qualys-host-list

    Given: A valid response from qualys
    When: Parsing for the incidentid
    Then: Return a valid id

    """
    with open("test_data/qualys_host_list_rawresponse.xml") as f:
        raw_response = f.read()
    mocker.patch.object(demisto, "executeCommand", return_value=[{"Contents": raw_response, "Type": "note"}])
    assert get_asset_id_for_ip("1.1.1.1") == "69291564"