apache · jolshan · Jan 26, 2024 · Nov 28, 2023 · Nov 29, 2023 · Dec 5, 2023
diff --git a/checkstyle/import-control.xml b/checkstyle/import-control.xml
@@ -333,6 +333,7 @@
 
     <subpackage name="consumer.group">
       <allow pkg="org.apache.kafka.tools"/>
+      <allow pkg="scala"/>
     </subpackage>
 
     <subpackage name="other">

diff --git a/core/src/main/scala/kafka/admin/ConsumerGroupCommand.scala b/core/src/main/scala/kafka/admin/ConsumerGroupCommand.scala
@@ -125,7 +125,7 @@ object ConsumerGroupCommand extends Logging {
     }
   }
 
-  private[admin] case class PartitionAssignmentState(group: String, coordinator: Option[Node], topic: Option[String],
+  case class PartitionAssignmentState(group: String, coordinator: Option[Node], topic: Option[String],
                                                 partition: Option[Int], offset: Option[Long], lag: Option[Long],
                                                 consumerId: Option[String], host: Option[String],
                                                 clientId: Option[String], logEndOffset: Option[Long])

diff --git a/core/src/test/scala/unit/kafka/admin/ConsumerGroupServiceTest.scala b/core/src/test/scala/unit/kafka/admin/ConsumerGroupServiceTest.scala