| 8 | ) |
| 9 | |
| 10 | func TestWithSingleConnection(t *testing.T) { |
| 11 | expectedName := "test" |
| 12 | var actualName string |
| 13 | |
| 14 | setSQL, getSQL := getSetSQL(DB.Dialector.Name()) |
| 15 | if len(setSQL) == 0 || len(getSQL) == 0 { |
| 16 | return |
| 17 | } |
| 18 | |
| 19 | err := DB.Connection(func(tx *gorm.DB) error { |
| 20 | if err := tx.Exec(setSQL, expectedName).Error; err != nil { |
| 21 | return err |
| 22 | } |
| 23 | |
| 24 | if err := tx.Raw(getSQL).Scan(&actualName).Error; err != nil { |
| 25 | return err |
| 26 | } |
| 27 | return nil |
| 28 | }) |
| 29 | if err != nil { |
| 30 | t.Errorf("WithSingleConnection should work, but got err %v", err) |
| 31 | } |
| 32 | |
| 33 | if actualName != expectedName { |
| 34 | t.Errorf("WithSingleConnection() method should get correct value, expect: %v, got %v", expectedName, actualName) |
| 35 | } |
| 36 | } |
| 37 | |
| 38 | func getSetSQL(driverName string) (string, string) { |
| 39 | switch driverName { |