Do we have Playbook to confirm preferred leader doesn t mismatch in Kafka after server is rebooted

Question

I'm working on creating playbook for kafka rolling restarts..

I am able to get rolling restarts working, but would like also to verify that messages are flowing after node1 is restarted in cluster and similarly for each node on-by-one in cluster.

I thought of having Leader match with Replicas first value as we have prefered leader enabled I guess in configurations.

Does anyone know how to verify each partition Leader of specific topic match with first replica using shell command?

I hope my question is clear and let me know if needed more information.