Added tests for batch with Query DSL