core/tests/components/cloud/test_utils.py

"""Test aiohttp request helper."""
from aiohttp import web

from homeassistant.components.cloud import utils


def test_serialize_text():
    """Test serializing a text response."""
    response = web.Response(status=201, text="Hello")
    assert utils.aiohttp_serialize_response(response) == {
        "status": 201,
        "body": "Hello",
        "headers": {"Content-Type": "text/plain; charset=utf-8"},
    }


def test_serialize_body_str():
    """Test serializing a response with a str as body."""
    response = web.Response(status=201, body="Hello")
    assert utils.aiohttp_serialize_response(response) == {
        "status": 201,
        "body": "Hello",
        "headers": {"Content-Length": "5", "Content-Type": "text/plain; charset=utf-8"},
    }


def test_serialize_body_None():
    """Test serializing a response with a str as body."""
    response = web.Response(status=201, body=None)
    assert utils.aiohttp_serialize_response(response) == {
        "status": 201,
        "body": None,
        "headers": {},
    }


def test_serialize_body_bytes():
    """Test serializing a response with a str as body."""
    response = web.Response(status=201, body=b"Hello")
    assert utils.aiohttp_serialize_response(response) == {
        "status": 201,
        "body": "Hello",
        "headers": {},
    }


def test_serialize_json():
    """Test serializing a JSON response."""
    response = web.json_response({"how": "what"})
    assert utils.aiohttp_serialize_response(response) == {
        "status": 200,
        "body": '{"how": "what"}',
        "headers": {"Content-Type": "application/json; charset=utf-8"},
    }