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')