| 451 | } |
| 452 | |
| 453 | func TestTransactionWithHooks(t *testing.T) { |
| 454 | user := GetUser("tTestTransactionWithHooks", Config{Account: true}) |
| 455 | DB.Create(&user) |
| 456 | |
| 457 | var err error |
| 458 | err = DB.Transaction(func(tx *gorm.DB) error { |
| 459 | return tx.Model(&User{}).Limit(1).Transaction(func(tx2 *gorm.DB) error { |
| 460 | return tx2.Scan(&User{}).Error |
| 461 | }) |
| 462 | }) |
| 463 | if err != nil { |
| 464 | t.Error(err) |
| 465 | } |
| 466 | |
| 467 | // method with hooks |
| 468 | err = DB.Transaction(func(tx1 *gorm.DB) error { |
| 469 | // callMethod do |
| 470 | tx2 := tx1.Find(&User{}).Session(&gorm.Session{NewDB: true}) |
| 471 | // trx in hooks |
| 472 | return tx2.Transaction(func(tx3 *gorm.DB) error { |
| 473 | return tx3.Where("user_id", user.ID).Delete(&Account{}).Error |
| 474 | }) |
| 475 | }) |
| 476 | if err != nil { |
| 477 | t.Error(err) |
| 478 | } |
| 479 | } |
| 480 | |
| 481 | func TestTransactionWithDefaultTimeout(t *testing.T) { |
| 482 | db, err := OpenTestConnection(&gorm.Config{DefaultTransactionTimeout: 2 * time.Second}) |