simpler test

apache · Aug 23, 2024 · 0c76ad9 · 0c76ad9
1 parent d107917
commit 0c76ad9
Showing 1 changed file with 12 additions and 28 deletions.
diff --git a/actor-typed-tests/src/test/scala/org/apache/pekko/actor/typed/BehaviorSpec.scala b/actor-typed-tests/src/test/scala/org/apache/pekko/actor/typed/BehaviorSpec.scala
@@ -379,34 +379,18 @@ class ReceiveBehaviorSpec extends Messages with BecomeWithLifecycle with Stoppab
 }
 
 class ReceiveMessageWithSameBehaviorSpec extends Messages {
-  override def behavior(monitor: ActorRef[Event]): (Behavior[Command], Aux) = behv(monitor, StateA) -> null
-  private def behv(monitor: ActorRef[Event], state: State): Behavior[Command] = {
-    SBehaviors.setup[Command] { context =>
-      SBehaviors
-        .receiveMessageWithSame[Command] {
-          case GetSelf =>
-            monitor ! Self(context.self)
-          case Miss =>
-            monitor ! Missed
-          case Ignore =>
-            monitor ! Ignored
-          case Ping =>
-            monitor ! Pong
-            behv(monitor, state)
-          case Swap =>
-            monitor ! Swapped
-            behv(monitor, state.next)
-          case GetState() =>
-            monitor ! state
-          case Stop       => SBehaviors.stopped
-          case _: AuxPing => SBehaviors.unhandled
-        }
-        .receiveSignal {
-          case (_, signal) =>
-            monitor ! ReceivedSignal(signal)
-            SBehaviors.same
-        }
-    }
+  override def behavior(monitor: ActorRef[Event]): (Behavior[Command], Aux) = behv(monitor) -> null
+  private def behv(monitor: ActorRef[Event]): Behavior[Command] = {
+    SBehaviors
+      .receiveMessageWithSame[Command] {
+        case Miss =>
+          monitor ! Missed
+        case Ignore =>
+          monitor ! Ignored
+        case Ping =>
+          monitor ! Pong
+        case _ =>
+      }
   }
 }