Added test for batch messages, change the count to equals