~aleteoryx/muditaos

ref: 5a8ae75844e31970e5665257aaff8d9f558ab06a muditaos/test/harness/harness.py -rw-r--r-- 1.2 KiB
5a8ae758 — RobertPiet [EGD-4448] ssettings variable register fixed (#1045) 5 years ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Copyright (c) 2017-2020, Mudita Sp. z.o.o. All rights reserved.
# For licensing, see https://github.com/mudita/MuditaOS/LICENSE.md

from harness.interface import CDCSerial as serial
from harness.interface.defs import key_codes
from harness.utils import send_keystoke, application_keypath


class Harness:
    connection = None
    is_phone_unlocked = False

    def __init__(self, port='/dev/ttyACM2'):
        self.connection = serial.CDCSerial(port)

    def get_connection(self):
        return self.connection

    def unlock_phone(self):
        if self.connection.is_phone_locked():
            self.connection.send_key(key_codes["enter"])
            self.connection.send_key(key_codes["#"])
            self.connection.send_key(3)
            self.connection.send_key(3)
            self.connection.send_key(3)
            self.connection.send_key(3)
            print("Phone unlocked")
        else:
            print("Phone already unlocked")

    def with_phone_unlocked(self, func):
        if not self.is_phone_unlocked:
            self.unlock_phone()
            self.is_phone_unlocked = True
            
        func(self.connection)

    def open_application(self, app):
        send_keystoke(application_keypath[app], self.connection)