add D555 to test-fw-update

Author: AviaAv

unit-tests/run-unit-tests.py

@@ -55,6 +55,7 @@ def usage():
     print( '        --no-reset           Do not try to reset any devices, with or without a hub' )
     print( '        --hub-reset          If a hub is available, reset the hub itself' )
     print( '        --custom-fw-d400          If custom fw provided flash it if its different that the current fw installed' )
+    print( '        --custom-fw-d555          If custom fw provided flash it if its different that the current fw installed' )
     print( '        --rslog              Enable LibRS logging (LOG_DEBUG etc.) to console in each test' )
     print( '        --skip-disconnected  Skip live test if required device is disconnected (only applies w/o a hub)' )
     print( '        --test-dir <>        Path to test dir; default: librealsense/unit-tests' )
@@ -84,7 +85,7 @@ def usage():
     opts, args = getopt.getopt( sys.argv[1:], 'hvqr:st:',
                                 longopts=['help', 'verbose', 'debug', 'quiet', 'regex=', 'stdout', 'tag=', 'list-tags',
                                           'list-tests', 'no-exceptions', 'context=', 'repeat=', 'retry=', 'config=', 'no-reset', 'hub-reset',
-                                          'rslog', 'skip-disconnected', 'live', 'not-live', 'device=', 'exclude-device=', 'test-dir=','skip-regex=','custom-fw-d400='] )
+                                          'rslog', 'skip-disconnected', 'live', 'not-live', 'device=', 'exclude-device=', 'test-dir=','skip-regex=','custom-fw-d400=','custom-fw-d555='] )
 except getopt.GetoptError as err:
     log.e( err )  # something like "option -a not recognized"
     usage()
@@ -104,6 +105,7 @@ def usage():
 hub_reset = False
 skip_disconnected = False
 custom_fw_path=''
+custom_fw_d555_path=''
 rslog = False
 only_live = False
 only_not_live = False
@@ -179,6 +181,9 @@ def usage():
     elif opt == '--custom-fw-d400':
         custom_fw_path = arg  # Store the custom firmware path
         log.i(f"custom firmware path was provided ${custom_fw_path}")
+    elif opt == '--custom-fw-d555':
+        custom_fw_d555_path = arg  # Store the custom D555 firmware path
+        log.i(f"custom D555 firmware path was provided ${custom_fw_d555_path}")
 
 def find_build_dir( dir ):
     """
@@ -469,6 +474,9 @@ def test_wrapper_( test, configuration=None, repetition=1, curr_retry=0, max_ret
     if test.name == "test-fw-update" and custom_fw_path:
         opts.append('--custom-fw-d400')
         opts.append(custom_fw_path)
+    if test.name == "test-fw-update" and custom_fw_d555_path:
+        opts.append('--custom-fw-d555')
+        opts.append(custom_fw_d555_path)
     try:
         test.run_test( configuration = configuration, log_path = log_path, opts = opts )
     except FileNotFoundError as e:

unit-tests/test-fw-update.py

@@ -6,6 +6,7 @@
 #test:timeout 500
 #test:donotrun:gha
 #test:device each(D400*)
+#test:device D555
 
 import sys
 import os
@@ -14,38 +15,25 @@
 import platform
 import pyrealsense2 as rs
 import pyrsutils as rsutils
-from rspy import devices, log, test, file, repo
+from rspy import log, test, file, repo
 import time
 import argparse
 
 # Parse command-line arguments
 parser = argparse.ArgumentParser(description="Test firmware update")
 parser.add_argument('--custom-fw-d400', type=str, help='Path to custom firmware file')
+parser.add_argument('--custom-fw-d555', type=str, help='Path to custom D555 firmware file')
 args = parser.parse_args()
 
 custom_fw_d400_path = args.custom_fw_d400
+custom_fw_d555_path = args.custom_fw_d555
 if custom_fw_d400_path:
-    log.i(f"Custom firmware path provided: {custom_fw_d400_path}")
+    log.i(f"Custom D400 firmware path provided: {custom_fw_d400_path}")
+elif custom_fw_d555_path:
+    log.i(f"Custom D555 firmware path provided: {custom_fw_d555_path}")
 else:
     log.i(f"No Custom firmware path provided. using bundled firmware")
 
-# This is the first test running, discover acroname modules.
-# Not relevant to MIPI devices running on jetson for LibCI
-if 'jetson' not in test.context:
-    if not devices.hub:
-        log.i( "No hub library found; skipping device FW update" )
-        sys.exit(0)
-    # Following will throw if no acroname module is found
-    from rspy import device_hub
-
-    if device_hub.create() is None:
-        log.f("No hub found")
-    # Remove acroname -- we're likely running inside run-unit-tests in which case the
-    # acroname hub is likely already connected-to from there and we'll get an error
-    # thrown ('failed to connect to acroname (result=11)'). We do not need it -- just
-    # needed to verify it is available above...
-    devices.hub = None
-
 
 def send_hardware_monitor_command(device, command):
     # byte_index = -1
@@ -73,7 +61,7 @@ def extract_version_from_filename(file_path):
         return None
 
     filename = os.path.basename(file_path)
-    match = re.search(r'(\d+)_(\d+)_(\d+)_(\d+)\.bin$', filename)
+    match = re.search(r'(\d+)[._](\d+)[._](\d+)[._](\d+)', filename)
     if match:
         groups = match.groups()
         if groups[3] == '0':
@@ -89,12 +77,15 @@ def extract_version_from_filename(file_path):
 
 def get_update_counter(device):
     product_line = device.get_info(rs.camera_info.product_line)
+    product_name = device.get_info(rs.camera_info.name)
     opcode = 0x09
     start_index = 0x30
     size = None
 
     if product_line == "D400":
         size = 0x2
+    elif "D555" in product_name:
+        return 0 # D555 does not have update counter
     else:
         log.f( "Incompatible product line:", product_line )
 
@@ -163,16 +154,11 @@ def find_image_or_exit( product_name, fw_version_regex = r'(\d+\.){3}(\d+)' ):
 if not fw_updater_exe:
     log.f( "Could not find the update tool file (rs-fw-update.exe)" )
 
-devices.query( monitor_changes = False )
-sn_list = devices.all()
-# acroname should ensure there is always 1 available device
-if len( sn_list ) != 1:
-    log.f( "Expected 1 device, got", len( sn_list ) )
-device = devices.get_first( sn_list ).handle
-log.d( 'found:', device )
+device, ctx = test.find_first_device_or_exit()
 product_line = device.get_info( rs.camera_info.product_line )
 product_name = device.get_info( rs.camera_info.name )
-log.d( 'product line:', product_line )
+serial_number = device.get_info(rs.camera_info.serial_number)
+log.d( 'product line:', product_line, 'serial number:', serial_number )
 ###############################################################################
 #
 
@@ -185,7 +171,7 @@ def find_image_or_exit( product_name, fw_version_regex = r'(\d+\.){3}(\d+)' ):
     try:
         # always flash signed fw when device on recovery befre flashing anything else
         image_file = find_image_or_exit(product_name)
-        cmd = [fw_updater_exe, '-r', '-f', image_file]
+        cmd = [fw_updater_exe, '-r', '-f', image_file, '-s', serial_number]
         log.d( 'running:', cmd )
         subprocess.run( cmd )
         recovered = True
@@ -199,19 +185,35 @@ def find_image_or_exit( product_name, fw_version_regex = r'(\d+\.){3}(\d+)' ):
         test.unexpected_exception()
         log.f( "Unexpected error while trying to recover device:", e )
     else:
-        devices.query( monitor_changes = False )
-        device = devices.get_first( devices.all() ).handle
+        device, ctx = test.find_first_device_or_exit()
+        # check we got the same serial number
+        new_serial_number = device.get_info(rs.camera_info.serial_number)
+        if new_serial_number != serial_number:
+            log.f( "Recovered device has different serial number:", new_serial_number )
 
 current_fw_version = rsutils.version( device.get_info( rs.camera_info.firmware_version ))
 log.d( 'current FW version:', current_fw_version )
-bundled_fw_version = rsutils.version( device.get_info( rs.camera_info.recommended_firmware_version ) )
-log.d( 'bundled FW version:', bundled_fw_version )
+bundled_fw_version = rsutils.version("")
+if device.supports( rs.camera_info.recommended_firmware_version ): # currently, D500 does not support recommended FW
+    bundled_fw_version = rsutils.version( device.get_info( rs.camera_info.recommended_firmware_version ) )
+    log.d( 'bundled FW version:', bundled_fw_version )
 custom_fw_d400_version = extract_version_from_filename(custom_fw_d400_path)
 log.d( 'custom FW D400 version:', custom_fw_d400_version )
-
-
-if (current_fw_version == bundled_fw_version and not custom_fw_d400_path) or \
-   (current_fw_version == custom_fw_d400_version):
+custom_fw_d555_version = extract_version_from_filename(custom_fw_d555_path)
+log.d( 'custom FW D555 version:', custom_fw_d555_version )
+
+# Determine which custom firmware to use based on product
+custom_fw_path = None
+custom_fw_version = None
+if product_line == "D400" and custom_fw_d400_path:
+    custom_fw_path = custom_fw_d400_path
+    custom_fw_version = custom_fw_d400_version
+elif "D555" in product_name and custom_fw_d555_path:
+    custom_fw_path = custom_fw_d555_path
+    custom_fw_version = custom_fw_d555_version
+
+if (current_fw_version == bundled_fw_version and not custom_fw_path) or \
+   (current_fw_version == custom_fw_version):
     if recovered or 'nightly' not in test.context:
         log.d('versions are same; skipping FW update')
         test.finish()
@@ -231,29 +233,33 @@ def find_image_or_exit( product_name, fw_version_regex = r'(\d+\.){3}(\d+)' ):
 if not bundled_fw_version.build():
     fw_version_regex += ".0"  # version drops the build if 0
 fw_version_regex = re.escape( fw_version_regex )
-image_file = find_image_or_exit(product_name, fw_version_regex) if not custom_fw_d400_path else custom_fw_d400_path
+image_file = find_image_or_exit(product_name, fw_version_regex) if not custom_fw_path else custom_fw_path
 # finding file containing image for FW update
 
-cmd = [fw_updater_exe, '-f', image_file]
-if custom_fw_d400_path:
+cmd = [fw_updater_exe, '-f', image_file, '-s', serial_number]
+if custom_fw_path:
     # Add '-u' only if the path doesn't include 'signed'
-    if 'signed' not in custom_fw_d400_path.lower():
+    if ('signed' not in custom_fw_path.lower()
+            and "d555" not in product_name.lower()): # currently -u is not supported for D555
         cmd.insert(1, '-u')
+
+# for DDS devices we need to close device and context to detect it back after FW update
+del device, ctx
 log.d( 'running:', cmd )
 sys.stdout.flush()
 subprocess.run( cmd )   # may throw
 
 # make sure update worked
 time.sleep(3) # MIPI devices do not re-enumerate so we need to give them some time to restart
-devices.query( monitor_changes = False )
-sn_list = devices.all()
-device = devices.get_first( sn_list ).handle
+device, ctx = test.find_first_device_or_exit()
 current_fw_version = rsutils.version( device.get_info( rs.camera_info.firmware_version ))
-test.check_equal(current_fw_version, bundled_fw_version if not custom_fw_d400_path else custom_fw_d400_version)  
+
+expected_fw_version = custom_fw_version if custom_fw_path else bundled_fw_version
+test.check_equal(current_fw_version, expected_fw_version)
 new_update_counter = get_update_counter( device )
 # According to FW: "update counter zeros if you load newer FW than (ever) before"
 # TODO: check why update counter is 255 when installing cutom fw
-if new_update_counter > 0 and not custom_fw_d400_version:
+if new_update_counter > 0 and not custom_fw_version:
     test.check_equal( new_update_counter, update_counter + 1 )
 
 test.finish()