130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type City struct {
|
|
gorm.Model
|
|
Name string `gorm:"unique"`
|
|
}
|
|
|
|
func TestPostgresErrorTranslator(t *testing.T) {
|
|
if DB.Dialector.Name() != "postgres" {
|
|
t.Skip()
|
|
}
|
|
|
|
DB.Migrator().DropTable(&City{})
|
|
|
|
if err := DB.AutoMigrate(&City{}); err != nil {
|
|
t.Fatalf("Failed to migrate cities table, got error: %v", err)
|
|
}
|
|
|
|
err := DB.Create(&City{Name: "Amsterdam"}).Error
|
|
if err != nil {
|
|
t.Fatalf("errors happened on create: %v", err)
|
|
}
|
|
|
|
// test errors to be translated
|
|
err = DB.Create(&City{Name: "Amsterdam"}).Error
|
|
if !errors.Is(err, gorm.ErrDuplicatedKey) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrDuplicatedKey, err)
|
|
}
|
|
|
|
// test default errors to not be translated
|
|
err = DB.Where("name = ?", "Kabul").First(&City{}).Error
|
|
if err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrRecordNotFound, err)
|
|
}
|
|
}
|
|
|
|
func TestMysqlErrorTranslator(t *testing.T) {
|
|
if DB.Dialector.Name() != "mysql" {
|
|
t.Skip()
|
|
}
|
|
|
|
DB.Migrator().DropTable(&City{})
|
|
|
|
if err := DB.AutoMigrate(&City{}); err != nil {
|
|
t.Fatalf("Failed to migrate cities table, got error: %v", err)
|
|
}
|
|
|
|
err := DB.Create(&City{Name: "Berlin"}).Error
|
|
if err != nil {
|
|
t.Fatalf("errors happened on create: %v", err)
|
|
}
|
|
|
|
// test errors to be translated
|
|
err = DB.Create(&City{Name: "Berlin"}).Error
|
|
if !errors.Is(err, gorm.ErrDuplicatedKey) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrDuplicatedKey, err)
|
|
}
|
|
|
|
// test default errors to not be translated
|
|
err = DB.Where("name = ?", "Istanbul").First(&City{}).Error
|
|
if err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrRecordNotFound, err)
|
|
}
|
|
}
|
|
|
|
func TestMssqlErrorTranslator(t *testing.T) {
|
|
if DB.Dialector.Name() != "mssql" {
|
|
t.Skip()
|
|
}
|
|
|
|
DB.Migrator().DropTable(&City{})
|
|
|
|
if err := DB.AutoMigrate(&City{}); err != nil {
|
|
t.Fatalf("Failed to migrate cities table, got error: %v", err)
|
|
}
|
|
|
|
err := DB.Create(&City{Name: "Paris"}).Error
|
|
if err != nil {
|
|
t.Fatalf("errors happened on create: %v", err)
|
|
}
|
|
|
|
// test errors to be translated
|
|
err = DB.Create(&City{Name: "Paris"}).Error
|
|
if !errors.Is(err, gorm.ErrDuplicatedKey) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrDuplicatedKey, err)
|
|
}
|
|
|
|
// test default errors to not be translated
|
|
err = DB.Where("name = ?", "Prague").First(&City{}).Error
|
|
if err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrRecordNotFound, err)
|
|
}
|
|
}
|
|
|
|
func TestSqliteErrorTranslator(t *testing.T) {
|
|
if DB.Dialector.Name() != "sqlite" {
|
|
t.Skip()
|
|
}
|
|
|
|
DB.Migrator().DropTable(&City{})
|
|
|
|
if err := DB.AutoMigrate(&City{}); err != nil {
|
|
t.Fatalf("Failed to migrate cities table, got error: %v", err)
|
|
}
|
|
|
|
err := DB.Create(&City{Name: "Madrid"}).Error
|
|
if err != nil {
|
|
t.Fatalf("errors happened on create: %v", err)
|
|
}
|
|
|
|
// test errors to be translated
|
|
err = DB.Create(&City{Name: "Madrid"}).Error
|
|
if !errors.Is(err, gorm.ErrDuplicatedKey) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrDuplicatedKey, err)
|
|
}
|
|
|
|
// test default errors to not be translated
|
|
err = DB.Where("name = ?", "Rome").First(&City{}).Error
|
|
if err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
t.Fatalf("expected err: %v got err: %v", gorm.ErrRecordNotFound, err)
|
|
}
|
|
}
|