fix some more tests

hughsimpson · Nov 7, 2023 · 2e49049 · 2e49049
1 parent 7811241
commit 2e49049
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/...ko/src/main/scala/kamon/instrumentation/pekko/instrumentations/ActorInstrumentation.scala b/...ko/src/main/scala/kamon/instrumentation/pekko/instrumentations/ActorInstrumentation.scala
@@ -32,6 +32,11 @@ class ActorInstrumentation extends InstrumentationBuilder {
 
   onType("org.apache.pekko.actor.dungeon.Dispatch")
     .advise(method("sendMessage").and(takesArguments(1)), classOf[SendMessageAdvice])
+    .advise(method("swapMailbox"), classOf[ActorCellSwapMailboxAdvice])
+
+  onType("org.apache.pekko.actor.dungeon.FaultHandling")
+    .advise(method("handleInvokeFailure"), classOf[HandleInvokeFailureMethodAdvice])
+    .advise(method("terminate"), classOf[TerminateMethodAdvice])
 
   /**
    * This is where most of the Actor processing magic happens. Handling of messages, errors and system messages.
@@ -40,10 +45,6 @@ class ActorInstrumentation extends InstrumentationBuilder {
     .mixin(classOf[HasActorMonitor.Mixin])
     .advise(isConstructor, classOf[ActorCellConstructorAdvice])
     .advise(method("invoke"), classOf[ActorCellInvokeAdvice])
-    .advise(method("handleInvokeFailure"), classOf[HandleInvokeFailureMethodAdvice])
-    .advise(method("sendMessage").and(takesArguments(1)), classOf[SendMessageAdvice])
-    .advise(method("terminate"), classOf[TerminateMethodAdvice])
-    .advise(method("swapMailbox"), classOf[ActorCellSwapMailboxAdvice])
     .advise(method("invokeAll$1"), classOf[InvokeAllMethodInterceptor])
 
   /**