108 lines
5.1 KiB
Python
108 lines
5.1 KiB
Python
import copy
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
from automation.pages import PageMap
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
|
|
class PageTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.page_map = PageMap(ROOT)
|
|
|
|
def frame(self, *page_ids):
|
|
image = Image.new('RGB', (562, 1002), (35, 50, 70))
|
|
for page_id in page_ids:
|
|
for spec in self.page_map.pages[page_id]['recognition'][0]:
|
|
with Image.open(ROOT / 'assets' / spec['template']) as template:
|
|
x, y = spec['search'][:2]
|
|
image.paste(template.convert('RGB'), (x + 4, y + 4))
|
|
return np.asarray(image)
|
|
|
|
def test_primary_pages_are_distinct(self):
|
|
for page_id in ('home', 'shop', 'heroes', 'king', 'territory', 'guild'):
|
|
with self.subTest(page=page_id):
|
|
result = self.page_map.identify(self.frame(page_id))
|
|
self.assertEqual(result['status'], 'identified')
|
|
self.assertEqual(result['page']['id'], page_id)
|
|
|
|
def test_white_or_darkened_frame_is_not_treated_as_home(self):
|
|
self.assertEqual(self.page_map.identify(np.full((1002, 562, 3), 255, dtype=np.uint8))['status'], 'unknown')
|
|
dark = (self.frame('home').astype(float) * .35).astype(np.uint8)
|
|
self.assertEqual(self.page_map.identify(dark)['status'], 'unknown')
|
|
|
|
def test_overlay_has_priority_and_does_not_guess_its_parent(self):
|
|
result = self.page_map.identify(self.frame('shop', 'reward'))
|
|
self.assertEqual(result['page']['id'], 'reward')
|
|
self.assertIsNone(self.page_map.pages['reward']['parent'])
|
|
|
|
def test_duplicate_identity_reports_ambiguity(self):
|
|
duplicate = copy.deepcopy(self.page_map.pages['territory'])
|
|
duplicate['id'] = 'another_territory'
|
|
self.page_map.pages[duplicate['id']] = duplicate
|
|
result = self.page_map.identify(self.frame('territory'))
|
|
self.assertEqual(result['status'], 'ambiguous')
|
|
self.assertIsNone(result['page'])
|
|
|
|
def test_specific_tab_supersedes_matching_overlay_and_background(self):
|
|
container = copy.deepcopy(self.page_map.pages['reward'])
|
|
container['id'] = 'test.container'
|
|
self.page_map.pages[container['id']] = container
|
|
# Keep the generic reward out of this fixture: it has no known parent.
|
|
self.page_map.pages['reward']['recognition'] = []
|
|
tab = copy.deepcopy(container)
|
|
tab.update(id='test.tab', parent='test.container', kind='tab')
|
|
self.page_map.pages[tab['id']] = tab
|
|
result = self.page_map.identify(self.frame('shop', 'test.tab'))
|
|
self.assertEqual(result['page']['id'], 'test.tab')
|
|
self.assertEqual(result['status'], 'identified')
|
|
|
|
def test_territory_to_recruit_route_includes_home_and_shop(self):
|
|
route = self.page_map.route('territory', 'recruit.lamp')
|
|
self.assertEqual([edge['to'] for edge in route], ['home', 'shop', 'recruit.lamp'])
|
|
self.assertIn('condition', route[-1])
|
|
|
|
def test_unverified_paths_are_excluded(self):
|
|
with self.assertRaises(ValueError):
|
|
self.page_map.route('home', 'guild.boss')
|
|
with self.assertRaises(ValueError):
|
|
self.page_map.route('reward', 'home')
|
|
|
|
def test_shared_progress_popup_requires_known_origin(self):
|
|
frame = self.frame('planet.progress')
|
|
self.assertEqual(self.page_map.identify(frame)['status'], 'ambiguous')
|
|
self.assertEqual(self.page_map.identify(frame, previous_page='territory.rank.board')['page']['id'],
|
|
'territory.rank.progress')
|
|
self.assertEqual(self.page_map.identify(frame, previous_page='planet.rank')['page']['id'],
|
|
'planet.progress')
|
|
self.assertEqual(self.page_map.identify(frame, previous_page='home')['status'], 'ambiguous')
|
|
|
|
def test_mode_selection_uses_its_distinct_card_label(self):
|
|
for page_id in ('coop.selection', 'arena.selection'):
|
|
with self.subTest(page=page_id):
|
|
result = self.page_map.identify(self.frame(page_id))
|
|
self.assertEqual(result['page']['id'], page_id)
|
|
|
|
def test_pet_weekly_challenge_is_not_home_weekly_tasks(self):
|
|
# Actual audit pixels retain the similar task title that previously
|
|
# matched home weekly tasks; the selected navigation label differs.
|
|
frame = np.asarray(Image.open(ROOT / 'tests/fixtures/pet_challenge_identity.png').convert('RGB'))
|
|
result = self.page_map.identify(frame)
|
|
self.assertEqual(result['page']['id'], 'coop.pet.challenge.weekly')
|
|
self.assertNotIn('tasks.weekly', [p['id'] for p in result['candidates']])
|
|
|
|
def test_history_does_not_resolve_unrelated_matches(self):
|
|
duplicate = copy.deepcopy(self.page_map.pages['territory'])
|
|
duplicate.update(id='test.unknown_territory', parent='home')
|
|
self.page_map.pages[duplicate['id']] = duplicate
|
|
result = self.page_map.identify(self.frame('territory'), previous_page='home')
|
|
self.assertEqual(result['status'], 'ambiguous')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|