Revert back old code, but keep Task { @mainactor

0xLeif · Oct 25, 2024 · a7ba8ac · a7ba8ac
1 parent e2a33d8
commit a7ba8ac
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/Sources/AppState/Application/Types/State/Application+State.swift b/Sources/AppState/Application/Types/State/Application+State.swift
@@ -34,7 +34,7 @@ extension Application {
                     )
                 else {
                     defer {
-                        Task { @MainActor in
+                        let setValue = {
                             shared.cache.set(
                                 value: Application.State(
                                     type: .state,
@@ -44,6 +44,17 @@ extension Application {
                                 forKey: scope.key
                             )
                         }
+                        #if (!os(Linux) && !os(Windows))
+                        if NSClassFromString("XCTest") == nil {
+                            Task { @MainActor in
+                                setValue()
+                            }
+                        } else {
+                            setValue()
+                        }
+                        #else
+                        setValue()
+                        #endif
                     }
                     return _value
                 }