from unittest.mock import patch
import pytest
@pytest.mark.parametrize(
"indexes, expected_results",
[
("0", [0]),
("0,1,2", [0, 1, 2]),
("0,2", [0, 2]),
("0,2-3", [0, 2, 3]),
("0,2-2", [0, 2]),
("2-3", [2, 3]),
],
)
def test_get_range_command_by_index_list(mocker, indexes, expected_results):
"""
Given
- List of indexes.
When
- Calling main method.
Then
- Return filtered list.
"""
import demistomock as demisto
from GetRange import main
with patch("GetRange.return_results") as return_results:
mocker.patch.object(demisto, "args", return_value={"range": indexes, "value": list(range(10))})
main()
assert return_results.call_args.kwargs.get("results") == expected_results