Add test for consumer