Surround

sky-uk · Aug 30, 2023 · 6023f24 · 6023f24
1 parent 55e77c4
commit 6023f24
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/it/src/test/scala/integration/LoadExampleIntSpec.scala b/it/src/test/scala/integration/LoadExampleIntSpec.scala
@@ -83,6 +83,6 @@ class LoadExampleIntSpec extends KafkaSpecBase[IO] {
   private def withKafkaContext(test: TestContext[IO] => IO[Assertion]): IO[Assertion] = {
     object testContext extends TestContext[IO]
     import testContext.*
-    embeddedKafka.use(_ => test(testContext))
+    embeddedKafka.surround(test(testContext))
   }
 }
diff --git a/src/test/scala/integration/TopicLoaderIntSpec.scala b/src/test/scala/integration/TopicLoaderIntSpec.scala
@@ -197,7 +197,7 @@ class TopicLoaderIntSpec extends KafkaSpecBase[IO] {
         assertion  <- loadAndRunR(NonEmptyList.one(testTopic1))(
                         _ => loadState.set(true),
                         r => topicState.getAndUpdate(_ :+ r).void
-                      ).use { _ =>
+                      ).surround {
                         for {
                           _         <- eventually(topicState.get.asserting(_ should contain theSameElementsAs preLoad))
                           _         <- loadState.get.asserting(_ shouldBe true)
@@ -221,6 +221,6 @@ class TopicLoaderIntSpec extends KafkaSpecBase[IO] {
   private def withKafkaContext(test: TestContext[IO] => IO[Assertion]): IO[Assertion] = {
     object testContext extends TestContext[IO]
     import testContext.*
-    embeddedKafka.use(_ => test(testContext))
+    embeddedKafka.surround(test(testContext))
   }
 }