test #405 for mysql

Author: vincentlauvlwj

ktorm-support-mysql/src/test/kotlin/org/ktorm/support/mysql/CommonTest.kt

@@ -14,6 +14,7 @@ import java.util.concurrent.ExecutionException
 import java.util.concurrent.Executors
 import java.util.concurrent.TimeUnit
 import java.util.concurrent.TimeoutException
+import kotlin.test.assertEquals
 
 /**
  * Created by vince on Dec 12, 2018.
@@ -532,4 +533,29 @@ class CommonTest : BaseMySqlTest() {
             .first()
         assertThat(mood1, equalTo(null))
     }
+
+    interface TestMultiGeneratedKey : Entity<TestMultiGeneratedKey> {
+        var id: Int
+        var k: String
+        var v: String
+    }
+
+    object TestMultiGeneratedKeys : Table<TestMultiGeneratedKey>("t_multi_generated_key") {
+        val id = int("id").primaryKey().bindTo { it.id }
+        val k = varchar("k").bindTo { it.k }
+        val v = varchar("v").bindTo { it.v }
+    }
+
+    @Test
+    fun testMultiGeneratedKey() {
+        val e = Entity.create<TestMultiGeneratedKey>()
+        e.v = "test~~"
+        database.sequenceOf(TestMultiGeneratedKeys).add(e)
+
+        val e1 = database.sequenceOf(TestMultiGeneratedKeys).first()
+        println(e1)
+        assertEquals(1, e1.id)
+        assertEquals("test~~", e1.v)
+        assert(e1.k.isNotEmpty())
+    }
 }

ktorm-support-mysql/src/test/resources/drop-mysql-data.sql

@@ -1,3 +1,4 @@
 drop table if exists t_department;
 drop table if exists t_employee;
 drop table if exists t_user;
+drop table if exists t_multi_generated_key;

ktorm-support-mysql/src/test/resources/init-mysql-data.sql

@@ -23,6 +23,12 @@ create table t_user(
    age int
 );
 
+create table t_multi_generated_key(
+    id int not null primary key auto_increment,
+    k varchar(128) default (uuid()),
+    v varchar(128)
+);
+
 insert into t_department(name, location) values ('tech', 'Guangzhou');
 insert into t_department(name, location) values ('finance', 'Beijing');
 

ktorm-support-postgresql/src/test/kotlin/org/ktorm/support/postgresql/CommonTest.kt

@@ -14,6 +14,7 @@ import java.util.concurrent.ExecutionException
 import java.util.concurrent.Executors
 import java.util.concurrent.TimeUnit
 import java.util.concurrent.TimeoutException
+import kotlin.test.assertEquals
 
 /**
  * Created by vince on Feb 13, 2019.
@@ -212,5 +213,11 @@ class CommonTest : BasePostgreSqlTest() {
         val e = Entity.create<TestMultiGeneratedKey>()
         e.v = "test~~"
         database.sequenceOf(TestMultiGeneratedKeys).add(e)
+
+        val e1 = database.sequenceOf(TestMultiGeneratedKeys).first()
+        println(e1)
+        assertEquals(1, e1.id)
+        assertEquals("test~~", e1.v)
+        assert(e1.k.isNotEmpty())
     }
 }