made MuteLog overrideable in Akka.Cluster.Testkit

Aaronontheweb · Apr 7, 2021 · 0fffc1a · 0fffc1a
1 parent eb40a29
commit 0fffc1a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/core/Akka.Cluster.TestKit/MultiNodeClusterSpec.cs b/src/core/Akka.Cluster.TestKit/MultiNodeClusterSpec.cs
@@ -175,7 +175,7 @@ protected override void AfterTermination()
 
         //TODO: ExpectedTestDuration?
 
-        void MuteLog(ActorSystem sys = null)
+        public virtual void MuteLog(ActorSystem sys = null)
         {
             if (sys == null) sys = Sys;
             if (!sys.Log.IsDebugEnabled)

diff --git a/src/core/Akka.Cluster.Tests.MultiNode/StressSpec.cs b/src/core/Akka.Cluster.Tests.MultiNode/StressSpec.cs
@@ -684,8 +684,10 @@ public class StressSpec : MultiNodeClusterSpec
         public int Step = 0;
         public int NbrUsedRoles = 0;
 
-        protected virtual void MuteLog(ActorSystem sys)
+        public override void MuteLog(ActorSystem sys = null)
         {
+            sys ??= Sys;
+            base.MuteLog(sys);
             Sys.EventStream.Publish(new Mute(new ErrorFilter(typeof(ApplicationException), new ContainsString("Simulated exception"))));
             MuteDeadLetters(sys, typeof(AggregatedClusterResult), typeof(StatsResult), typeof(PhiResult), typeof(RetryTick));
         }