Update migrator tests

This commit is contained in:
TeoV
2018-04-27 11:19:59 -04:00
committed by Dan Christian Bogos
parent 858e337c39
commit 7c8f91ee3d
4 changed files with 31 additions and 28 deletions

View File

@@ -759,9 +759,12 @@ func (self *SQLStorage) SetSMCost(smc *SMCost) error {
func (self *SQLStorage) RemoveSMCost(smc *SMCost) error {
tx := self.db.Begin()
if err := tx.Where(&SessionsCostsSQL{Cgrid: smc.CGRID,
RunID: smc.RunID}).Delete(SessionsCostsSQL{}).Error; err != nil {
var rmParam *SessionsCostsSQL
if smc != nil {
rmParam = &SessionsCostsSQL{Cgrid: smc.CGRID,
RunID: smc.RunID}
}
if err := tx.Where(rmParam).Delete(SessionsCostsSQL{}).Error; err != nil {
tx.Rollback()
return err
}