48 lines
1.6 KiB
Go
48 lines
1.6 KiB
Go
package repository
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"testing"
|
||
|
||
"github.com/go-sql-driver/mysql"
|
||
"github.com/jackc/pgx/v5/pgconn"
|
||
)
|
||
|
||
type memberSQLiteCodeError int
|
||
|
||
func (e memberSQLiteCodeError) Error() string { return "测试数据库错误" }
|
||
func (e memberSQLiteCodeError) Code() int { return int(e) }
|
||
|
||
func TestProjectMemberUniqueViolationClassification(t *testing.T) {
|
||
for _, tc := range []struct {
|
||
name string
|
||
err error
|
||
want bool
|
||
}{
|
||
{"nil", nil, false},
|
||
{"untyped_constraint", errors.New("constraint failed"), false},
|
||
{"sqlite_primary", memberSQLiteCodeError(1555), true},
|
||
{"sqlite_unique", memberSQLiteCodeError(2067), true},
|
||
{"sqlite_trigger", memberSQLiteCodeError(1811), false},
|
||
{"sqlite_foreign_key", memberSQLiteCodeError(787), false},
|
||
{"sqlite_check", memberSQLiteCodeError(275), false},
|
||
{"sqlite_not_null", memberSQLiteCodeError(1299), false},
|
||
{"mysql_unique", &mysql.MySQLError{Number: 1062}, true},
|
||
{"mysql_trigger", &mysql.MySQLError{Number: 1644}, false},
|
||
{"mysql_foreign_key", &mysql.MySQLError{Number: 1452}, false},
|
||
{"postgres_unique", &pgconn.PgError{Code: "23505"}, true},
|
||
{"postgres_foreign_key", &pgconn.PgError{Code: "23503"}, false},
|
||
{"postgres_trigger", &pgconn.PgError{Code: "P0001"}, false},
|
||
} {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
if got := projectMemberUniqueViolation(tc.err); got != tc.want {
|
||
t.Fatalf("成员冲突错误分类不准确：%t，期望 %t", got, tc.want)
|
||
}
|
||
if tc.err != nil && projectMemberUniqueViolation(fmt.Errorf("包装错误：%w", tc.err)) != tc.want {
|
||
t.Fatal("包装错误后的分类不一致")
|
||
}
|
||
})
|
||
}
|
||
}
|