Add integration test to verify experiment completion

Author: owbone

src/litlogger/experiment.py

@@ -476,6 +476,8 @@ def finalize(self, status: str | None = None, print_summary: bool = True) -> Non
                 url=self._url,
             )
 
+        #self._done_event.wait()
+
     def print_url(self) -> None:
         """Print the experiment URL and initialization info with styled output."""
         self._printer.experiment_start(

tests/integrations/test_standalone_lifecycle.py

@@ -13,6 +13,7 @@
 from lightning_sdk.lightning_cloud.openapi.models import LitLoggerServiceDeleteMetricsStreamBody
 from lightning_sdk.utils.resolve import _get_cloud_url
 from litlogger.api.client import LitRestClient
+from litlogger.types import PhaseType
 
 # Suppress deprecation warnings from legacy API usage in integration tests
 pytestmark = pytest.mark.filterwarnings("ignore::DeprecationWarning")
@@ -400,3 +401,32 @@ def test_get_or_create_experiment_metrics():
         project_id=teamspace_id,
         body=LitLoggerServiceDeleteMetricsStreamBody(ids=[original_id, new_experiment.id]),
     )
+
+
+@pytest.mark.cloud()
+def test_experiment_completion():
+    """Test that finalize correctly marks the experiment as complete."""
+    exp = litlogger.init(
+        name=f"lightning_experiment_completion-{uuid.uuid4().hex}",
+        teamspace="oss-litlogger",
+    )
+
+    exp.log_metrics({"my_metric": 0}, step=0)
+    exp.finalize()
+    
+    stream_id = exp._metrics_store.id
+
+    client = LitRestClient()
+    response = client.lit_logger_service_get_metrics_stream(
+        project_name=exp._teamspace.name,
+        project_owner_name=exp._teamspace.owner.name,
+        id=stream_id,
+    )
+    
+    client.lit_logger_service_delete_metrics_stream(
+        project_id=exp._teamspace.id,
+        body=LitLoggerServiceDeleteMetricsStreamBody(ids=[stream_id]),
+    )
+
+    assert response is not None
+    assert response.phase == PhaseType.COMPLETED, f"Expected metrics stream status to be 'COMPLETED', but got {response.metrics_stream.phase}"