from unittest.mock import MagicMock

import pytest

from falcon import MEDIA_TEXT, ResponseOptions

from _util import create_resp  # NOQA


@pytest.fixture(params=[True, False])
def resp(request):
    return create_resp(asgi=request.param)


def test_response_set_content_type_set(resp):
    resp._set_media_type(MEDIA_TEXT)
    assert resp._headers['content-type'] == MEDIA_TEXT


def test_response_set_content_type_not_set(resp):
    assert 'content-type' not in resp._headers

    resp._set_media_type()
    assert 'content-type' not in resp._headers


def test_response_get_headers(resp):
    resp.append_header('x-things1', 'thing-1')
    resp.append_header('x-things2', 'thing-2')
    resp.append_header('X-Things3', 'Thing-3')

    resp.set_cookie('Chocolate', 'Chip')

    headers = resp.headers
    assert headers['x-things1'] == 'thing-1'
    assert headers['x-things2'] == 'thing-2'
    assert headers['x-things3'] == 'Thing-3'

    assert 'set-cookie' not in headers


def test_response_attempt_to_set_read_only_headers(resp):
    resp.append_header('x-things1', 'thing-1')
    resp.append_header('x-things2', 'thing-2')
    resp.append_header('x-things3', 'thing-3a')
    resp.append_header('X-Things3', 'thing-3b')

    with pytest.raises(AttributeError):
        resp.headers = {'x-things4': 'thing-4'}

    headers = resp.headers
    assert headers['x-things1'] == 'thing-1'
    assert headers['x-things2'] == 'thing-2'
    assert headers['x-things3'] == 'thing-3a, thing-3b'


def test_response_removed_stream_len(resp):
    with pytest.raises(AttributeError):
        resp.stream_len = 128

    with pytest.raises(AttributeError):
        resp.stream_len


def test_response_option_mimetype_init(monkeypatch):
    mock = MagicMock()
    mock.inited = False
    monkeypatch.setattr('falcon.response.mimetypes', mock)

    ro = ResponseOptions()

    assert ro.static_media_types is mock.types_map
    mock.reset_mock()
    mock.inited = True
    ro = ResponseOptions()
    assert ro.static_media_types is mock.types_map
    mock.init.assert_not_called()