Refactor mock a bit
Refactor mock a bit

aybabtme/sarama,VividCortex/sarama,vrischmann/sarama,DataDog/sarama,cep21/sarama,raisemarketplace/sarama,winlinvip/sarama,rafrombrc/sarama,rafrombrc/sarama,winlinvip/sarama,Neuw84/sarama,remerge/sarama,LastManStanding/sarama,jepst/sarama,DataDog/sarama,remerge/sarama,jimmykuo/sarama,pronix/sarama,kraveio/sarama,mailgun/sarama,crast/sarama,kraveio/sarama,Shopify/sarama,scalingdata/sarama,kmiku7/sarama,betable/sarama,HelixEducation/sarama,mailgun/sarama,yangfan876/sarama,pronix/sarama,kmiku7/sarama,vrischmann/sarama,cep21/sarama,gyroscope-innovations/legacy_sarama,xiaotianlis/sarama,scalingdata/sarama,Shopify/sarama,Neuw84/sarama,LastManStanding/sarama,jimmykuo/sarama,crast/sarama,jepst/sarama,VividCortex/sarama,sclasen/sarama,gyroscope-innovations/legacy_sarama,uovobw/sarama,HelixEducation/sarama,betable/sarama,aybabtme/sarama,raisemarketplace/sarama,uovobw/sarama,yangfan876/sarama