Add test for default column names.

As per @harward's excellent commit: f803f1a
pardom-zz · Apr 6, 2014 · 72d4a4c · 72d4a4c
1 parent 87bb981
commit 72d4a4c
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/tests/src/com/activeandroid/test/MockModel.java b/tests/src/com/activeandroid/test/MockModel.java
@@ -17,8 +17,22 @@
  */
 
 import com.activeandroid.Model;
+import com.activeandroid.annotation.Column;
 import com.activeandroid.annotation.Table;
 
+import java.util.Date;
+
 @Table(name = "MockModel")
 public class MockModel extends Model {
+    @Column
+    public Date dateField;
+
+    @Column
+    public double doubleField;
+
+    @Column
+    public int intField;
+
+    @Column
+    public boolean booleanField;
 }
diff --git a/tests/src/com/activeandroid/test/ModelTest.java b/tests/src/com/activeandroid/test/ModelTest.java
@@ -16,9 +16,12 @@
 
 package com.activeandroid.test;
 
+import com.activeandroid.Cache;
 import com.activeandroid.Model;
+import com.activeandroid.TableInfo;
 import com.activeandroid.annotation.Table;
 
+import java.lang.reflect.Field;
 import java.util.HashSet;
 import java.util.Set;
 
@@ -125,6 +128,20 @@ public void testHashCodeDifferentRows() {
 		assertEquals(2, set.size());
 	}
 
+    /**
+     * Column names should default to the field name.
+     */
+    public void testColumnNamesDefaulToFieldNames() {
+        TableInfo tableInfo = Cache.getTableInfo(MockModel.class);
+
+        for ( Field field : tableInfo.getFields() ) {
+            // Id column is a special case, we'll ignore that one.
+            if ( field.getName().equals("mId") ) continue;
+
+            assertEquals(field.getName(), tableInfo.getColumnName(field));
+        }
+    }
+
 	/**
 	 * Mock model as we need 2 different model classes.
 	 */