Allow selecting a column from a table

2 files changed, 32 insertions, 0 deletions

diff --git a/src/test/java/com/iciql/test/ModelsTest.java b/src/test/java/com/iciql/test/ModelsTest.java
index ff27bf5..a5474df 100644
--- a/src/test/java/com/iciql/test/ModelsTest.java
+++ b/src/test/java/com/iciql/test/ModelsTest.java
@@ -160,4 +160,19 @@ public class ModelsTest {
 		SupportedTypes s2 = db.from(SupportedTypes.SAMPLE).where(SupportedTypes.SAMPLE.id).is(1).selectFirst();
 		assertEquals(40, s2.myString.length());
 	}
+
+	@Test
+	public void testColumnSelection() {
+		List<SupportedTypes> original = SupportedTypes.createList();
+		db.insertAll(original);
+		List<String> myStrings = db.from(SupportedTypes.SAMPLE)
+				.select(SupportedTypes.SAMPLE.myString);
+		assertEquals(10, myStrings.size());
+
+		List<Integer> ids = db.from(SupportedTypes.SAMPLE)
+				.orderByDesc(SupportedTypes.SAMPLE.id)
+				.selectDistinct(SupportedTypes.SAMPLE.id);
+		assertEquals(10, ids.size());
+		assertEquals("[9, 8, 7, 6, 5, 4, 3, 2, 1, 0]", ids.toString());
+	}
 }
diff --git a/src/test/java/com/iciql/test/PrimitivesTest.java b/src/test/java/com/iciql/test/PrimitivesTest.java
index 0e8bf20..938fd32 100644
--- a/src/test/java/com/iciql/test/PrimitivesTest.java
+++ b/src/test/java/com/iciql/test/PrimitivesTest.java
@@ -99,4 +99,21 @@ public class PrimitivesTest {
 		}

 		db.close();

 	}

+

+	@Test

+	public void testPrimitiveColumnSelection() {

+		Db db = IciqlSuite.openNewDb();

+

+		// insert random models in reverse order

+		List<PrimitivesModel> models = PrimitivesModel.getList();

+		PrimitivesModel model = models.get(0);

+		Collections.reverse(models);

+		// insert them in reverse order

+		db.insertAll(models);

+

+		PrimitivesModel p = new PrimitivesModel();

+		List<Long> list = db.from(p)/*.orderByDesc(p.myLong)*/.select(p.myLong);

+		assertEquals(models.size(), list.size());

+		assertEquals("[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]", list.toString());

+	}

 }