def test_service_setup(self, mock_getsrv, mock_getchar, mock_getdesc):
"""Test service setup"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
services = []
for uuid in [UUIDS.SERVICE_GENERIC, UUIDS.SERVICE_HW01_B]:
services.append(Service(self.h, uuid, 0x0, 0xffff))
mock_getsrv.side_effect = services
characteristics = []
for uuid in [
UUIDS.CHARACTERISTIC_DEVICE_NAME, UUIDS.CHARACTERISTIC_TX,
UUIDS.CHARACTERISTIC_RX
]:
characteristics.append([
Characteristic(self.h, uuid, 0x47, [0b00000010, 0b00001000],
0x47)
])
mock_getchar.side_effect = characteristics
mock_desc = [Descriptor(self.h, UUIDS.NOTIFICATION_DESCRIPTOR, 0x12)]
mock_getdesc.return_value = mock_desc
with mock.patch('bluepy.btle.Descriptor.read',
return_value=(b'\x01\x00')):
with mock.patch('bluepy.btle.Descriptor.write'):
with self.assertLogs('HW01') as log:
self.h.setup_services()
self.assertEqual(log.output, ['INFO:HW01:Setup RX/TX communication'])
def test_valid_datetime(self, mock_raw):
"""Test setting a date and time"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
with self.assertLogs('HW01') as log:
self.h.set_datetime(2015, 10, 25, 7, 28)
self.assertEqual(
log.output, ['INFO:HW01:Date and time set to 2015-10-25 07:28:00'])
def test_device_name(self, mock_device_name):
"""Test device name"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
mock_device_char = Characteristic(
self.h, '00000000-0000-1000-8000-00805f9b34fb', 0xff,
[0b00000010, 0b00001000], 0xff)
self.h.device_char = mock_device_char
self.assertEqual(self.h.get_device_name(), 'HW01')
def test_invalid_datetime(self):
"""Test setting a date time with invalid month"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
with self.assertRaises(ValueError) as _:
with self.assertLogs('HW01') as log:
self.h.set_datetime(2018, 99, 1)
self.assertEqual(
log.output,
['ERROR:HW01:Failed to set date and time: month must be in 1..12'])
def test_bluetooth_version(self, mock_raw):
"""Test Bluetooth version number"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
mock_raw.return_value = 'BT+VER:105.013.032'
self.assertEqual(self.h.get_version(), '105.013.032')
with self.assertLogs('HW01') as log:
mock_raw.return_value = 'fail'
self.h.get_version()
self.assertEqual(log.output,
['ERROR:HW01:Failed to parse Bluetooth version'])
def test_notification_for_unexpected_handle(self):
"""Test Bluetooth notification for unexpected handle"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
mock_char = Characteristic(self.h, UUIDS.CHARACTERISTIC_TX, 0x14,
[0b00010000], 0x14)
tx_delegate = TXDelegate(mock_char.getHandle(), self.h._log)
self.h.setDelegate(tx_delegate)
self.assertEqual(tx_delegate.data, None)
# Trigger notification with expected handle
self.h.delegate.handleNotification(0xff, 'AT+??')
self.assertEqual(tx_delegate.data, None)
def test_connect_without_address(self):
"""Test connection to device without providing mac address"""
self.h = HW01()
self.assertEqual(self.h.mac_address, None)
with self.assertRaises(SystemExit) as _:
with self.assertLogs('HW01') as log:
self.h.connect()
self.assertEqual(log.output, [
'INFO:HW01:No mac address provided, start BTLE scanning for HW01...',
'ERROR:HW01:Fail to scan BTLE'
])
def test_connect_with_bad_address(self):
"""Test connection to device with bad mac address"""
mac_address = 'XX:XX:XX:XX:XX:XX'
self.h = HW01(mac_address)
self.assertEqual(self.h.mac_address, mac_address)
with self.assertRaises(SystemExit) as _:
with self.assertLogs('HW01') as log:
self.h.connect()
self.assertEqual(log.output, [
'INFO:HW01:Connecting to %s' % mac_address,
'ERROR:HW01:Failed to established connection to %s' % mac_address
])
def test_distance_unit(self, mock_raw):
"""Test setting displayed distance unit"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
mock_raw.side_effect = ['AT+UNITS:1', 'AT+UNITS:0']
units = ['metric', 'imperial']
with self.assertLogs('HW01') as log:
for unit in units:
self.h.set_distance_unit(unit)
self.assertEqual(log.output, [
'INFO:HW01:Distance unit set to metric',
'INFO:HW01:Distance unit set to imperial'
])
def test_valid_current_datetime(self, mock_raw): #, mock_now):
"""Test setting a date and time to current local time"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
class MockDatetime(datetime.datetime):
@classmethod
def now(cls):
return cls(2019, 1, 15, 23, 12, 51)
datetime.datetime = MockDatetime
with self.assertLogs('HW01') as log:
self.h.set_datetime()
self.assertEqual(
log.output, ['INFO:HW01:Date and time set to 2019-01-15 23:12:51'])
def test_timeformat(self, mock_raw):
"""Test setting a time format to 12h/24h"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
mock_raw.side_effect = [
'AT+TIMEFORMAT:1', 'AT+TIMEFORMAT:0', 'AT+TIMEFORMAT:1',
'AT+TIMEFORMAT:1'
]
formats = ['24h', '12h', 'unknown', None]
with self.assertLogs('HW01') as log:
for format in formats:
self.h.set_timeformat(format)
self.assertEqual(log.output, [
'INFO:HW01:Time format set to 24h',
'INFO:HW01:Time format set to 12h',
'INFO:HW01:Time format set to 24h',
'INFO:HW01:Time format set to 24h'
])
#!/usr/bin/env python3
import sys
from hw01 import HW01
if __name__ == '__main__':
# Rock'n'Roll
h = HW01('XX:XX:XX:XX:XX:XX', debug=True)
h.connect()
h.setup_services()
try:
battery = h.get_battery_level()
print(battery)
h.set_distance_unit('metric')
version = h.get_version()
print('Bluetooth HW version: %s', version)
except KeyboardInterrupt:
pass
def test_distance_unknown_unit(self, mock_raw):
"""Test setting displayed distance unit with unknown unit"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
self.assertRaises(KeyError, self.h.set_distance_unit('unknown'))
def test_serial_number(self, _):
"""Test device serial number"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
self.assertEqual(self.h.get_serial_number(), 'HW19999999')
def test_battery_level(self, _):
"""Test battery level"""
self.h = HW01('XX:XX:XX:XX:XX:XX')
self.assertEqual(self.h.get_battery_level(), 80)