From bd9ecc64ad2b00aecfad0b97a21fdd23ca95f4fd Mon Sep 17 00:00:00 2001
From: GreyPlane <31082046+GreyPlane@users.noreply.github.com>
Date: Wed, 24 Aug 2022 14:54:52 +0800
Subject: [PATCH] fix: using createUnhandledMessageProbe to avoid message
 delivered before subscription (#31498)

---
 .../akka/actor/typed/scaladsl/ActorContextDelegateSpec.scala  | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/akka-actor-typed-tests/src/test/scala/akka/actor/typed/scaladsl/ActorContextDelegateSpec.scala b/akka-actor-typed-tests/src/test/scala/akka/actor/typed/scaladsl/ActorContextDelegateSpec.scala
index fbd8e9c83b..c404754ab1 100644
--- a/akka-actor-typed-tests/src/test/scala/akka/actor/typed/scaladsl/ActorContextDelegateSpec.scala
+++ b/akka-actor-typed-tests/src/test/scala/akka/actor/typed/scaladsl/ActorContextDelegateSpec.scala
@@ -7,7 +7,6 @@ package akka.actor.typed.scaladsl
 import akka.actor.UnhandledMessage
 import akka.actor.testkit.typed.TestKitSettings
 import akka.actor.testkit.typed.scaladsl.{ FishingOutcomes, LogCapturing, ScalaTestWithActorTestKit, TestProbe }
-import akka.actor.typed.eventstream.EventStream
 import akka.actor.typed.{ ActorRef, Behavior }
 import org.scalatest.wordspec.AnyWordSpecLike
 
@@ -71,8 +70,7 @@ class ActorContextDelegateSpec extends ScalaTestWithActorTestKit with AnyWordSpe
     }
 
     "publish unhandled message to eventStream as UnhandledMessage and switch to delegator behavior" in {
-      val deadLetters = TestProbe[UnhandledMessage]("probeDeadLetters")
-      system.eventStream ! EventStream.Subscribe[UnhandledMessage](deadLetters.ref)
+      val deadLetters = testKit.createUnhandledMessageProbe()
 
       val probe = TestProbe[Event]()
       val behv = Behaviors.setup[PingPongCommand] { implicit context =>