Add tests for safe loggers

JuliaGPU · Aug 27, 2024 · df2c888 · df2c888
1 parent 6894f14
commit df2c888
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/test/Project.toml b/test/Project.toml
@@ -2,6 +2,7 @@
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
+Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 REPL = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb"
 ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"

diff --git a/test/util_tests.jl b/test/util_tests.jl
@@ -19,4 +19,49 @@ end
     @test GPUCompiler.mangle_sig(Tuple{typeof(sin), XX{Int64(-10)}}) == "_Z3sin2XXILln10EE"  # "sin(XX<-10l>)"
 end
 
+@testset "safe loggers" begin
+    using Logging: Logging
+
+    struct YieldingLogger <: Logging.AbstractLogger
+        logger::Logging.AbstractLogger
+        YieldingLogger() = new(Logging.current_logger())
+    end
+
+    function Logging.handle_message(logger::YieldingLogger, args...)
+        yield()
+        return Logging.handle_message(logger.logger, args...)
+    end
+
+    Logging.shouldlog(::YieldingLogger, ::Any...) = true
+    Logging.min_enabled_level(::YieldingLogger) = Logging.Debug
+
+    GPUCompiler.@locked function f()
+        GPUCompiler.@safe_debug "safe_debug"
+        GPUCompiler.@safe_info "safe_info"
+        GPUCompiler.@safe_warn "safe_warn"
+        GPUCompiler.@safe_error "safe_error"
+    end
+
+    @test begin
+        @sync begin
+            Threads.@spawn begin
+                sleep(0.1)
+                @debug "debug"
+                sleep(0.1)
+                @info "info"
+                sleep(0.1)
+                @warn "warn"
+                sleep(0.1)
+                @error "error"
+                sleep(0.1)
+            end
+            Threads.@spawn Logging.with_logger(YieldingLogger()) do
+                sleep(0.1)
+                f()
+            end
+        end
+        true
+    end
+end
+
 end