import pytest
import demistomock as demisto
from IncidentsCheck_Widget_CommandsNames import main, random


@pytest.mark.parametrize(
    "list_, expected",
    [
        (
            [{"Contents": "name 1,name 2"}],
            '[{"data": [1], "name": "name 1", "color": "#0003e8"}, {"data": [1], "name": "name 2", "color": "#0003e8"}]',
        ),
        ([{"Contents": ""}], '[{"data": [0], "name": "N/A", "color": "#00CD33"}]'),
        ([{}], '[{"data": [0], "name": "N/A", "color": "#00CD33"}]'),
    ],
)
def test_script(mocker, list_, expected):
    mocker.patch.object(random, "randint", return_value=1000)
    mocker.patch.object(demisto, "executeCommand", return_value=list_)
    mocker.patch.object(demisto, "results")

    main()

    contents = demisto.results.call_args[0][0]
    assert contents == expected