class TestDfuTransportSerial(unittest.TestCase):
DEVKEY_PORT = "NORDICSEMI_PCA10028_1_PORT"
def setUp(self):
setup_logging()
# Assert that environment variables are setUp before starting tests.
# TODO: create generic functionality for fetching environment variables that map
# TODO: communication ports to PCA versions
# TODO: setup target nRF5X device to a given state (bootloader+sd+application)
if self.DEVKEY_PORT not in os.environ:
self.fail(
"Environment variable {0} not found. "
"Must specify serial port with development kit connected.".
format(self.DEVKEY_PORT))
self.transport = DfuTransportSerial(os.environ[self.DEVKEY_PORT],
baud_rate=38400,
flow_control=True)
def tearDown(self):
if self.transport and self.transport.is_open():
self.transport.close()
def test_open_close(self):
self.transport.open()
self.assertTrue(self.transport.is_open())
self.transport.close()
self.assertFalse(self.transport.is_open())
def test_dfu_methods(self):
def timeout_callback(log_message):
logging.debug("timeout_callback. Message: %s", log_message)
def progress_callback(progress, log_message, done):
logging.debug("Log message: %s, Progress: %d, done: %s",
log_message, progress, done)
def error_callback(log_message=""):
logging.error("Log message: %s", log_message)
self.transport.register_events_callback(DfuEvent.TIMEOUT_EVENT,
timeout_callback)
self.transport.register_events_callback(DfuEvent.PROGRESS_EVENT,
progress_callback)
self.transport.register_events_callback(DfuEvent.ERROR_EVENT,
error_callback())
firmware = ''
test_firmware_path = os.path.join("firmwares",
"pca10028_nrf51422_xxac_blinky.bin")
with open(test_firmware_path, 'rb') as f:
while True:
data = f.read()
if data:
firmware += data
else:
break
crc = crc16.calc_crc16(firmware, 0xffff)
self.transport.open()
# Sending start DFU command to target
self.transport.send_start_dfu(HexType.APPLICATION,
app_size=len(firmware),
softdevice_size=0,
bootloader_size=0)
# Sending DFU init packet to target
init_packet_vars = {
PacketField.DEVICE_TYPE: 1,
PacketField.DEVICE_REVISION: 2,
PacketField.APP_VERSION: 0xfffa,
PacketField.REQUIRED_SOFTDEVICES_ARRAY: [0x005a],
PacketField.NORDIC_PROPRIETARY_OPT_DATA_FIRMWARE_CRC16: crc
}
pkt = Packet(init_packet_vars)
self.transport.send_init_packet(pkt.generate_packet())
# Sending firmware to target
self.transport.send_firmware(firmware)
# Validating firmware
self.transport.send_validate_firmware()
self.transport.send_activate_firmware()
self.transport.close()
class TestDfuTransportSerial(unittest.TestCase):
DEVKEY_PORT = "NORDICSEMI_PCA10028_1_PORT"
def setUp(self):
setup_logging()
# Assert that environment variables are setUp before starting tests.
# TODO: create generic functionality for fetching environment variables that map
# TODO: communication ports to PCA versions
# TODO: setup target nRF5X device to a given state (bootloader+sd+application)
if self.DEVKEY_PORT not in os.environ:
self.fail("Environment variable {0} not found. "
"Must specify serial port with development kit connected."
.format(self.DEVKEY_PORT))
self.transport = DfuTransportSerial(os.environ[self.DEVKEY_PORT],
baud_rate=38400,
flow_control=True)
def tearDown(self):
if self.transport and self.transport.is_open():
self.transport.close()
def test_open_close(self):
self.transport.open()
self.assertTrue(self.transport.is_open())
self.transport.close()
self.assertFalse(self.transport.is_open())
def test_dfu_methods(self):
def timeout_callback(log_message):
logging.debug("timeout_callback. Message: %s", log_message)
def progress_callback(progress, log_message, done):
logging.debug("Log message: %s, Progress: %d, done: %s", log_message, progress, done)
def error_callback(log_message=""):
logging.error("Log message: %s", log_message)
self.transport.register_events_callback(DfuEvent.TIMEOUT_EVENT, timeout_callback)
self.transport.register_events_callback(DfuEvent.PROGRESS_EVENT, progress_callback)
self.transport.register_events_callback(DfuEvent.ERROR_EVENT, error_callback())
firmware = ''
test_firmware_path = os.path.join("firmwares", "pca10028_nrf51422_xxac_blinky.bin")
with open(test_firmware_path, 'rb') as f:
while True:
data = f.read()
if data:
firmware += data
else:
break
crc = crc16.calc_crc16(firmware, 0xffff)
self.transport.open()
# Sending start DFU command to target
self.transport.send_start_dfu(HexType.APPLICATION,
app_size=len(firmware),
softdevice_size=0,
bootloader_size=0)
# Sending DFU init packet to target
init_packet_vars = {
PacketField.DEVICE_TYPE: 1,
PacketField.DEVICE_REVISION: 2,
PacketField.APP_VERSION: 0xfffa,
PacketField.REQUIRED_SOFTDEVICES_ARRAY: [0x005a],
PacketField.NORDIC_PROPRIETARY_OPT_DATA_FIRMWARE_CRC16: crc
}
pkt = Packet(init_packet_vars)
self.transport.send_init_packet(pkt.generate_packet())
# Sending firmware to target
self.transport.send_firmware(firmware)
# Validating firmware
self.transport.send_validate_firmware()
self.transport.send_activate_firmware()
self.transport.close()
