summaryrefslogtreecommitdiff
path: root/src/test/java/com/amazon/carbonado/repo
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/java/com/amazon/carbonado/repo')
-rw-r--r--src/test/java/com/amazon/carbonado/repo/jdbc/H2SchemaResolver.java4
-rw-r--r--src/test/java/com/amazon/carbonado/repo/jdbc/TestH2.java31
2 files changed, 35 insertions, 0 deletions
diff --git a/src/test/java/com/amazon/carbonado/repo/jdbc/H2SchemaResolver.java b/src/test/java/com/amazon/carbonado/repo/jdbc/H2SchemaResolver.java
index 9afeb43..bc82139 100644
--- a/src/test/java/com/amazon/carbonado/repo/jdbc/H2SchemaResolver.java
+++ b/src/test/java/com/amazon/carbonado/repo/jdbc/H2SchemaResolver.java
@@ -135,6 +135,10 @@ public class H2SchemaResolver implements SchemaResolver {
b.append(" NOT NULL");
}
+ if (property.isAutomatic()) {
+ b.append(" AUTO_INCREMENT");
+ }
+
if (property.getSequenceName() != null) {
sequenceNames.add(property.getSequenceName());
}
diff --git a/src/test/java/com/amazon/carbonado/repo/jdbc/TestH2.java b/src/test/java/com/amazon/carbonado/repo/jdbc/TestH2.java
index c969838..8f2cfaa 100644
--- a/src/test/java/com/amazon/carbonado/repo/jdbc/TestH2.java
+++ b/src/test/java/com/amazon/carbonado/repo/jdbc/TestH2.java
@@ -25,6 +25,7 @@ import java.io.*;
import java.sql.DriverManager;
+import com.amazon.carbonado.stored.AutomaticAndAltKey;
import junit.framework.TestSuite;
import org.apache.commons.dbcp.BasicDataSource;
@@ -77,6 +78,36 @@ public class TestH2 extends com.amazon.carbonado.TestStorables {
}
}
+ public void test_automaticProperty() throws Exception {
+ Storage<AutomaticAndAltKey> storage =
+ getRepository().storageFor(AutomaticAndAltKey.class);
+
+ // Insert some record and expect the ID to be set automatically.
+ for (int i=1; i <= 5; i++) {
+ AutomaticAndAltKey record = storage.prepare();
+ record.setName("row" + i);
+ record.insert();
+ }
+
+ // Load by primary key
+ for (int i=1; i <= 5; i++) {
+ AutomaticAndAltKey record = storage.prepare();
+ record.setID(i);
+ assertTrue(record.tryLoad());
+ assertEquals(i, record.getID());
+ assertEquals("row" + i, record.getName());
+ }
+
+ // Load by alternate key
+ for (int i=1; i <= 5; i++) {
+ AutomaticAndAltKey record = storage.prepare();
+ record.setName("row" + i);
+ assertTrue(record.tryLoad());
+ assertEquals(i, record.getID());
+ assertEquals("row" + i, record.getName());
+ }
+ }
+
public void test_propertyOther() throws Exception {
Storage<WithPropertyOther> storage = getRepository().storageFor(WithPropertyOther.class);
WithPropertyOther other = storage.prepare();