Merge pull request #3844 from dimagi/cs/fix_failing_test

test: fix two flaky tests that fail intermittently in CI

Author: SmittieC

apps/pipelines/tests/test_add_start_end_nodes.py

@@ -258,9 +258,7 @@ def test_remove_start_end_nodes(team):
     remove_all_start_end_nodes(Node)
     pipeline.refresh_from_db()
 
-    assert pipeline.node_set.all().count() == 2
-    assert pipeline.node_set.all()[0].flow_id == passthrough_1["id"]
-    assert pipeline.node_set.all()[1].flow_id == passthrough_2["id"]
+    assert set(pipeline.node_set.values_list("flow_id", flat=True)) == {passthrough_1["id"], passthrough_2["id"]}
     assert len(pipeline.data["edges"]) == 1
     assert pipeline.data["edges"][0] == {
         "id": "1->2",

apps/service_providers/tests/test_langfuse_client_manager.py

@@ -99,12 +99,12 @@ def test_prune_stale_clients(client_manager, config, langfuse_mock):
     # Arrange
     other_config = {"public_key": "other_key", "secret_key": "other_secret"}
     first_client = client_manager.get(config)
-
-    time.sleep(0.4)
     client_manager.get(other_config)  # Get second client
     assert len(client_manager.key_timestamps) == 2
 
-    time.sleep(0.1)  # Sleep longer than the stale timeout
+    # Backdate the first client so only it is past the stale timeout
+    client_manager.key_timestamps[config["public_key"]] -= client_manager.stale_timeout + 1
+
     client_manager._prune_stale()
 
     assert len(client_manager.key_timestamps) == 1