import demistomock as demisto
from SetRDPOverallScore import main
def test_happy_path_score_zero(mocker):
"""
Given:
- OverallScore zero
When:
- Calling the main function
Then:
- Ensure that the expected HTML is returned
"""
mocker.patch.object(demisto, "results", return_value={"OverallScore": "0"})
expected_html = (
"0/100
"
"No suspicious strings found
"
)
main()
result = demisto.results.call_args[0][0]["Contents"]
assert result == expected_html
def test_happy_path_score_50(mocker):
"""
Given:
- OverallScore 50
When:
- Calling the main function
Then:
- Ensure that the expected HTML is returned as expected
"""
mocker.patch.object(demisto, "results")
mocker.patch.object(demisto, "context", return_value={"OverallScore": "50"})
expected_html = "50/100
"
main()
result = demisto.results.call_args[0][0]
assert result["Contents"] == expected_html
def test_happy_path_score_90(mocker):
"""
Given:
- OverallScore 50
When:
- Calling the main function
Then:
- Ensure that the expected HTML is returned
"""
mocker.patch.object(demisto, "results")
mocker.patch.object(demisto, "context", return_value={"OverallScore": "90"})
expected_html = "90/100
"
main()
result = demisto.results.call_args[0][0]
assert result["Contents"] == expected_html
def test_happy_path_score_100(mocker):
"""
Given:
- OverallScore 100
When:
- Calling the main function
Then:
- Ensure that the expected HTML is returned
"""
mocker.patch.object(demisto, "results")
mocker.patch.object(demisto, "context", return_value={"OverallScore": "100"})
expected_html = "100/100
"
main()
result = demisto.results.call_args[0][0]
assert result["Contents"] == expected_html