def test_main_success(mocker):
"""
When main function is called, get_html_representation should be called.
"""
mocker.patch.object(demisto, 'args', return_value=INDICATOR_DATA)
mocker.patch.object(ChronicleIsolatedHostnameWidgetScript,
'get_html_representation',
return_value='')
ChronicleIsolatedHostnameWidgetScript.main()
assert ChronicleIsolatedHostnameWidgetScript.get_html_representation.called
def test_main_failure(mock_return_error, capfd, mocker):
"""
When main function gets some exception then valid message should be printed.
"""
mocker.patch.object(demisto, 'args', return_value=INDICATOR_DATA)
mocker.patch.object(ChronicleIsolatedHostnameWidgetScript,
'get_html_representation',
side_effect=Exception)
with capfd.disabled():
ChronicleIsolatedHostnameWidgetScript.main()
mock_return_error.assert_called_once_with('Could not load widget:\n')