From df869c95d057f9b36178dbe4fec63252b5f69f34 Mon Sep 17 00:00:00 2001
From: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Date: Thu, 14 Dec 2023 14:59:45 +0100
Subject: [PATCH] tests: mark lit drive as flaky (#19155)

---
 tests/tests_app/storage/test_drive.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/tests_app/storage/test_drive.py b/tests/tests_app/storage/test_drive.py
index 92e2b81544..5c6456f384 100644
--- a/tests/tests_app/storage/test_drive.py
+++ b/tests/tests_app/storage/test_drive.py
@@ -46,6 +46,7 @@ class SyncFlowLITDrives(LightningFlow):
         self.stop()
 
 
+@pytest.mark.flaky(reruns=3, reruns_delay=5)  # todo: likely dead feature, fine to crash...
 def test_synchronization_lit_drive(tmpdir):
     if os.path.exists("a.txt"):
         os.remove("a.txt")
@@ -104,12 +105,14 @@ class LITDriveFlow(LightningFlow):
             self.stop()
 
 
+@pytest.mark.flaky(reruns=3, reruns_delay=5)  # todo: likely dead feature, fine to crash...
 def test_lit_drive_transferring_files():
     app = LightningApp(LITDriveFlow())
     MultiProcessRuntime(app, start_server=False).dispatch()
     os.remove("a.txt")
 
 
+@pytest.mark.flaky(reruns=3, reruns_delay=5)  # todo: likely dead feature, fine to crash...
 def test_lit_drive():
     with pytest.raises(Exception, match="Unknown protocol for the drive 'id' argument"):
         Drive("invalid_drive_id")