From 1228ca2a4202faa79e89ff5a073975dc5250131c Mon Sep 17 00:00:00 2001
From: Gregorius Soedharmo <arkatufus@yahoo.com>
Date: Tue, 24 Oct 2023 02:03:41 +0700
Subject: [PATCH] Issue #6948 unit test

---
 src/core/Akka.Tests/Actor/BugFix6948Spec.cs | 91 +++++++++++++++++++++
 src/core/Akka.Tests/Akka.Tests.csproj       |  1 +
 2 files changed, 92 insertions(+)
 create mode 100644 src/core/Akka.Tests/Actor/BugFix6948Spec.cs

diff --git a/src/core/Akka.Tests/Actor/BugFix6948Spec.cs b/src/core/Akka.Tests/Actor/BugFix6948Spec.cs
new file mode 100644
index 00000000000..b6102870815
--- /dev/null
+++ b/src/core/Akka.Tests/Actor/BugFix6948Spec.cs
@@ -0,0 +1,91 @@
+//-----------------------------------------------------------------------
+// <copyright file="BugFix6948Spec.cs" company="Akka.NET Project">
+//     Copyright (C) 2009-2023 Lightbend Inc. <http://www.lightbend.com>
+//     Copyright (C) 2013-2023 .NET Foundation <https://github.com/akkadotnet/akka.net>
+// </copyright>
+//-----------------------------------------------------------------------
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Akka.Actor;
+using Akka.Event;
+using Akka.TestKit;
+using FluentAssertions;
+using FluentAssertions.Extensions;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Akka.Tests.Actor;
+
+public class BugFix6948Spec : AkkaSpec
+{
+    public BugFix6948Spec(ITestOutputHelper output): base("akka.loglevel=DEBUG", output)
+    {
+    }
+
+    [Fact(DisplayName = "#6948 ActorSystem should terminate cleanly on fatal exceptions")]
+    public async Task ActorSystem_ShouldTerminateCleanlyOnOOM()
+    {
+        var actor = Sys.ActorOf(Props.Create(() => new OutOfMemoryActor()));
+        actor.Tell("die");
+        ExpectMsg("dead");
+        
+        var shutdown = CoordinatedShutdown.Get(Sys);
+        shutdown.AddTask(CoordinatedShutdown.PhaseBeforeActorSystemTerminate, "z", () =>
+        {
+            var pressure = new List<long>();
+            try
+            {
+                foreach (var i in Enumerable.Range(0, 50_000_000))
+                {
+                    pressure.Add(i);
+                }
+            }
+            catch (OutOfMemoryException ex)
+            {
+                Log.Error(ex, "I died (CoordinatedShutdown)");
+                throw;
+            }
+
+            return Task.FromResult(Done.Instance);
+        });
+
+        var terminated = false;
+        Sys.RegisterOnTermination(() =>
+        {
+            terminated = true;
+        });
+        
+        await shutdown.Run(CoordinatedShutdown.UnknownReason.Instance);
+        await Sys.Terminate().WaitAsync(5.Seconds());
+        terminated.Should().BeTrue();
+    }
+
+    private class OutOfMemoryActor : ReceiveActor
+    {
+        private readonly List<long> _memoryPressure = new ();
+        
+        public OutOfMemoryActor()
+        {
+            var log = Context.GetLogger();
+            
+            Receive<string>(_ =>
+            {
+                try
+                {
+                    foreach (var i in Enumerable.Range(0, 50_000_000))
+                    {
+                        _memoryPressure.Add(i);
+                    }
+                }
+                catch (OutOfMemoryException e)
+                {
+                    log.Error(e, "I died (Actor)");
+                    Sender.Tell("dead");
+                }
+            });
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/core/Akka.Tests/Akka.Tests.csproj b/src/core/Akka.Tests/Akka.Tests.csproj
index 2941f3a10ef..bc0ed3c34ff 100644
--- a/src/core/Akka.Tests/Akka.Tests.csproj
+++ b/src/core/Akka.Tests/Akka.Tests.csproj
@@ -5,6 +5,7 @@
     <AssemblyTitle>Akka.Tests</AssemblyTitle>
     <TargetFrameworks Condition=" '$(OS)' == 'Windows_NT' ">$(NetFrameworkTestVersion);$(NetTestVersion)</TargetFrameworks>
     <TargetFrameworks Condition=" '$(OS)' != 'Windows_NT' ">$(NetTestVersion)</TargetFrameworks>
+    <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
   </PropertyGroup>
 
   <ItemGroup>