diff --git a/discord_ui/scout_view.py b/discord_ui/scout_view.py index a5a64ba..7dcecff 100644 --- a/discord_ui/scout_view.py +++ b/discord_ui/scout_view.py @@ -11,7 +11,7 @@ import logging import discord -from api_calls import db_post +from api_calls import db_get, db_post from helpers.main import get_team_by_owner, get_card_embeds from helpers.scouting import ( SCOUT_TOKENS_PER_DAY, @@ -207,12 +207,14 @@ class ScoutButton(discord.ui.Button): ) # Consume a scout token + current = await db_get("current") await db_post( "rewards", payload={ "name": "Scout Token", "team_id": scouter_team["id"], "season": PD_SEASON, + "week": current["week"], "created": int_timestamp(), }, ) diff --git a/tests/scouting/test_scout_view.py b/tests/scouting/test_scout_view.py index 186c0b1..0a2532e 100644 --- a/tests/scouting/test_scout_view.py +++ b/tests/scouting/test_scout_view.py @@ -210,12 +210,14 @@ class TestScoutButtonSuccess: @pytest.mark.asyncio @patch("discord_ui.scout_view.get_card_embeds", new_callable=AsyncMock) @patch("discord_ui.scout_view.db_post", new_callable=AsyncMock) + @patch("discord_ui.scout_view.db_get", new_callable=AsyncMock) @patch("discord_ui.scout_view.get_scout_tokens_used", new_callable=AsyncMock) @patch("discord_ui.scout_view.get_team_by_owner", new_callable=AsyncMock) async def test_successful_scout_creates_card_copy( self, mock_get_team, mock_get_tokens, + mock_db_get, mock_db_post, mock_card_embeds, sample_cards,