From e6cdb01aa14e97f91cf550017d33d9114a36b584 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johan=20Andr=C3=A9n?= <johan@markatta.com>
Date: Wed, 29 May 2019 08:29:49 +0200
Subject: [PATCH] Delegate stop method on TestKitJunitResource  #27007

---
 .../testkit/typed/javadsl/TestKitJunitResource.scala | 12 ++++++++++++
 .../testkit/typed/javadsl/BehaviorTestKitTest.java   |  2 ++
 2 files changed, 14 insertions(+)

diff --git a/akka-actor-testkit-typed/src/main/scala/akka/actor/testkit/typed/javadsl/TestKitJunitResource.scala b/akka-actor-testkit-typed/src/main/scala/akka/actor/testkit/typed/javadsl/TestKitJunitResource.scala
index 8b4eac6922..da89ae0110 100644
--- a/akka-actor-testkit-typed/src/main/scala/akka/actor/testkit/typed/javadsl/TestKitJunitResource.scala
+++ b/akka-actor-testkit-typed/src/main/scala/akka/actor/testkit/typed/javadsl/TestKitJunitResource.scala
@@ -4,6 +4,8 @@
 
 package akka.actor.testkit.typed.javadsl
 
+import java.time.Duration
+
 import akka.actor.testkit.typed.TestKitSettings
 import akka.actor.testkit.typed.internal.TestKitUtils
 import akka.actor.typed.ActorRef
@@ -127,6 +129,16 @@ final class TestKitJunitResource(_kit: ActorTestKit) extends ExternalResource {
    */
   def createTestProbe[M](name: String): TestProbe[M] = testKit.createTestProbe(name)
 
+  /**
+   * See corresponding method on [[ActorTestKit]]
+   */
+  def stop[T](ref: ActorRef[T], max: Duration): Unit = testKit.stop(ref, max)
+
+  /**
+   * See corresponding method on [[ActorTestKit]]
+   */
+  def stop[T](ref: ActorRef[T]): Unit = testKit.stop(ref)
+
   override def after(): Unit = {
     testKit.shutdownTestKit()
   }
diff --git a/akka-actor-testkit-typed/src/test/java/akka/actor/testkit/typed/javadsl/BehaviorTestKitTest.java b/akka-actor-testkit-typed/src/test/java/akka/actor/testkit/typed/javadsl/BehaviorTestKitTest.java
index d1f0215c83..184b17fd00 100644
--- a/akka-actor-testkit-typed/src/test/java/akka/actor/testkit/typed/javadsl/BehaviorTestKitTest.java
+++ b/akka-actor-testkit-typed/src/test/java/akka/actor/testkit/typed/javadsl/BehaviorTestKitTest.java
@@ -72,6 +72,7 @@ public class BehaviorTestKitTest extends JUnitSuite {
     private final Class<Object> clazz;
     private final Function<Object, Command> f;
 
+    @SuppressWarnings("unchecked")
     public CreateMessageAdapter(Class clazz, Function<Object, Command> f) {
       this.clazz = clazz;
       this.f = f;
@@ -314,6 +315,7 @@ public class BehaviorTestKitTest extends JUnitSuite {
     BehaviorTestKit<Command> test = BehaviorTestKit.create(behavior);
     SpawnChildren adaptedMessage = new SpawnChildren(1);
     test.run(new CreateMessageAdapter(String.class, o -> adaptedMessage));
+    @SuppressWarnings("unchecked")
     Effect.MessageAdapter<String, SpawnChildren> mAdapter =
         test.<Effect.MessageAdapter>expectEffectClass(Effect.MessageAdapter.class);
     assertEquals(String.class, mAdapter.messageClass());