From 9e00e78694669c4bd3838db9748a872371919dbe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timo=20Westk=C3=A4mper?= <timo.westkamper@mysema.com>
Date: Fri, 1 Apr 2011 06:17:48 +0000
Subject: [PATCH] added tests

---
 .../com/mysema/query/lucene/LuceneQueryTest.java  | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/querydsl-lucene/src/test/java/com/mysema/query/lucene/LuceneQueryTest.java b/querydsl-lucene/src/test/java/com/mysema/query/lucene/LuceneQueryTest.java
index f180a063d..78030164f 100644
--- a/querydsl-lucene/src/test/java/com/mysema/query/lucene/LuceneQueryTest.java
+++ b/querydsl-lucene/src/test/java/com/mysema/query/lucene/LuceneQueryTest.java
@@ -23,6 +23,7 @@ import java.util.Locale;
 import org.apache.lucene.analysis.standard.StandardAnalyzer;
 import org.apache.lucene.document.Document;
 import org.apache.lucene.document.Field;
+import org.apache.lucene.document.MapFieldSelector;
 import org.apache.lucene.document.NumericField;
 import org.apache.lucene.document.Field.Index;
 import org.apache.lucene.document.Field.Store;
@@ -413,6 +414,13 @@ public class LuceneQueryTest {
         assertNull(document.get("year"));
     }
 
+    @Test
+    public void Load_List_FieldSelector(){
+        Document document = query.where(title.ne("")).load(new MapFieldSelector("title")).list().get(0);
+        assertNotNull(document.get("title"));
+        assertNull(document.get("year"));
+    }
+
     @Test
     public void Load_SingleResult(){
         Document document = query.where(title.ne("")).load(title).singleResult();
@@ -420,6 +428,13 @@ public class LuceneQueryTest {
         assertNull(document.get("year"));
     }
 
+    @Test
+    public void Load_SingleResult_FieldSelector(){
+        Document document = query.where(title.ne("")).load(new MapFieldSelector("title")).singleResult();
+        assertNotNull(document.get("title"));
+        assertNull(document.get("year"));
+    }
+
     @Test
     public void SingleResult() {
         assertNotNull(query.where(title.ne("")).singleResult());