func save(db *gorm.DB, data []types) err {
  tx := db.Begin()

  err := tx.Save(data).Error
  if err != nil {
    tx.Roolback()
    return err
  }

  tx.Commit()

  return nil
}

func TestSave(t *testing.T) {
  t.Run("success", func(t *testing.T) {
    db, err := gorm.Open("postgres", databaseURL)
    tx := db.Begin()
    teardown := func() {
      err := tx.Rollback().Error
      err = db.Close()
    }
    t.Cleanup(teardown)
    
    got, _ := db.save(deviceID)
    
    assert.Equal(t, want, got)
  })
}