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
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
import unittest
|
|
|
|
import pygatt
|
|
import serial
|
|
from pygatt import BLEAddressType
|
|
from pygatt.backends.bgapi.bgapi import MAX_CONNECTION_ATTEMPTS
|
|
from pygatt.backends.bgapi.device import BGAPIBLEDevice
|
|
|
|
from pylgbst.comms_pygatt import GattoolConnection
|
|
from tests import log
|
|
|
|
|
|
class SerialMock(serial.Serial):
|
|
def write(self, data):
|
|
self.is_open = True
|
|
log.debug("Write data to serial: %s", data)
|
|
return len(data)
|
|
|
|
def flush(self, *args, **kwargs):
|
|
pass
|
|
|
|
def close(self):
|
|
pass
|
|
|
|
def read(self, size=1):
|
|
return bytes()
|
|
|
|
|
|
class BGAPIBLEDeviceMock(BGAPIBLEDevice):
|
|
def subscribe(self, uuid, callback=None, indication=False):
|
|
log.debug("Mock subscribing")
|
|
|
|
def char_write_handle(self, char_handle, value, wait_for_response=False):
|
|
log.debug("Mock write: %s", value)
|
|
|
|
|
|
class BlueGigaBackendMock(pygatt.BGAPIBackend):
|
|
def _open_serial_port(self, max_connection_attempts=MAX_CONNECTION_ATTEMPTS):
|
|
log.debug("Mock open serial port")
|
|
self._ser = SerialMock()
|
|
|
|
def expect(self, expected, *args, **kargs):
|
|
log.debug("Mock expect")
|
|
data = {
|
|
"packet_type": 0x04,
|
|
"sender": "abcdef".encode('ascii'),
|
|
"data": [1, 2, 3],
|
|
"rssi": 1
|
|
}
|
|
self._ble_evt_gap_scan_response(data)
|
|
|
|
def connect(self, address, timeout=5, address_type=BLEAddressType.public, interval_min=60, interval_max=76,
|
|
supervision_timeout=100, latency=0):
|
|
log.debug("Mock connect")
|
|
device = BGAPIBLEDeviceMock("address", 0, self)
|
|
return device
|
|
|
|
|
|
class BlueGigaTests(unittest.TestCase):
|
|
def test_1(self):
|
|
obj = GattoolConnection()
|
|
obj.backend = BlueGigaBackendMock
|
|
obj.connect(u'66:65:64:63:62:61')
|
|
obj.write(0, "test".encode('ascii'))
|
|
obj.set_notify_handler(lambda x: None)
|