From c76b0fc9dbb7ef63d2b3b49ae547915066ce377a Mon Sep 17 00:00:00 2001
From: Kai Moritz <kai@juplo.de>
Date: Wed, 13 Sep 2023 22:04:29 +0200
Subject: [PATCH] WIP

---
 .../implementation/kafka/KafkaChatHomeServiceTest.java        | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/test/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaChatHomeServiceTest.java b/src/test/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaChatHomeServiceTest.java
index a18de256..6e6fcb14 100644
--- a/src/test/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaChatHomeServiceTest.java
+++ b/src/test/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaChatHomeServiceTest.java
@@ -55,13 +55,15 @@ public class KafkaChatHomeServiceTest extends ChatHomeServiceWithShardsTest
   static class KafkaChatHomeTestConfiguration
   {
     @Bean
-    KafkaServicesApplicationRunner.WorkAssignor workAssignor()
+    KafkaServicesApplicationRunner.WorkAssignor workAssignor(
+        ChatRoomChannel chatRoomChannel)
     {
       return consumer ->
       {
         List<TopicPartition> assignedPartitions =
             List.of(new TopicPartition(TOPIC, 2));
         consumer.assign(assignedPartitions);
+        chatRoomChannel.onPartitionsAssigned(assignedPartitions);
       };
     }
 
-- 
2.20.1