/
githubmirror
/
gorm
Обзор
Документация
Войти
/
githubmirror
/
gorm
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/connection_test.go
46 строк
963 B
li-jin-gou
fix: omit not work when use join (#5034)
28 янв 2022, 17:16
Не верифицирован
28 янв 2022, 17:16
c0bea44
Код
Авторство
О чём код?
package tests_test import ( "fmt" "testing" "gorm.io/driver/mysql" "gorm.io/gorm" ) func TestWithSingleConnection(t *testing.T) { expectedName := "test" var actualName string setSQL, getSQL := getSetSQL(DB.Dialector.Name()) if len(setSQL) == 0 || len(getSQL) == 0 { return } err := DB.Connection(func(tx *gorm.DB) error { if err := tx.Exec(setSQL, expectedName).Error; err != nil { return err } if err := tx.Raw(getSQL).Scan(&actualName).Error; err != nil { return err } return nil }) if err != nil { t.Errorf(fmt.Sprintf("WithSingleConnection should work, but got err %v", err)) } if actualName != expectedName { t.Errorf("WithSingleConnection() method should get correct value, expect: %v, got %v", expectedName, actualName) } } func getSetSQL(driverName string) (string, string) { switch driverName { case mysql.Dialector{}.Name(): return "SET @testName := ?", "SELECT @testName" default: return "", "" } }