1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-26 14:08:21 +00:00

Use http.HTTPStatus in components/m* (#58251)

This commit is contained in:
Ville Skyttä
2021-10-23 00:54:36 +03:00
committed by GitHub
parent b49b975999
commit 77120a5137
16 changed files with 119 additions and 112 deletions

View File

@@ -1,11 +1,11 @@
"""The tests for the mailbox component."""
from hashlib import sha1
from http import HTTPStatus
import pytest
from homeassistant.bootstrap import async_setup_component
import homeassistant.components.mailbox as mailbox
from homeassistant.const import HTTP_INTERNAL_SERVER_ERROR, HTTP_NOT_FOUND
@pytest.fixture
@@ -21,7 +21,7 @@ async def test_get_platforms_from_mailbox(mock_http_client):
url = "/api/mailbox/platforms"
req = await mock_http_client.get(url)
assert req.status == 200
assert req.status == HTTPStatus.OK
result = await req.json()
assert len(result) == 1
assert result[0].get("name") == "DemoMailbox"
@@ -32,7 +32,7 @@ async def test_get_messages_from_mailbox(mock_http_client):
url = "/api/mailbox/messages/DemoMailbox"
req = await mock_http_client.get(url)
assert req.status == 200
assert req.status == HTTPStatus.OK
result = await req.json()
assert len(result) == 10
@@ -45,7 +45,7 @@ async def test_get_media_from_mailbox(mock_http_client):
url = f"/api/mailbox/media/DemoMailbox/{msgsha}"
req = await mock_http_client.get(url)
assert req.status == 200
assert req.status == HTTPStatus.OK
data = await req.read()
assert sha1(data).hexdigest() == mp3sha
@@ -60,11 +60,11 @@ async def test_delete_from_mailbox(mock_http_client):
for msg in [msgsha1, msgsha2]:
url = f"/api/mailbox/delete/DemoMailbox/{msg}"
req = await mock_http_client.delete(url)
assert req.status == 200
assert req.status == HTTPStatus.OK
url = "/api/mailbox/messages/DemoMailbox"
req = await mock_http_client.get(url)
assert req.status == 200
assert req.status == HTTPStatus.OK
result = await req.json()
assert len(result) == 8
@@ -74,7 +74,7 @@ async def test_get_messages_from_invalid_mailbox(mock_http_client):
url = "/api/mailbox/messages/mailbox.invalid_mailbox"
req = await mock_http_client.get(url)
assert req.status == HTTP_NOT_FOUND
assert req.status == HTTPStatus.NOT_FOUND
async def test_get_media_from_invalid_mailbox(mock_http_client):
@@ -83,7 +83,7 @@ async def test_get_media_from_invalid_mailbox(mock_http_client):
url = f"/api/mailbox/media/mailbox.invalid_mailbox/{msgsha}"
req = await mock_http_client.get(url)
assert req.status == HTTP_NOT_FOUND
assert req.status == HTTPStatus.NOT_FOUND
async def test_get_media_from_invalid_msgid(mock_http_client):
@@ -92,7 +92,7 @@ async def test_get_media_from_invalid_msgid(mock_http_client):
url = f"/api/mailbox/media/DemoMailbox/{msgsha}"
req = await mock_http_client.get(url)
assert req.status == HTTP_INTERNAL_SERVER_ERROR
assert req.status == HTTPStatus.INTERNAL_SERVER_ERROR
async def test_delete_from_invalid_mailbox(mock_http_client):
@@ -101,4 +101,4 @@ async def test_delete_from_invalid_mailbox(mock_http_client):
url = f"/api/mailbox/delete/mailbox.invalid_mailbox/{msgsha}"
req = await mock_http_client.delete(url)
assert req.status == HTTP_NOT_FOUND
assert req.status == HTTPStatus.NOT_FOUND