Add hub version in regression test report

PiperOrigin-RevId: 602583331

ui_automator/ui_automator.py

diff --git a/ui_automator/ui_automator.py b/ui_automator/ui_automator.py
index 8a3e7c2..663e6b7 100644
--- a/ui_automator/ui_automator.py
+++ b/ui_automator/ui_automator.py
@@ -89,6 +89,15 @@
         ' test report. e.g. `--dut X123123123,LIGHT,Matter`.'
     ),
 )
+_HUB_VERSION_KEY = 'hub_version'
+_HUB = flags.DEFINE_string(
+    name='hub',
+    default=None,
+    help=(
+        'Use --hub {HubVersion} to include this field in'
+        ' regression test report. e.g. `--hub 10.1.3`.'
+    ),
+)
 
 
 class RegTestSuiteType(enum.Enum):
@@ -409,6 +418,9 @@
     if _DUT.value:
       temp_info['dut'] = f"<{', '.join(_DUT.value)}>"
 
+    if _HUB.value:
+      temp_info[_HUB_VERSION_KEY] = _HUB.value
+
     report_info: test_reporter.ReportInfo = {**temp_info}
 
     return report_info

ui_automator/ui_automator_test.py

diff --git a/ui_automator/ui_automator_test.py b/ui_automator/ui_automator_test.py
index 02fb559..661bd67 100644
--- a/ui_automator/ui_automator_test.py
+++ b/ui_automator/ui_automator_test.py
@@ -1039,6 +1039,7 @@
     fake_report_info: test_reporter.ReportInfo = {
         'gha_version': '0.0.0',
         'gms_core_version': '0.0.1',
+        'hub_version': '10.1.3',
     }
     mock_get_report_info.return_value = fake_report_info
     mock_get_all_instances.return_value = [self.mock_android_device]
@@ -1075,6 +1076,20 @@
 
     self.assertDictEqual(report_info, {})
 
+  @flagsaver.flagsaver((ui_automator._HUB, '10.1.3'))
+  def test_get_report_info_includes_hub_value_from_flag_input(self):
+    report_info = self.ui_automator.get_report_info()
+
+    self.assertEqual(len(report_info), 1)
+    self.assertEqual(report_info.get('hub_version'), '10.1.3')
+
+  @flagsaver.flagsaver((ui_automator._HUB, None))
+  def test_get_report_info_returns_empty_dict_without_hub_flag_input(
+      self,
+  ):
+    report_info = self.ui_automator.get_report_info()
+
+    self.assertDictEqual(report_info, {})
 
 if __name__ == '__main__':
   unittest.main()