golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql/sqlgraph: report check constraint violation with IsCheckConstraintError (#4404)

Browse Source 
* feat(dialect): Support check constraint violation

* chore: Add unit tests for IsCheckConstraintError
This commit is contained in:
Henry Le
2025-06-20 15:11:59 +07:00
committed by GitHub
parent 86238d83fc
commit ace82484b8
2 changed files with 57 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
dialect/sql/sqlgraph/graph_test.go
View File

@@ -2569,6 +2569,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint bool
expectedFK bool
expectedUnique bool
expectedCheck bool
}{
{
name: "MySQL FK",
@@ -2578,6 +2579,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "SQLite FK",
@@ -2585,6 +2587,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "Postgres FK",
@@ -2592,6 +2595,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "MySQL FK",
@@ -2600,6 +2604,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "SQLite FK",
@@ -2607,6 +2612,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "Postgres FK",
@@ -2614,6 +2620,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: true,
expectedUnique: false,
expectedCheck: false,
},
{
name: "MySQL Unique",
@@ -2621,6 +2628,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: false,
expectedUnique: true,
expectedCheck: false,
},
{
name: "SQLite Unique",
@@ -2628,6 +2636,7 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: false,
expectedUnique: true,
expectedCheck: false,
},
{
name: "Postgres Unique",
@@ -2635,6 +2644,31 @@ func TestIsConstraintError(t *testing.T) {
expectedConstraint: true,
expectedFK: false,
expectedUnique: true,
expectedCheck: false,
},
{
name: "MySQL Check",
errMessage: `insert node to table "users": Error 3819: Check constraint 'users_age_check' is violated`,
expectedConstraint: true,
expectedFK: false,
expectedUnique: false,
expectedCheck: true,
},
{
name: "SQLite Check",
errMessage: `insert node to table "users": CHECK constraint failed: age >= 18`,
expectedConstraint: true,
expectedFK: false,
expectedUnique: false,
expectedCheck: true,
},
{
name: "Postgres Check",
errMessage: `insert node to table "users": pq: new row for relation "users" violates check constraint "users_age_check"`,
expectedConstraint: true,
expectedFK: false,
expectedUnique: false,
expectedCheck: true,
},
}
for _, tt := range tests {
@@ -2643,6 +2677,7 @@ func TestIsConstraintError(t *testing.T) {
require.EqualValues(t, tt.expectedConstraint, IsConstraintError(err))
require.EqualValues(t, tt.expectedFK, IsForeignKeyConstraintError(err))
require.EqualValues(t, tt.expectedUnique, IsUniqueConstraintError(err))
require.EqualValues(t, tt.expectedCheck, IsCheckConstraintError(err))
})
}
}