aa-killtracker 0.17.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

killtracker/tests/models/test_webhook.py

@@ -1,150 +0,0 @@
-import json
-
-import dhooks_lite
-import requests_mock
-
-from django.core.cache import cache
-from django.test import TestCase
-
-from app_utils.json import JSONDateTimeDecoder
-
-from killtracker.exceptions import WebhookTooManyRequests
-from killtracker.models import Webhook
-from killtracker.tests.testdata.helpers import LoadTestDataMixin
-
-
-class TestWebhookQueue(LoadTestDataMixin, TestCase):
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass()
-
-    def setUp(self) -> None:
-        self.webhook_1.main_queue.clear()
-        self.webhook_1.error_queue.clear()
-
-    def test_reset_failed_messages(self):
-        message = "Test message"
-        self.webhook_1.error_queue.enqueue(message)
-        self.webhook_1.error_queue.enqueue(message)
-        self.assertEqual(self.webhook_1.error_queue.size(), 2)
-        self.assertEqual(self.webhook_1.main_queue.size(), 0)
-        self.webhook_1.reset_failed_messages()
-        self.assertEqual(self.webhook_1.error_queue.size(), 0)
-        self.assertEqual(self.webhook_1.main_queue.size(), 2)
-
-    def test_discord_message_asjson_normal(self):
-        embed = dhooks_lite.Embed(description="my_description")
-        result = Webhook._discord_message_asjson(
-            content="my_content",
-            username="my_username",
-            avatar_url="my_avatar_url",
-            embeds=[embed],
-        )
-        message_python = json.loads(result, cls=JSONDateTimeDecoder)
-        expected = {
-            "content": "my_content",
-            "embeds": [{"description": "my_description", "type": "rich"}],
-            "username": "my_username",
-            "avatar_url": "my_avatar_url",
-        }
-        self.assertDictEqual(message_python, expected)
-
-    def test_discord_message_asjson_empty(self):
-        with self.assertRaises(ValueError):
-            Webhook._discord_message_asjson("")
-
-
-@requests_mock.Mocker()
-class TestWebhookSendMessage(LoadTestDataMixin, TestCase):
-    def setUp(self) -> None:
-        self.message = Webhook._discord_message_asjson(content="Test message")
-        cache.clear()
-
-    def test_when_send_ok_returns_true(self, requests_mocker):
-        # given
-        requests_mocker.register_uri(
-            "POST",
-            self.webhook_1.url,
-            status_code=200,
-            json={
-                "name": "test webhook",
-                "type": 1,
-                "channel_id": "199737254929760256",
-                "token": "3d89bb7572e0fb30d8128367b3b1b44fecd1726de135cbe28a41f8b2f777c372ba2939e72279b94526ff5d1bd4358d65cf11",
-                "avatar": None,
-                "guild_id": "199737254929760256",
-                "id": "223704706495545344",
-                "application_id": None,
-                "user": {
-                    "username": "test",
-                    "discriminator": "7479",
-                    "id": "190320984123768832",
-                    "avatar": "b004ec1740a63ca06ae2e14c5cee11f3",
-                    "public_flags": 131328,
-                },
-            },
-        )
-        # when
-        response = self.webhook_1.send_message_to_webhook(self.message)
-        # then
-        self.assertTrue(response.status_ok)
-        self.assertTrue(requests_mocker.called)
-
-    def test_when_send_not_ok_returns_false(self, requests_mocker):
-        # given
-        requests_mocker.register_uri("POST", self.webhook_1.url, status_code=404)
-        # when
-        response = self.webhook_1.send_message_to_webhook(self.message)
-        # then
-        self.assertFalse(response.status_ok)
-        self.assertTrue(requests_mocker.called)
-
-    def test_too_many_requests_normal(self, requests_mocker):
-        # given
-        requests_mocker.register_uri(
-            "POST",
-            self.webhook_1.url,
-            status_code=429,
-            json={
-                "global": False,
-                "message": "You are being rate limited.",
-                "retry_after": 2000,
-            },
-            headers={
-                "x-ratelimit-remaining": "5",
-                "x-ratelimit-reset-after": "60",
-                "Retry-After": "2000",
-            },
-        )
-        # when/then
-        try:
-            self.webhook_1.send_message_to_webhook(self.message)
-        except Exception as ex:
-            self.assertIsInstance(ex, WebhookTooManyRequests)
-            self.assertEqual(ex.retry_after, 2002)
-        else:
-            self.fail("Did not raise excepted exception")
-
-        self.assertAlmostEqual(
-            cache.ttl(self.webhook_1._blocked_cache_key()), 2002, delta=5
-        )
-
-    def test_too_many_requests_no_retry_value(self, requests_mocker):
-        # given
-        requests_mocker.register_uri(
-            "POST",
-            self.webhook_1.url,
-            status_code=429,
-            headers={
-                "x-ratelimit-remaining": "5",
-                "x-ratelimit-reset-after": "60",
-            },
-        )
-        # when/then
-        try:
-            self.webhook_1.send_message_to_webhook(self.message)
-        except Exception as ex:
-            self.assertIsInstance(ex, WebhookTooManyRequests)
-            self.assertEqual(ex.retry_after, WebhookTooManyRequests.DEFAULT_RESET_AFTER)
-        else:
-            self.fail("Did not raise excepted exception")

killtracker/tests/test_exceptions.py

@@ -1,12 +0,0 @@
-from django.test import TestCase
-
-from killtracker.exceptions import WebhookTooManyRequests
-
-
-class TestExceptions(TestCase):
-    def test_exception(self):
-        ex = WebhookTooManyRequests(10)
-        self.assertEqual(ex.retry_after, 10)
-
-        ex = WebhookTooManyRequests()
-        self.assertEqual(ex.retry_after, 600)