Ironscales

import json
from Ironscales import fetch_incidents, get_open_incidents_command


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


def test_ironscales_fetch_incident(mocker):
    incidents_mocked_data = util_load_json("test_data/test_get_open_incidents.json")
    last_run = {"data": [0]}
    mocked_client = mocker.Mock()
    mocked_client.get_open_incidents.return_value = {"incident_ids": [0, 1]}
    mocked_client.get_incident.return_value = incidents_mocked_data
    result = fetch_incidents(mocked_client, last_run)
    result_to_compare = [
        {
            "name": "Ironscales incident: IS-1",
            "occurred": "2021-07-06T06:34:00.070Z",
            "rawJSON": json.dumps(incidents_mocked_data),
        }
    ]
    assert result == ([0, 1], result_to_compare)


def test_ironscales_get_open_incidents_empty(mocker):
    mocked_client = mocker.Mock()
    mocked_client.get_open_incidents.return_value = {"incident_ids": []}
    result = get_open_incidents_command(mocked_client, {})
    assert result == "No open incidents were found"