From 1777a6d24abffb933105e6fdd148368a4c841d50 Mon Sep 17 00:00:00 2001
From: Dean Herbert <pe@ppy.sh>
Date: Fri, 1 Jul 2022 16:06:18 +0900
Subject: [PATCH] Add attribute to retry flaky tests on normal CI runs

---
 osu.Game/Tests/FlakyTestAttribute.cs | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 osu.Game/Tests/FlakyTestAttribute.cs

diff --git a/osu.Game/Tests/FlakyTestAttribute.cs b/osu.Game/Tests/FlakyTestAttribute.cs
new file mode 100644
index 0000000000..a5646f5d00
--- /dev/null
+++ b/osu.Game/Tests/FlakyTestAttribute.cs
@@ -0,0 +1,26 @@
+// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
+// See the LICENCE file in the repository root for full licence text.
+
+#nullable disable
+using System;
+using NUnit.Framework;
+
+namespace osu.Game.Tests
+{
+    /// <summary>
+    /// An attribute to mark any flaky tests.
+    /// Will add a retry count unless environment variable `FAIL_FLAKY_TESTS` is set to `1`.
+    /// </summary>
+    public class FlakyTestAttribute : RetryAttribute
+    {
+        public FlakyTestAttribute()
+            : this(10)
+        {
+        }
+
+        public FlakyTestAttribute(int tryCount)
+            : base(Environment.GetEnvironmentVariable("FAIL_FLAKY_TESTS") == "1" ? 0 : tryCount)
+        {
+        }
+    }
+}