fix: [tabbasedhierarchy] fix issues with parsing remotectl_dumpstate

EC-DIGIT-CSIRC · Nov 6, 2024 · 1b6f992 · 1b6f992
1 parent fdc7b23
commit 1b6f992
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/sysdiagnose/utils/tabbasedhierarchy.py b/src/sysdiagnose/utils/tabbasedhierarchy.py
@@ -10,7 +10,7 @@ def parse_tab_based_hierarchal_file(path: str) -> list | dict:
 
 
 def parse_block(lines: list) -> list | dict:
-    result = {}
+    result = None
     n = 0
     while n < len(lines):
         line = lines[n]
@@ -64,6 +64,8 @@ def parse_block(lines: list) -> list | dict:
                 if not result:
                     result = {}
                 result[key.strip()] = clean_value(value)
+            elif not key and isinstance(result, dict):
+                result[line.strip()] = ""
             else:
                 if not result:
                     result = []

diff --git a/tests/test_tabbasedhierarchy.py b/tests/test_tabbasedhierarchy.py
@@ -39,6 +39,7 @@ def test_remotectldumpstateblock(self):
             '			Properties: {',
             '				UsesRemoteXPC => true',
             '			}',
+            '		foobar',
             '		ssh',
             '			Properties: {',
             '				Legacy => true',
@@ -84,6 +85,7 @@ def test_remotectldumpstateblock(self):
                             "UsesRemoteXPC": "true"
                         }
                     },
+                    "foobar": "",
                     "ssh": {
                         "Properties": {
                             "Legacy": "true"