def test_provide_callback(self):
"""
XBeeBase constructor should accept a callback function
"""
self.xbee = XBeeBase(self.serial,
callback=self.callback,
error_callback=self.error_callback)
def test_read(self):
"""
_wait_for_frame should properly read a frame of data
"""
device = FakeReadDevice(b'\x7E\x00\x01\x00\xFF')
xbee = XBeeBase(device)
frame = xbee._wait_for_frame()
self.assertEqual(frame.data, b'\x00')
def test_read_invalid_followed_by_valid(self):
"""
_wait_for_frame should skip invalid data
"""
device = FakeReadDevice(b'\x7E\x00\x01\x00\xFA' +
b'\x7E\x00\x01\x05\xFA')
xbee = XBeeBase(device)
frame = xbee._wait_for_frame()
self.assertEqual(frame.data, b'\x05')
def test_read_escaped(self):
"""
_wait_for_frame should properly read a frame of data
Verify that API mode 2 escaped bytes are read correctly
"""
device = FakeReadDevice('\x7E\x00\x04\x7D\x5E\x7D\x5D\x7D\x31\x7D\x33\xE0')
xbee = XBeeBase(device,escaped=True)
frame = xbee._wait_for_frame()
self.assertEqual(frame.data, '\x7E\x7D\x11\x13')
def test_write_escaped(self):
"""
_write method should write the expected data to the serial
device
"""
device = FakeDevice()
xbee = XBeeBase(device, escaped=True)
xbee._write(b'\x7E\x01\x7D\x11\x13')
# Check resuting state of fake device
expected_frame = b'\x7E\x00\x05\x7D\x5E\x01\x7D\x5D\x7D\x31\x7D\x33\xDF'
self.assertEqual(device.data, expected_frame)
def test_write_again(self):
"""
_write method should write the expected data to the serial
device
"""
device = FakeDevice()
xbee = XBeeBase(device)
xbee._write(b'\x00\x01\x02')
# Check resuting state of fake device
expected_frame = b'\x7E\x00\x03\x00\x01\x02\xFC'
self.assertEqual(device.data, expected_frame)
def test_write_again(self):
"""
_write method should write the expected data to the serial
device
"""
device = Serial()
xbee = XBeeBase(device)
xbee._write(b'\x00\x01\x02')
# Check resuting state of fake device
expected_frame = b'\x7E\x00\x03\x00\x01\x02\xFC'
result_frame = device.get_data_written()
self.assertEqual(result_frame, expected_frame)
def setUp(self):
self.base = XBeeBase(None)
def setUp(self):
"""
Set up a base class XBeeBase object which does not have
api_commands or api_responses defined
"""
self.xbee = XBeeBase(None)
