Add max history length

brewtils/models.py

@@ -452,11 +452,14 @@ def __init__(self, heartbeat=None, history=None):
         self.heartbeat = heartbeat
         self.history = history or []
 
-    def set_status_heartbeat(self, status):
+    def set_status_heartbeat(self, status, max_history=None):
 
         self.heartbeat = datetime.utcnow()
         self.history.append(StatusHistory(status=copy.deepcopy(status), heartbeat=self.heartbeat))
 
+        if max_history and len(self.history) > max_history:
+            self.history = self.history[(max_history * -1):]
+
     def __str__(self):
         return self.heartbeat
 

test/models_test.py

@@ -20,6 +20,7 @@
     RequestTemplate,
     LegacyRole,
     Subscriber,
+    StatusInfo,
     Topic,
 )
 from pytest_lazyfixture import lazy_fixture
@@ -713,3 +714,24 @@ def test_repr(self, topic1, subscriber1):
             topic1.name,
             [subscriber1],
         )
+
+
+class TestStatusInfo:
+
+    def test_max_history(self):
+        status_info = StatusInfo()
+
+        max_length = 5
+
+        for _ in range(10):
+            status_info.set_status_heartbeat("RUNNING", max_history=max_length)
+
+        assert len(status_info.history) == max_length
+
+    def test_history(self):
+        status_info = StatusInfo()
+
+        for _ in range(10):
+            status_info.set_status_heartbeat("RUNNING")
+
+        assert len(status_info.history) == 10