Added Skynet test

andresgutierrez · Oct 18, 2023 · a048297 · a048297
1 parent b14ab8b
commit a048297
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 9 deletions.
diff --git a/Nixie.Tests/Actors/SenderActor.cs b/Nixie.Tests/Actors/SenderActor.cs
@@ -29,18 +29,18 @@ public SenderPongActor(IActorContext<SenderPongActor, SenderRequest> context)
         this.context = context;
     }
 
-    public async Task Receive(SenderRequest message)
-    {
-        await Task.Yield();
-
+    public Task Receive(SenderRequest message)
+    {        
         if (message.Type != SenderRequestType.Broadcast)
-            return;
+            return Task.CompletedTask;
 
         if (context.Sender is IActorRef<SenderActor, SenderRequest> senderActor)
             senderActor.Send(new SenderRequest(SenderRequestType.Pong));
 
         if (context.Sender is IActorRef<NobodyActor, object> nobodyActor)
             nobodyActor.Send(new SenderRequest(SenderRequestType.Pong));
+
+        return Task.CompletedTask;
     }
 }
 
@@ -70,10 +70,8 @@ public void IncrMessage()
         receivedMessages++;
     }
 
-    public async Task Receive(SenderRequest message)
-    {
-        await Task.Yield();
-
+    public Task Receive(SenderRequest message)
+    {        
         if (message.Type == SenderRequestType.Broadcast)
         {
             for (int i = 0; i < 10; i++)
@@ -88,5 +86,7 @@ public async Task Receive(SenderRequest message)
 
         if (message.Type == SenderRequestType.Pong)
             IncrMessage();
+
+        return Task.CompletedTask;
     }
 }
diff --git a/Nixie.Tests/TestSender.cs b/Nixie.Tests/TestSender.cs
@@ -1,4 +1,5 @@
 
+using System.Diagnostics;
 using Nixie.Tests.Actors;
 
 namespace Nixie.Tests;
@@ -81,4 +82,25 @@ public async Task TestBroadcastMixedMessagesAndUseSenderToReply()
 
         Assert.Equal(100, ((SenderActor)actor.Runner.Actor!).GetMessages());
     }
+
+    [Fact]
+    public async Task TestSkynet()
+    {
+        using ActorSystem asx = new();
+
+        IActorRef<SenderActor, SenderRequest> actor = asx.Spawn<SenderActor, SenderRequest>();
+
+        Stopwatch st = Stopwatch.StartNew();
+
+        SenderRequest request = new(SenderRequestType.Broadcast);
+
+        for (int i = 0; i < 100000; i++)
+            actor.Send(request);
+
+        await asx.Wait();
+
+        Console.WriteLine("Time=", st.ElapsedMilliseconds);
+
+        Assert.Equal(1000000, ((SenderActor)actor.Runner.Actor!).GetMessages());
+    }
 }