mirror of
https://github.com/undera/pylgbst.git
synced 2020-11-18 19:37:26 -08:00
16b1612cc6
* Cosmetics * Harmonograph demo * Cleanup * Original * Original * Cosmetics * Original file * Fixes * cosmetics * separate classes * Cosmetics * Cosmetics * fix tests * Remove plotter tests * Add bluegiga * Rename it * Progress * Fix tests * Cosmetics * Found a way for pygatt! * Playing with gatt * Fix hung subscribe * rename class * add test * skeleton for autodetect * safer order * Fix tests * Fix test * Add dbus install * another try * 2 * 3 * 34 * 6 * 7 * Isolate some tests * 8 * back to roots * Try more * 9 * Help * rep * site-packs * Fix? * Py3 come on * dbus * busss * dev null! * Fix test * Cleanup * Fix tests * Fix after review * add package * FIx package paths * Cosmetics * Update * More doc
142 lines
5.4 KiB
Python
142 lines
5.4 KiB
Python
import time
|
|
import unittest
|
|
|
|
from pylgbst.movehub import MoveHub, MOVE_HUB_HARDWARE_HANDLE, PORT_LED, COLOR_RED
|
|
from pylgbst.peripherals import LED, TiltSensor, COLORS
|
|
from tests import log, HubMock, ConnectionMock, Thread
|
|
|
|
HANDLE = MOVE_HUB_HARDWARE_HANDLE
|
|
|
|
|
|
class GeneralTest(unittest.TestCase):
|
|
def _wait_notifications_handled(self, hub):
|
|
hub.connection.running = False
|
|
for _ in range(1, 180):
|
|
time.sleep(1)
|
|
log.debug("Waiting for notifications to process...")
|
|
if hub.connection.finished:
|
|
log.debug("Done waiting for notifications to process")
|
|
break
|
|
|
|
def test_led(self):
|
|
hub = HubMock()
|
|
led = LED(hub, PORT_LED)
|
|
led.set_color(COLOR_RED)
|
|
self.assertEqual("0801813201510009", hub.writes[0][1])
|
|
|
|
def test_tilt_sensor(self):
|
|
hub = HubMock()
|
|
hub.notify_mock.append((HANDLE, '0f00 04 3a 0128000000000100000001'))
|
|
time.sleep(1)
|
|
|
|
def callback(param1, param2=None, param3=None):
|
|
if param2 is None:
|
|
log.debug("Tilt: %s", TiltSensor.DUO_STATES[param1])
|
|
else:
|
|
log.debug("Tilt: %s %s %s", param1, param2, param3)
|
|
|
|
self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
|
|
hub.tilt_sensor.subscribe(callback)
|
|
hub.notify_mock.append((HANDLE, "0500453a05"))
|
|
hub.notify_mock.append((HANDLE, "0a00473a010100000001"))
|
|
time.sleep(1)
|
|
self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
|
|
hub.tilt_sensor.subscribe(callback, TiltSensor.MODE_2AXIS_SIMPLE)
|
|
|
|
hub.notify_mock.append((HANDLE, "0500453a09"))
|
|
time.sleep(1)
|
|
|
|
self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
|
|
hub.tilt_sensor.subscribe(callback, TiltSensor.MODE_2AXIS_FULL)
|
|
hub.notify_mock.append((HANDLE, "0600453a04fe"))
|
|
time.sleep(1)
|
|
|
|
self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
|
|
hub.tilt_sensor.unsubscribe(callback)
|
|
self._wait_notifications_handled(hub)
|
|
# TODO: assert
|
|
|
|
def test_motor(self):
|
|
conn = ConnectionMock()
|
|
conn.notifications.append((14, '0900 04 39 0227003738'))
|
|
hub = HubMock(conn)
|
|
time.sleep(0.1)
|
|
|
|
conn.notifications.append((14, '050082390a'))
|
|
hub.motor_AB.timed(1.5)
|
|
self.assertEqual("0d018139110adc056464647f03", conn.writes[0][1])
|
|
|
|
conn.notifications.append((14, '050082390a'))
|
|
hub.motor_AB.angled(90)
|
|
self.assertEqual("0f018139110c5a0000006464647f03", conn.writes[1][1])
|
|
|
|
def test_capabilities(self):
|
|
conn = ConnectionMock()
|
|
conn.notifications.append((14, '0f00 04 01 0125000000001000000010'))
|
|
conn.notifications.append((14, '0f00 04 02 0126000000001000000010'))
|
|
conn.notifications.append((14, '0f00 04 37 0127000100000001000000'))
|
|
conn.notifications.append((14, '0f00 04 38 0127000100000001000000'))
|
|
conn.notifications.append((14, '0900 04 39 0227003738'))
|
|
conn.notifications.append((14, '0f00 04 32 0117000100000001000000'))
|
|
conn.notifications.append((14, '0f00 04 3a 0128000000000100000001'))
|
|
conn.notifications.append((14, '0f00 04 3b 0115000200000002000000'))
|
|
conn.notifications.append((14, '0f00 04 3c 0114000200000002000000'))
|
|
conn.notifications.append((14, '0f00 8202 01'))
|
|
conn.notifications.append((14, '0f00 8202 0a'))
|
|
|
|
self._inject_notification(conn, '1200 0101 06 4c45474f204d6f766520487562', 1)
|
|
self._inject_notification(conn, '1200 0108 06 4c45474f204d6f766520487562', 2)
|
|
self._inject_notification(conn, '0900 47 3c 0227003738', 3)
|
|
self._inject_notification(conn, '0600 45 3c 020d', 4)
|
|
self._inject_notification(conn, '0900 47 3c 0227003738', 5)
|
|
hub = MoveHub(conn)
|
|
# demo_all(hub)
|
|
self._wait_notifications_handled(hub)
|
|
|
|
def test_color_sensor(self):
|
|
#
|
|
hub = HubMock()
|
|
hub.notify_mock.append((HANDLE, '0f00 04010125000000001000000010'))
|
|
time.sleep(1)
|
|
|
|
def callback(color, unk1, unk2=None):
|
|
name = COLORS[color] if color is not None else 'NONE'
|
|
log.info("Color: %s %s %s", name, unk1, unk2)
|
|
|
|
self._inject_notification(hub, '0a00 4701090100000001', 1)
|
|
hub.color_distance_sensor.subscribe(callback)
|
|
|
|
hub.notify_mock.append((HANDLE, "08004501ff0aff00"))
|
|
time.sleep(1)
|
|
# TODO: assert
|
|
self._inject_notification(hub, '0a00 4701090100000001', 1)
|
|
hub.color_distance_sensor.unsubscribe(callback)
|
|
self._wait_notifications_handled(hub)
|
|
|
|
def test_button(self):
|
|
hub = HubMock()
|
|
time.sleep(1)
|
|
|
|
def callback(pressed):
|
|
log.info("Pressed: %s", pressed)
|
|
|
|
hub.notify_mock.append((HANDLE, "060001020600"))
|
|
hub.button.subscribe(callback)
|
|
|
|
hub.notify_mock.append((HANDLE, "060001020601"))
|
|
hub.notify_mock.append((HANDLE, "060001020600"))
|
|
time.sleep(1)
|
|
# TODO: assert
|
|
hub.button.unsubscribe(callback)
|
|
self._wait_notifications_handled(hub)
|
|
|
|
def _inject_notification(self, hub, notify, pause):
|
|
def inject():
|
|
time.sleep(pause)
|
|
if isinstance(hub, ConnectionMock):
|
|
hub.notifications.append((HANDLE, notify))
|
|
else:
|
|
hub.notify_mock.append((HANDLE, notify))
|
|
|
|
Thread(target=inject).start()
|