def test_get_teams_going_to_right_function(): """ Given: - An app client object When: - Calling function get_teams Case A: "team_id" = 1234 Case B: No arguments Then: - Ensure the right function was called Case A: Called get_team Case B: Called list_teams """ mock_client = OpsGenieV3.Client(base_url="") mock_client.get_team = mock.MagicMock() OpsGenieV3.get_teams(mock_client, {"team_id": 1234}) assert mock_client.get_team.called mock_client.list_teams = mock.MagicMock() OpsGenieV3.get_teams(mock_client, {}) assert mock_client.list_teams.called
def test_get_teams(mocker): """ Given: - An app client object When: - Calling function get_teams Then: - Ensure the return data is correct """ mock_client = OpsGenieV3.Client(base_url="") mocker.patch.object( mock_client, 'list_teams', return_value=util_load_json('test_data/get_teams.json')) res = OpsGenieV3.get_teams(mock_client, {}) assert len(res.outputs) == 2