Add tests for Batch Extension.