Allow the use of FileCheck

test/Project.toml

@@ -2,8 +2,10 @@
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
+IOCapture = "b5f81e59-6552-4d32-b1f0-c071b021bf89"
 InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
 LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
+LLVM_jll = "86de99a1-58d6-5da7-8064-bd56ce2e322c"
 Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"

test/bpf.jl

@@ -1,38 +1,68 @@
 @testset "No-op" begin
-    kernel() = 0
+    mod = @eval module $(gensym())
+        kernel() = 0
+    end
 
-    output = sprint(io->BPF.code_native(io, kernel, ()))
-    @test occursin("\tr0 = 0\n\texit", output)
+    @test @filecheck begin
+        check"CHECK-LABEL: julia_kernel_{{[0-9_]*}}:"
+        check"CHECK: r0 = 0"
+        check"CHECK-NEXT: exit"
+        BPF.code_native(mod.kernel, ())
+    end
 end
 @testset "Return argument" begin
-    kernel(x) = x
+    mod = @eval module $(gensym())
+        kernel(x) = x
+    end
 
-    output = sprint(io->BPF.code_native(io, kernel, (UInt64,)))
-    @test occursin("\tr0 = r1\n\texit", output)
+    @test @filecheck begin
+        check"CHECK-LABEL: julia_kernel_{{[0-9_]*}}:"
+        check"CHECK: r0 = r1"
+        check"CHECK-NEXT: exit"
+        BPF.code_native(mod.kernel, (UInt64,))
+    end
 end
 @testset "Addition" begin
-    kernel(x) = x+1
+    mod = @eval module $(gensym())
+        kernel(x) = x+1
+    end
 
-    output = sprint(io->BPF.code_native(io, kernel, (UInt64,)))
-    @test occursin("\tr0 = r1\n\tr0 += 1\n\texit", output)
+    @test @filecheck begin
+        check"CHECK-LABEL: julia_kernel_{{[0-9_]*}}:"
+        check"CHECK: r0 = r1"
+        check"CHECK-NEXT: r0 += 1"
+        check"CHECK-NEXT: exit"
+        BPF.code_native(mod.kernel, (UInt64,))
+    end
 end
 @testset "Errors" begin
-    kernel(x) = fakefunc(x)
+    mod = @eval module $(gensym())
+        kernel(x) = fakefunc(x)
+    end
 
-    @test_throws GPUCompiler.InvalidIRError BPF.code_execution(kernel, (UInt64,))
+    @test_throws GPUCompiler.InvalidIRError BPF.code_execution(mod.kernel, (UInt64,))
 end
 @testset "Function Pointers" begin
     @testset "valid" begin
-        goodcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
-        kernel(x) = goodcall(x)
+        mod = @eval module $(gensym())
+            goodcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
+            kernel(x) = goodcall(x)
+        end
 
-        output = sprint(io->BPF.code_native(io, kernel, (Int,)))
-        @test occursin(r"\tcall .*\n\texit", output)
+        @test @filecheck begin
+            check"CHECK-LABEL: julia_kernel_{{[0-9_]*}}:"
+            check"CHECK: call"
+            check"CHECK-NEXT: exit"
+            BPF.code_native(mod.kernel, (Int,))
+        end
     end
+
     @testset "invalid" begin
-        badcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3000 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
-        kernel(x) = badcall(x)
+        mod = @eval module $(gensym())
+            badcall(x) = Base.llvmcall("%2 = call i64 inttoptr (i64 3000 to i64 (i64)*)(i64 %0)\nret i64 %2", Int, Tuple{Int}, x)
+            kernel(x) = badcall(x)
+        end
 
-        @test_throws GPUCompiler.InvalidIRError BPF.code_execution(kernel, (Int,))
+        @test_throws GPUCompiler.InvalidIRError BPF.code_execution(mod.kernel, (Int,))
     end
 end