Started to write tests for QueryWrapper