1 files changed, 9 insertions, 2 deletions
diff --git a/container-compiler-plugin/src/test/java/io/trygvis/persistence/EntityMirrorTest.java b/container-compiler-plugin/src/test/java/io/trygvis/persistence/EntityMirrorTest.java
index 52b33ca..85e3741 100644
--- a/container-compiler-plugin/src/test/java/io/trygvis/persistence/EntityMirrorTest.java
+++ b/container-compiler-plugin/src/test/java/io/trygvis/persistence/EntityMirrorTest.java
@@ -71,7 +71,7 @@ public class EntityMirrorTest {
             myTable.add(field);
         }
 
-        SqlUnitModel unit = new SqlUnitModel().add(myTable).add(new SequenceMirror("id_seq"));
+        SqlUnitModel unit = new SqlUnitModel().add(myTable).add(new SequenceMirror("seq-gen", "id_seq", 0, 0));
 
         assertEquals(myTable.insertIntoSql(unit), insert);
         assertEquals(myTable.deleteFromSql(), delete);
@@ -92,9 +92,16 @@ public class EntityMirrorTest {
                         "    stmt.executeUpdate();",
                         "}")
                 },
-                new Object[]{new FieldMirror[]{age}, join(
+                new Object[]{new FieldMirror[]{year}, join(
+                        "try(java.sql.PreparedStatement stmt = con.prepareStatement(insertIntoSql)) {",
+                        "    stmt.setInt(1, o.getYear());",
+                        "    stmt.executeUpdate();",
+                        "}")
+                },
+                new Object[]{new FieldMirror[]{age, year, idLong}, join(
                         "try(java.sql.PreparedStatement stmt = con.prepareStatement(insertIntoSql)) {",
                         "    stmt.setInt(1, o.age);",
+                        "    stmt.setInt(2, o.getYear());",
                         "    stmt.executeUpdate();",
                         "}")
                 },