Add tests for OR queries