87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
import unittest
|
|
from pathlib import Path
|
|
|
|
from PIL import Image
|
|
|
|
from automation.engine import Matcher, Region, load_flow
|
|
from automation.tasks import TaskRunner
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
|
|
class TaskDesktop:
|
|
def __init__(self, images):
|
|
self.images = images
|
|
self.index = 0
|
|
self.clicks = []
|
|
|
|
def screenshot(self):
|
|
return self.images[min(self.index, len(self.images) - 1)]
|
|
|
|
def click(self, point, region, size, hold=0.2):
|
|
self.clicks.append((point, hold))
|
|
self.index += 1
|
|
|
|
|
|
class TaskTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.flow = load_flow(ROOT / 'workflows/task_claim.json')
|
|
self.matcher = Matcher(ROOT / 'assets')
|
|
self.region = Region(0, 0, 562, 1002)
|
|
|
|
def image(self, state, claim=False, dot=None):
|
|
image = Image.new('RGB', (562, 1002), (42, 58, 73))
|
|
if state in self.flow['states']:
|
|
for spec in self.flow['states'][state]:
|
|
template = Image.open(ROOT / 'assets' / spec['template']).convert('RGB')
|
|
x, y, _, _ = spec['search']
|
|
image.paste(template, (x + 5, y + 5))
|
|
if claim:
|
|
spec = self.flow['claim_button']
|
|
template = Image.open(ROOT / 'assets' / spec['template']).convert('RGB')
|
|
x, y, _, _ = spec['search']
|
|
image.paste(template, (x + 5, y + 5))
|
|
if dot == 'home':
|
|
spec = self.flow['task_dot'][0]
|
|
elif dot:
|
|
spec = self.flow['tab_dots'][dot]
|
|
else:
|
|
spec = None
|
|
if spec:
|
|
template = Image.open(ROOT / 'assets' / spec['template']).convert('RGB')
|
|
x, y, _, _ = spec['search']
|
|
image.paste(template, (x + 5, y + 5))
|
|
return image
|
|
|
|
def runner(self, images):
|
|
desktop = TaskDesktop(images)
|
|
return desktop, TaskRunner(desktop, self.matcher, self.flow,
|
|
lambda _: self.region, log=lambda _: None, timeout=0.8)
|
|
|
|
def test_claims_current_and_dotted_other_panel(self):
|
|
images = [
|
|
self.image('home', dot='home'),
|
|
self.image('daily', claim=True),
|
|
self.image('reward'),
|
|
self.image('daily', dot='weekly'),
|
|
self.image('weekly', claim=True),
|
|
self.image('reward'),
|
|
self.image('weekly'),
|
|
self.image('home'),
|
|
]
|
|
desktop, runner = self.runner(images)
|
|
self.assertEqual(runner.run(), 'claimed')
|
|
self.assertEqual(len(desktop.clicks), 7)
|
|
self.assertEqual(desktop.clicks[0][0], self.flow['clicks']['open_tasks'])
|
|
self.assertEqual(desktop.clicks[3][0], self.flow['clicks']['open_weekly'])
|
|
self.assertEqual(desktop.clicks[-1][0], self.flow['clicks']['close'])
|
|
|
|
def test_without_home_dot_does_not_open_tasks(self):
|
|
desktop, runner = self.runner([self.image('home')])
|
|
self.assertEqual(runner.run(), 'no_task_dot')
|
|
self.assertEqual(desktop.clicks, [])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|