flushChanges & delete throw SQLException

kotlin-orm · Jul 8, 2024 · 46e18c9 · 46e18c9
1 parent fb0c257
commit 46e18c9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/ktorm-core/src/main/kotlin/org/ktorm/entity/Entity.kt b/ktorm-core/src/main/kotlin/org/ktorm/entity/Entity.kt
@@ -23,6 +23,7 @@ import java.io.ObjectInputStream
 import java.io.ObjectOutputStream
 import java.io.Serializable
 import java.lang.reflect.Proxy
+import java.sql.SQLException
 import kotlin.reflect.KClass
 import kotlin.reflect.full.isSubclassOf
 import kotlin.reflect.jvm.jvmErasure
@@ -168,6 +169,7 @@ public interface Entity<E : Entity<E>> : Serializable {
      * @see add
      * @see update
      */
+    @Throws(SQLException::class)
     public fun flushChanges(): Int
 
     /**
@@ -191,6 +193,7 @@ public interface Entity<E : Entity<E>> : Serializable {
      * @see update
      * @see flushChanges
      */
+    @Throws(SQLException::class)
     public fun delete(): Int
 
     /**

diff --git a/ktorm-core/src/test/kotlin/org/ktorm/entity/EntityTest.kt b/ktorm-core/src/test/kotlin/org/ktorm/entity/EntityTest.kt
@@ -1,5 +1,6 @@
 package org.ktorm.entity
 
+import org.h2.jdbc.JdbcSQLIntegrityConstraintViolationException
 import org.junit.Test
 import org.ktorm.BaseTest
 import org.ktorm.database.Database
@@ -397,6 +398,19 @@ class EntityTest : BaseTest() {
         assert(e.flushChanges() == 1)
     }
 
+    @Test
+    fun testExceptionThrowsByProxy() {
+        try {
+            val e = database.employees.find { it.id eq 1 } ?: throw AssertionError()
+            e.department = Department()
+            e.flushChanges()
+
+            throw AssertionError("failed")
+        } catch (e: JdbcSQLIntegrityConstraintViolationException) {
+            assert(e.message!!.contains("NULL not allowed for column \"department_id\""))
+        }
+    }
+
     @Test
     fun testHasColumnValue() {
         val p1 = Parent()