add tests for system shutdown (#2071)

asynkron · Oct 27, 2023 · b06ae68 · b06ae68
1 parent 07ef068
commit b06ae68
Showing 1 changed file with 70 additions and 0 deletions.
diff --git a/tests/Proto.Actor.Tests/ActorTests.cs b/tests/Proto.Actor.Tests/ActorTests.cs
@@ -14,6 +14,76 @@ internal class MyAutoRespondMessage : IAutoRespond
 
 public class ActorTests
 {
+    [Fact]
+    public async Task CanNotSpawnAfterShutdown()
+    {
+        await using var system = new ActorSystem();
+        var context = system.Root;
+
+        PID SpawnActorFromFunc(Receive receive) => context.Spawn(Props.FromFunc(receive));
+
+        var pid = SpawnActorFromFunc(ctx =>
+            {
+                if (ctx.Message is string)
+                {
+                    ctx.Respond("hey");
+                }
+
+                return Task.CompletedTask;
+            }
+        );
+
+        var reply = await context.RequestAsync<object>(pid, "hello");
+        Assert.Equal("hey", reply);
+
+        await system.ShutdownAsync();
+
+        var pid2 = SpawnActorFromFunc(ctx =>
+            {
+                if (ctx.Message is string)
+                {
+                    ctx.Respond("hey");
+                }
+
+                return Task.CompletedTask;
+            }
+        );
+
+        Assert.Same(system.DeadLetterPid, pid2);
+    }
+
+    [Fact]
+    public async Task CanNotSendUserMessageAfterShutdown()
+    {
+        await using var system = new ActorSystem();
+        var context = system.Root;
+
+        PID SpawnActorFromFunc(Receive receive) => context.Spawn(Props.FromFunc(receive));
+
+        var pid = SpawnActorFromFunc(ctx =>
+            {
+                if (ctx.Message is string)
+                {
+                    ctx.Respond("hey");
+                }
+
+                return Task.CompletedTask;
+            }
+        );
+
+        var reply = await context.RequestAsync<object>(pid, "hello");
+        Assert.Equal("hey", reply);
+
+        await system.ShutdownAsync();
+
+        //expect DeadLetterException
+
+        await Assert.ThrowsAsync<DeadLetterException>(async () =>
+        {
+            await context.RequestAsync<object>(pid, "hello");
+        }); 
+    }
+
     [Fact]
     public async Task RequestActorAsync()
     {