test: Allow any key from the batch we just wrote