minor fix

mobly/suite_runner.py

@@ -123,6 +123,7 @@ def suite_end(self):
     self._end_time = utils.get_current_epoch_time()
 
   def set_extras(self, extras):
+    """Sets extra information. Must be serializable."""
     self._extras = extras
 
   def to_dict(self):

tests/mobly/suite_runner_test.py

@@ -260,7 +260,6 @@ def test_run_suite_class_finds_suite_class_when_not_in_main_module(
   ):
     mock_test_runner = mock_test_runner_class.return_value
     mock_test_runner.results.is_all_pass = True
-    mock_test_runner
     tmp_file_path = self._gen_tmp_config_file()
     mock_cli_args = ['test_binary', f'--config={tmp_file_path}']
 
@@ -281,13 +280,25 @@ def test_run_suite_class_records_suite_info(self, mock_time, _):
     )
     expected_record.suite_begin()
     expected_record.suite_end()
+    expected_record.set_extras(
+        {
+            'extra-key-0': 'extra-value-0',
+            'extra-key-1': 'extra-value-1',
+        }
+    )
     expected_summary_entry = expected_record.to_dict()
     expected_summary_entry['Type'] = (
         suite_runner.TestSummaryEntryType.SUITE_INFO.value
     )
 
     class FakeTestSuite(base_suite.BaseSuite):
 
+      def get_suite_info(self):
+        return {
+            'extra-key-0': 'extra-value-0',
+            'extra-key-1': 'extra-value-1',
+        }
+
       def setup_suite(self, config):
         super().setup_suite(config)
         self.add_test_class(FakeTest1)