use effective_duration

Signed-off-by: Benny Zlotnik <bzlotnik@redhat.com>

Author: bennyz

python/packages/jumpstarter/jumpstarter/client/lease.py

@@ -289,7 +289,7 @@ def _notify_lease_ending(self, remaining: timedelta) -> None:
 
     def _get_lease_end_time(self, lease) -> datetime | None:
         """Extract the end time from a lease response, or None if not available."""
-        if not (lease.effective_begin_time and lease.effective_duration):
+        if not (lease.effective_begin_time and lease.duration):
             return None
         if lease.effective_end_time:
             return lease.effective_end_time

python/packages/jumpstarter/jumpstarter/client/lease_test.py

@@ -396,15 +396,15 @@ def _make_lease(self):
 
     def test_returns_none_when_no_begin_time(self):
         lease = self._make_lease()
-        response = Mock(effective_begin_time=None, effective_duration=timedelta(minutes=30))
+        response = Mock(effective_begin_time=None, duration=timedelta(minutes=30))
 
         assert lease._get_lease_end_time(response) is None
 
     def test_returns_none_when_no_duration(self):
         lease = self._make_lease()
         response = Mock(
             effective_begin_time=datetime.now(tz=timezone.utc),
-            effective_duration=None,
+            duration=None,
         )
 
         assert lease._get_lease_end_time(response) is None
@@ -414,7 +414,7 @@ def test_returns_effective_end_time_when_present(self):
         end_time = datetime(2025, 6, 1, 12, 0, 0, tzinfo=timezone.utc)
         response = Mock(
             effective_begin_time=datetime(2025, 6, 1, 11, 0, 0, tzinfo=timezone.utc),
-            effective_duration=timedelta(hours=1),
+            duration=timedelta(hours=1),
             effective_end_time=end_time,
         )
 
@@ -426,7 +426,7 @@ def test_calculates_end_time_when_no_effective_end(self):
         duration = timedelta(hours=2)
         response = Mock(
             effective_begin_time=begin,
-            effective_duration=duration,
+            effective_duration=timedelta(hours=1),  # elapsed time, not used for calculation
             effective_end_time=None,
             duration=duration,
         )