Merge pull request #108 from Achllle/fix/test_bagfile_topic

Retry reading bag file messages without backslash in unit tests
team-vigir · Feb 9, 2020 · 86c80ec · 86c80ec
2 parents c06c6f4 + a3c9373
commit 86c80ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/flexbe_testing/src/flexbe_testing/data_provider.py b/flexbe_testing/src/flexbe_testing/data_provider.py
@@ -32,7 +32,10 @@ def parse(self, value):
             # message data
             if (isinstance(value, str) and len(value) > 1 and value[0] == '/' and value[1] != '/' and
                     self._bag is not None):
-                (_, result, _) = list(self._bag.read_messages(topics=[value]))[0]
+                try:
+                    (_, result, _) = list(self._bag.read_messages(topics=[value]))[0]
+                except IndexError:
+                    (_, result, _) = list(self._bag.read_messages(topics=[value[1:]]))[0]
             # anonymous function
             elif isinstance(value, str) and value.startswith('lambda '):
                 result = eval(value)