# Copyright (C) 2020 by Cathy Hu # Copyright (C) 2020 by Martin Rey # # SPDX-License-Identifier: 0BSD from fastapi import status from pytest import fixture, mark from kibicara.platforms import telegram from kibicara.platforms.telegram.model import Telegram @fixture(scope="function") def disable_spawner(monkeypatch): class DoNothing: def start(self, bot): assert bot is not None monkeypatch.setattr(telegram.webapi, "spawner", DoNothing()) @mark.parametrize("body", [{"api_token": "string", "welcome_message": "string"}]) def test_telegram_create_bot( event_loop, client, disable_spawner, hood_id, auth_header, monkeypatch, body, ): def check_token_mock(token): return True monkeypatch.setattr(telegram.webapi, "check_token", check_token_mock) response = client.post( "/api/hoods/{0}/telegram/".format(hood_id), json=body, headers=auth_header, ) assert response.status_code == status.HTTP_201_CREATED bot_id = response.json()["id"] telegram_obj = event_loop.run_until_complete(Telegram.objects.get(id=bot_id)) assert response.json()["api_token"] == body["api_token"] == telegram_obj.api_token assert ( response.json()["welcome_message"] == body["welcome_message"] == telegram_obj.welcome_message ) assert response.json()["hood"]["id"] == telegram_obj.hood.id assert telegram_obj.enabled @mark.parametrize("body", [{"api_token": "string", "welcome_message": "string"}]) def test_telegram_invalid_api_token( event_loop, client, disable_spawner, hood_id, auth_header, monkeypatch, body, ): response = client.post( "/api/hoods/{0}/telegram/".format(hood_id), json=body, headers=auth_header, ) assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY def test_telegram_create_telegram_invalid_id(client, auth_header): response = client.post("/api/hoods/1337/telegram/", headers=auth_header) assert response.status_code == status.HTTP_404_NOT_FOUND response = client.post("/api/hoods/wrong/telegram/", headers=auth_header) assert response.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY def test_telegram_create_unauthorized(client, hood_id): response = client.post("/api/hoods/{hood_id}/telegram/") assert response.status_code == status.HTTP_401_UNAUTHORIZED