Update mock_dlt_message.py (#12)

Allows the capablites to have mock data which has a storage timestamp
bmwcarit · May 8, 2024 · b70a24e · b70a24e
1 parent eb290b6
commit b70a24e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/dltlyse/mock_dlt_message.py b/src/dltlyse/mock_dlt_message.py
@@ -34,3 +34,12 @@ class MockStorageHeader(object):
     def __init__(self, msec=0, sec=0):
         self.microseconds = msec
         self.seconds = sec
+
+
+class MockDLTMessageWithTime(MockDLTMessage):
+    "MockDLTMessage but with storage_timestamp"
+
+    @property
+    def storage_timestamp(self):
+        """Fake storage timestamp"""
+        return float("{}.{}".format(self.storageheader.seconds, self.storageheader.microseconds))