show lease expected release time instead of start time

"jmp get exporters --with leases" shows lease status and when is the lease expiring. That's more useful that current start time which doesn't tell you for how long it may be held.
"jmp get leases" now shows lease begin time (the time it was acquired) in addition to duration.

Author: michalskrivanek

packages/jumpstarter/jumpstarter/client/grpc.py

@@ -32,7 +32,7 @@ def add_display_columns(table, options: WithOptions = None):
     if options.show_leases:
         table.add_column("LEASED BY")
         table.add_column("LEASE STATUS")
-        table.add_column("START TIME")
+        table.add_column("EXPECTED RELEASE")
 
 
 def add_exporter_row(table, exporter, options: WithOptions = None, lease_info: tuple[str, str, str] | None = None):
@@ -45,10 +45,10 @@ def add_exporter_row(table, exporter, options: WithOptions = None, lease_info: t
     row_data.append(",".join(("{}={}".format(k, v) for k, v in sorted(exporter.labels.items()))))
     if options.show_leases:
         if lease_info:
-            lease_client, lease_status, start_time = lease_info
+            lease_client, lease_status, expected_release = lease_info
         else:
-            lease_client, lease_status, start_time = "", "Available", ""
-        row_data.extend([lease_client, lease_status, start_time])
+            lease_client, lease_status, expected_release = "", "Available", ""
+        row_data.extend([lease_client, lease_status, expected_release])
 
     table.add_row(*row_data)
 
@@ -97,10 +97,11 @@ def rich_add_rows(self, table, options: WithOptions = None):
         if options and options.show_leases and self.lease:
             lease_client = self.lease.client
             lease_status = self.lease.get_status()
-            start_time = ""
-            if self.lease.effective_begin_time:
-                start_time = self.lease.effective_begin_time.strftime("%Y-%m-%d %H:%M:%S")
-            lease_info = (lease_client, lease_status, start_time)
+            expected_release = ""
+            if self.lease.effective_begin_time and self.lease.effective_duration:
+                release_time = self.lease.effective_begin_time + self.lease.effective_duration
+                expected_release = release_time.strftime("%Y-%m-%d %H:%M:%S")
+            lease_info = (lease_client, lease_status, expected_release)
         elif options and options.show_leases:
             lease_info = ("", "Available", "")
         add_exporter_row(table, self, options, lease_info)
@@ -165,15 +166,25 @@ def from_protobuf(cls, data: client_pb2.Lease) -> Lease:
     def rich_add_columns(cls, table):
         table.add_column("NAME", no_wrap=True)
         table.add_column("SELECTOR")
+        table.add_column("BEGIN TIME")
         table.add_column("DURATION")
         table.add_column("CLIENT")
         table.add_column("EXPORTER")
 
     def rich_add_rows(self, table):
+        begin_time = ""
+        if self.effective_begin_time:
+            begin_time = self.effective_begin_time.strftime("%Y-%m-%d %H:%M:%S")
+
+        duration = ""
+        if self.effective_duration:
+            duration = str(self.effective_duration)
+
         table.add_row(
             self.name,
             self.selector,
-            str(self.duration),
+            begin_time,
+            duration,
             self.client,
             self.exporter,
         )

packages/jumpstarter/jumpstarter/client/grpc_test.py

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timedelta
 from io import StringIO
 from unittest.mock import Mock
 
@@ -48,15 +48,15 @@ def test_with_leases_columns(self):
         add_display_columns(table, options)
 
         columns = [col.header for col in table.columns]
-        assert columns == ["NAME", "LABELS", "LEASED BY", "LEASE STATUS", "START TIME"]
+        assert columns == ["NAME", "LABELS", "LEASED BY", "LEASE STATUS", "EXPECTED RELEASE"]
 
     def test_with_all_columns(self):
         table = Table()
         options = WithOptions(show_online=True, show_leases=True)
         add_display_columns(table, options)
 
         columns = [col.header for col in table.columns]
-        assert columns == ["NAME", "ONLINE", "LABELS", "LEASED BY", "LEASE STATUS", "START TIME"]
+        assert columns == ["NAME", "ONLINE", "LABELS", "LEASED BY", "LEASE STATUS", "EXPECTED RELEASE"]
 
 
 class TestAddExporterRow:
@@ -91,7 +91,7 @@ def test_row_with_lease_info(self):
         add_exporter_row(table, exporter, options, lease_info)
 
         assert len(table.rows) == 1
-        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, START TIME
+        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, EXPECTED RELEASE
 
     def test_row_with_lease_info_available(self):
         table = Table()
@@ -115,7 +115,7 @@ def test_row_with_all_options(self):
         add_exporter_row(table, exporter, options, lease_info)
 
         assert len(table.rows) == 1
-        assert len(table.columns) == 6  # NAME, ONLINE, LABELS, LEASED BY, LEASE STATUS, START TIME
+        assert len(table.columns) == 6  # NAME, ONLINE, LABELS, LEASED BY, LEASE STATUS, EXPECTED RELEASE
 
 
 class TestExporterList:
@@ -124,6 +124,7 @@ def create_test_lease(self, client="test-client", status="Active"):
         lease.client = client
         lease.get_status.return_value = status
         lease.effective_begin_time = datetime(2023, 1, 1, 10, 0, 0)
+        lease.effective_duration = timedelta(hours=1)
         return lease
 
     def test_exporter_without_lease(self):
@@ -175,7 +176,7 @@ def test_exporter_with_lease_display(self):
         exporter.rich_add_rows(table, options)
 
         assert len(table.rows) == 1
-        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, START TIME
+        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, EXPECTED RELEASE
 
         # Test actual table content by rendering it
         console = Console(file=StringIO(), width=120)
@@ -187,7 +188,7 @@ def test_exporter_with_lease_display(self):
         assert "type=device" in output
         assert "test-client" in output
         assert "Active" in output
-        assert "2023-01-01 10:00:00" in output
+        assert "2023-01-01 11:00:00" in output  # Expected release: begin_time (10:00:00) + duration (1h)
 
     def test_exporter_without_lease_but_show_leases(self):
         exporter = Exporter(
@@ -203,7 +204,7 @@ def test_exporter_without_lease_but_show_leases(self):
         exporter.rich_add_rows(table, options)
 
         assert len(table.rows) == 1
-        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, START TIME
+        assert len(table.columns) == 5  # NAME, LABELS, LEASED BY, LEASE STATUS, EXPECTED RELEASE
 
         # Test actual table content by rendering it
         console = Console(file=StringIO(), width=120)
@@ -285,7 +286,7 @@ def test_exporter_all_features_display(self):
         exporter_offline_no_lease.rich_add_rows(table, options)
 
         assert len(table.rows) == 2
-        assert len(table.columns) == 6  # NAME, ONLINE, LABELS, LEASED BY, LEASE STATUS, START TIME
+        assert len(table.columns) == 6  # NAME, ONLINE, LABELS, LEASED BY, LEASE STATUS, EXPECTED RELEASE
 
         # Test actual table content by rendering it
         console = Console(file=StringIO(), width=150)
@@ -302,7 +303,7 @@ def test_exporter_all_features_display(self):
         assert "full-test-client" in output  # Lease client
         assert "Active" in output  # Lease status
         assert "Available" in output  # Available status for no lease
-        assert "2023-01-01 10:00:00" in output  # Lease start time
+        assert "2023-01-01 11:00:00" in output  # Expected release time (begin_time + duration)
 
     def test_exporter_lease_info_extraction(self):
         """Test that lease information is correctly extracted from lease objects"""
@@ -325,10 +326,13 @@ def test_exporter_lease_info_extraction(self):
         if options.show_leases and exporter.lease:
             lease_client = exporter.lease.client
             lease_status = exporter.lease.get_status()
-            start_time = exporter.lease.effective_begin_time.strftime("%Y-%m-%d %H:%M:%S")
-            lease_info = (lease_client, lease_status, start_time)
+            expected_release = ""
+            if exporter.lease.effective_begin_time and exporter.lease.effective_duration:
+                release_time = exporter.lease.effective_begin_time + exporter.lease.effective_duration
+                expected_release = release_time.strftime("%Y-%m-%d %H:%M:%S")
+            lease_info = (lease_client, lease_status, expected_release)
 
-            assert lease_info == ("my-client", "Expired", "2023-01-01 10:00:00")
+            assert lease_info == ("my-client", "Expired", "2023-01-01 11:00:00")
 
     def test_exporter_no_lease_info_extraction(self):
         """Test that default lease information is used when no lease exists"""