示例#1
0
def test_get_alerts_going_to_right_function():
    """
    Given:
        - An app client object
    When:
        - Calling function get_alerts
        Case A: "alert-id" = 1234
        Case B: No arguments
    Then:
        - Ensure the right function was called
        Case A: Called get_alert
        Case B: Called list_alerts
    """
    mock_client = OpsGenieV3.Client(base_url="")
    mock_client.get_alert = mock.MagicMock()
    OpsGenieV3.get_alerts(mock_client, {"alert-id": 1234})
    assert mock_client.get_alert.called
    OpsGenieV3.list_alerts = mock.MagicMock()
    OpsGenieV3.get_alerts(mock_client, {})
    assert OpsGenieV3.list_alerts.called
示例#2
0
def test_get_alerts(mocker):
    """
    Given:
        - An app client object
        - Limit = 1
    When:
        - Calling function list_alerts
    Then:
        - Ensure the return data is correct
    """
    mocker.patch('CommonServerPython.get_demisto_version',
                 return_value={"version": "6.2.0"})
    mock_client = OpsGenieV3.Client(base_url="")
    mocker.patch.object(
        mock_client,
        'list_alerts',
        return_value=util_load_json('test_data/get_alerts.json'))
    res = OpsGenieV3.get_alerts(mock_client, {"limit": 1})
    assert (len(res.outputs) == 1)