Add more tests

didier/data/embeds/wordle.py

@@ -105,7 +105,7 @@ class WordleEmbed(EmbedBaseModel):
         rows = [" ".join(row) for row in emojis]
 
         # Don't reveal anything if we only want to show the colours
-        if not only_colours:
+        if not only_colours and self.game is not None:
             for i, guess in enumerate(self.game.guesses):
                 rows[i] += f"   ||{guess.upper()}||"
 
@@ -126,7 +126,7 @@ class WordleErrorEmbed(EmbedBaseModel):
     message: str
 
     @overrides
-    def to_embed(self) -> discord.Embed:
+    def to_embed(self, **kwargs: dict) -> discord.Embed:
         embed = discord.Embed(colour=discord.Colour.red(), title="Wordle")
         embed.description = self.message
         embed.set_footer(text=footer())

didier/didier.py

@@ -27,7 +27,7 @@ class Didier(commands.Bot):
     error_channel: discord.abc.Messageable
     initial_extensions: tuple[str, ...] = ()
     http_session: ClientSession
-    wordle_words: set[str, ...] = set()
+    wordle_words: set[str] = set()
 
     def __init__(self):
         activity = discord.Activity(type=discord.ActivityType.playing, name=settings.DISCORD_STATUS_MESSAGE)

tests/test_database/test_crud/test_wordle.py

@@ -1,8 +1,12 @@
+from datetime import datetime, timedelta
+
 import pytest
+from freezegun import freeze_time
 
 from database.crud import wordle as crud
+from database.enums import TempStorageKey
 from database.mongo_types import MongoCollection, MongoDatabase
-from database.schemas.mongo import WordleGame
+from database.schemas.mongo import TemporaryStorage, WordleGame
 
 
 @pytest.fixture
@@ -37,6 +41,36 @@ async def test_get_active_wordle_game_none(mongodb: MongoDatabase, test_user_id:
 
 
 async def test_get_active_wordle_game(mongodb: MongoDatabase, wordle_game: WordleGame):
-    """Test getting an active game when there is none"""
+    """Test getting an active game when there is one"""
     result = await crud.get_active_wordle_game(mongodb, wordle_game.user_id)
     assert result.dict(by_alias=True) == wordle_game.dict(by_alias=True)
+
+
+async def test_get_daily_word_none(mongodb: MongoDatabase):
+    """Test getting the daily word when the database is empty"""
+    result = await crud.get_daily_word(mongodb)
+    assert result is None
+
+
+@freeze_time("2022-07-30")
+async def test_get_daily_word_not_today(mongodb: MongoDatabase):
+    """Test getting the daily word when there is an entry, but not for today"""
+    day = datetime.today() - timedelta(days=1)
+    collection = mongodb[TemporaryStorage.collection()]
+
+    word = "testword"
+    await collection.insert_one({"key": TempStorageKey.WORDLE_WORD, "day": day, "word": word})
+
+    assert await crud.get_daily_word(mongodb) is None
+
+
+@freeze_time("2022-07-30")
+async def test_get_daily_word_present(mongodb: MongoDatabase):
+    """Test getting the daily word when there is one for today"""
+    day = datetime.today()
+    collection = mongodb[TemporaryStorage.collection()]
+
+    word = "testword"
+    await collection.insert_one({"key": TempStorageKey.WORDLE_WORD, "day": day, "word": word})
+
+    assert await crud.get_daily_word(mongodb) == word