From 42124a738a3cd2e791deca99f0cd4e5046bbeade Mon Sep 17 00:00:00 2001
From: DrChainsaw <christian.kyril.skarby@gmail.com>
Date: Wed, 30 Sep 2020 00:43:57 +0200
Subject: [PATCH] Test that no new Context is created

---
 test/scheduler.jl | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/test/scheduler.jl b/test/scheduler.jl
index 13d6fad9..2641a40b 100644
--- a/test/scheduler.jl
+++ b/test/scheduler.jl
@@ -66,4 +66,31 @@ end
     end
     @everywhere (pop!(Dagger.PROCESSOR_CALLBACKS); empty!(Dagger.OSPROC_CACHE))
 
+    @testset "Add new workers" begin
+        using Distributed
+        ps1 = addprocs(2, exeflags="--project");
+    
+        @everywhere begin
+           using Dagger, Distributed
+           # Condition to guarantee that processing is not completed before we add new workers
+           c = Condition()
+           function testfun(i)
+               i < 2 && return myid()
+               wait(c)
+               return myid()
+           end
+        end
+    
+        ts = delayed(vcat)((delayed(testfun)(i) for i in 1:4)...);
+        job = @async collect(Context(ps1), ts);
+    
+        ps2 = addprocs(2, exeflags="--project");
+    
+        while !istaskdone(job)
+            @everywhere ps1 notify(c)
+        end
+        @test fetch(job) |> unique |> sort == ps1
+
+        wait(rmprocs(vcat(ps1,ps2)))
+    end
 end