diff --git a/tests/shitpost.py b/tests/shitpost.py index 0d560e7..a142f76 100644 --- a/tests/shitpost.py +++ b/tests/shitpost.py @@ -3,6 +3,7 @@ import dotenv import os import discord import requests +from datetime import datetime from discord.ext import commands import Shitpost @@ -169,3 +170,13 @@ class TestShitpost(unittest.TestCase): message = self.cog.zalgofy(inmsg) self.assertIsInstance(message, str) self.assertEqual(len(message), len(inmsg)*5) + + def test_zaagmans(self): + zaag = self.cog.get_zaagmans() + if zaag is None: + self.skipTest(f"Server at iszaagmansallangsgeweest.nl has an error, or is unavailable") + else: + vandaag = datetime.now() + dag = vandaag.weekday() + truth = True if dag > 2 or (dag == 2 and vandaag.hour >= 12) else False + self.assertEqual(truth, zaag)