# coding=utf-8
import json

import requests
import responses

from ibm_cloud_sdk_core import ApiException


@responses.activate
def test_api_exception():
    """Test APIException class"""
    responses.add(
        responses.GET,
        'https://test.com',
        status=500,
        body=json.dumps({'error': 'sorry', 'msg': 'serious error'}),
        content_type='application/json',
    )

    mock_response = requests.get('https://test.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception is not None
    assert exception.message == 'sorry'

    responses.add(
        responses.GET,
        'https://test-again.com',
        status=500,
        body=json.dumps(
            {
                "errors": [
                    {
                        "message": "sorry again",
                    }
                ],
            }
        ),
        content_type='application/json',
    )
    mock_response = requests.get('https://test-again.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception.message == 'sorry again'

    responses.add(
        responses.GET,
        'https://test-once-more.com',
        status=500,
        body=json.dumps({'message': 'sorry once more'}),
        content_type='application/json',
    )
    mock_response = requests.get('https://test-once-more.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception.message == 'sorry once more'

    responses.add(
        responses.GET,
        'https://test-msg.com',
        status=500,
        body=json.dumps({'msg': 'serious error'}),
        content_type='application/json',
    )
    mock_response = requests.get('https://test-msg.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception.message == 'Internal Server Error'

    responses.add(
        responses.GET,
        'https://test-errormessage.com',
        status=500,
        body=json.dumps({'errorMessage': 'IAM error message'}),
        content_type='application/json',
    )
    mock_response = requests.get('https://test-errormessage.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception.message == 'IAM error message'

    responses.add(
        responses.GET,
        'https://test-for-text.com',
        status=500,
        headers={'X-Global-Transaction-ID': 'xx'},
        body="plain text error",
    )
    mock_response = requests.get('https://test-for-text.com', timeout=None)
    exception = ApiException(500, http_response=mock_response)
    assert exception.message == 'plain text error'
    assert str(exception) == 'Error: plain text error, Status code: 500 , X-global-transaction-id: xx'