From f128a72fd6c2b5a24822faaca464c71684a30b26 Mon Sep 17 00:00:00 2001 From: Deko Date: Sat, 4 Mar 2023 11:12:37 +0100 Subject: [PATCH] Add a test for main --- app/tests/test_main.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 app/tests/test_main.py diff --git a/app/tests/test_main.py b/app/tests/test_main.py new file mode 100644 index 0000000..05f1953 --- /dev/null +++ b/app/tests/test_main.py @@ -0,0 +1,36 @@ +import os +from unittest import mock + +import requests_mock + +from app.main import Main + + +def test_create_main(mock_loggers): + with ( + mock.patch.dict( + os.environ, + { + "STREAMER_NAME": "streamer_name", + "DISCORD_WEBHOOK_URL": "URL", + "TWITCH_CLIENT_ID": "id", + "TWITCH_CLIENT_SECRET": "secret", + }, + ), + requests_mock.Mocker() as requests_mocker, + ): + # We authorize + requests_mocker.post( + "https://id.twitch.tv/oauth2/token", json={"access_token": "token"} + ) + # We cache the profile image URL of the streamer + requests_mocker.get( + "https://api.twitch.tv/helix/users?login=streamer_name", + json={"data": [{"profile_image_url": "image"}]}, + ) + + main = Main() + + assert main.discord_client + assert main.twitch_client + assert main.profile_image == "image"