mirror of
https://github.com/rangermix/TwitchDropsMiner.git
synced 2026-05-26 07:08:04 +00:00
c9fefed1a9
- Implemented 'Benefit Type' filtering (Item, Badge, etc.) in backend and settings UI. - Redesigned 'Wanted Drops Queue' as a card-based, masonry layout. - Added 'Game Groups' to Wanted Queue for better organization. - Added variable height support for Inventory cards. - Added proxy support and verification logic (from previous steps). - Added unit tests for new filters and UI logic.
135 lines
4.5 KiB
Python
135 lines
4.5 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock, Mock
|
|
from src.web.gui_manager import WebGUIManager
|
|
from src.core.client import Twitch, State
|
|
from src.models.campaign import DropsCampaign
|
|
from src.models.game import Game
|
|
from src.models.drop import TimedDrop
|
|
from src.models.benefit import Benefit, BenefitType
|
|
|
|
class TestWantedItems(unittest.TestCase):
|
|
def setUp(self):
|
|
# Mock Twitch Client
|
|
self.twitch = MagicMock(spec=Twitch)
|
|
self.twitch.settings = MagicMock()
|
|
self.twitch.state_change.return_value = lambda: None
|
|
|
|
# Mock dependencies created in __init__
|
|
# We can't easily mock internal creation of sub-managers without patching,
|
|
# but for get_wanted_tree we only need self._twitch.settings and self._twitch.inventory
|
|
|
|
# However, WebGUIManager __init__ calls self.twitch.state_change
|
|
|
|
self.gui = WebGUIManager(self.twitch)
|
|
# Suppress broadcaster
|
|
self.gui._broadcaster = MagicMock()
|
|
|
|
def test_get_wanted_tree(self):
|
|
# Setup Settings
|
|
self.twitch.settings.games_to_watch = ["Game1", "Game2"]
|
|
self.twitch.settings.mining_benefits = {"BADGE": True, "DIRECT_ENTITLEMENT": False}
|
|
|
|
# Setup Inventory
|
|
|
|
# Campaign 1: Game1, Drop with BADGE (Wanted)
|
|
c1 = MagicMock(spec=DropsCampaign)
|
|
c1.id = "c1_id"
|
|
c1.name = "Campaign1"
|
|
c1.campaign_url = "http://url1"
|
|
c1.game = Game({"id": 1, "name": "Game1", "boxArtURL": "http://img1"})
|
|
|
|
d1 = MagicMock(spec=TimedDrop)
|
|
d1.name = "Drop1"
|
|
d1.is_claimed = False
|
|
b1 = MagicMock(spec=Benefit)
|
|
b1.name = "Badge1"
|
|
b1.type = BenefitType.BADGE
|
|
b1.is_wanted.return_value = True
|
|
d1.benefits = [b1]
|
|
c1.drops = [d1]
|
|
|
|
# Campaign 2: Game2, Drop with DIRECT_ENTITLEMENT (Unwanted)
|
|
c2 = MagicMock(spec=DropsCampaign)
|
|
c2.id = "c2_id"
|
|
c2.name = "Campaign2"
|
|
c2.campaign_url = "http://url2"
|
|
c2.game = Game({"id": 2, "name": "Game2", "boxArtURL": "http://img2"})
|
|
|
|
d2 = MagicMock(spec=TimedDrop)
|
|
d2.name = "Drop2"
|
|
d2.is_claimed = False
|
|
b2 = MagicMock(spec=Benefit)
|
|
b2.name = "Item1"
|
|
b2.type = BenefitType.DIRECT_ENTITLEMENT
|
|
b2.is_wanted.return_value = False
|
|
d2.benefits = [b2]
|
|
c2.drops = [d2]
|
|
|
|
# Campaign 3: Game3 (Not in watch list), Drop with BADGE (Wanted but wrong game)
|
|
c3 = MagicMock(spec=DropsCampaign)
|
|
c3.id = "c3_id"
|
|
c3.name = "Campaign3"
|
|
c3.campaign_url = "http://url3"
|
|
c3.game = Game({"id": 3, "name": "Game3", "boxArtURL": "http://img3"})
|
|
|
|
d3 = MagicMock(spec=TimedDrop)
|
|
d3.name = "Drop3"
|
|
d3.is_claimed = False
|
|
b3 = MagicMock(spec=Benefit)
|
|
b3.name = "Badge2"
|
|
b3.type = BenefitType.BADGE
|
|
b3.is_wanted.return_value = True
|
|
d3.benefits = [b3]
|
|
c3.drops = [d3]
|
|
|
|
self.twitch.inventory = [c1, c2, c3]
|
|
|
|
# Execute
|
|
result = self.gui.get_wanted_tree()
|
|
|
|
# Verify
|
|
# Expected: Game1 only
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(result[0]["game_name"], "Game1")
|
|
self.assertEqual(result[0]["game_icon"], "http://img1")
|
|
|
|
campaigns = result[0]["campaigns"]
|
|
self.assertEqual(len(campaigns), 1)
|
|
self.assertEqual(campaigns[0]["name"], "Campaign1")
|
|
self.assertEqual(len(campaigns[0]["drops"]), 1)
|
|
self.assertEqual(campaigns[0]["drops"][0]["name"], "Drop1")
|
|
self.assertEqual(campaigns[0]["drops"][0]["benefits"], ["Badge1"])
|
|
|
|
def test_get_wanted_tree_claimed_filtering(self):
|
|
# Setup Settings
|
|
self.twitch.settings.games_to_watch = ["Game1"]
|
|
self.twitch.settings.mining_benefits = {"BADGE": True}
|
|
|
|
# Setup Inventory
|
|
# Drop is claimed -> Should be hidden
|
|
c1 = MagicMock(spec=DropsCampaign)
|
|
c1.id = "c1_id"
|
|
c1.name = "Campaign1"
|
|
c1.campaign_url = "http://url1"
|
|
c1.game = Game({"id": 1, "name": "Game1", "boxArtURL": "http://img1"})
|
|
|
|
d1 = MagicMock(spec=TimedDrop)
|
|
d1.name = "Drop1"
|
|
d1.is_claimed = True
|
|
b1 = MagicMock(spec=Benefit)
|
|
b1.name = "Badge1"
|
|
b1.type = BenefitType.BADGE
|
|
d1.benefits = [b1]
|
|
c1.drops = [d1]
|
|
|
|
self.twitch.inventory = [c1]
|
|
|
|
# Execute
|
|
result = self.gui.get_wanted_tree()
|
|
|
|
# Verify
|
|
self.assertEqual(len(result), 0)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|