60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
# Copyright (C) 2020 by Cathy Hu <cathy.hu@fau.de>
|
|
# Copyright (C) 2020 by Martin Rey <martin.rey@mailbox.org>
|
|
#
|
|
# SPDX-License-Identifier: 0BSD
|
|
|
|
from fastapi import status
|
|
from ormantic.exceptions import NoMatch
|
|
import pytest
|
|
|
|
from kibicara.platforms.telegram.model import Telegram, TelegramUser
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"bot", [{"api_token": "apitoken123", "welcome_message": "msg"}]
|
|
)
|
|
@pytest.mark.anyio
|
|
async def test_telegram_delete_bot(asyncclient, bot, telegram, auth_header):
|
|
await TelegramUser.objects.create(user_id=1234, bot=telegram.id)
|
|
await TelegramUser.objects.create(user_id=5678, bot=telegram.id)
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/{0}/telegram/{1}".format(telegram.hood.id, telegram.id),
|
|
headers=auth_header,
|
|
)
|
|
assert response.status_code == status.HTTP_204_NO_CONTENT
|
|
with pytest.raises(NoMatch):
|
|
await Telegram.objects.get(id=telegram.id)
|
|
with pytest.raises(NoMatch):
|
|
await TelegramUser.objects.get(id=telegram.id)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_telegram_delete_bot_invalid_id(asyncclient, auth_header, hood_id):
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/1337/telegram/123", headers=auth_header
|
|
)
|
|
assert response.status_code == status.HTTP_404_NOT_FOUND
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/wrong/telegram/123", headers=auth_header
|
|
)
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/{0}/telegram/7331".format(hood_id), headers=auth_header
|
|
)
|
|
assert response.status_code == status.HTTP_404_NOT_FOUND
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/{0}/telegram/wrong".format(hood_id), headers=auth_header
|
|
)
|
|
assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"bot", [{"api_token": "apitoken123", "welcome_message": "msg"}]
|
|
)
|
|
@pytest.mark.anyio
|
|
async def test_telegram_delete_bot_unauthorized(asyncclient, bot, telegram):
|
|
response = await asyncclient.delete(
|
|
"/api/hoods/{0}/telegram/{1}".format(telegram.hood.id, telegram.id)
|
|
)
|
|
assert response.status_code == status.HTTP_401_UNAUTHORIZED
|