@Override
    public void afterAll() {
        if (TEST_STREAM != null) {
            try {
                TEST_STREAM.shutdown();
            } catch (StreamsException ignore) {
            }
        }

        if (container != null) {
            container.stop();
        }
    }
Copy of https://github.com/chbatey/kafka-unit/blob/master/src/main/java/info/batey/kafka/unit/KafkaUnit.java
Some codes have been modified for testing from the copied code.
agent-module/plugins-it/kafka-it/kafka-3-it/src/test/java/test/pinpoint/plugin/kafka/KafkaStreamsUnitServer.java