Add mock test for batching (not working right now)