uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-12 02:26:26 +05:00
Code Issues Packages Projects Releases Wiki Activity

ApierV2.RemTP

Browse Source
This commit is contained in:
DanB
2014-10-09 14:11:58 +02:00
parent bafb0b0eb8
commit 6c02605cfd
2 changed files with 59 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
engine/storage_sql.go
View File

@@ -177,6 +177,18 @@ func (self *SQLStorage) SetTPTiming(tpid string, tm *utils.TPTiming) error {
}
func (self *SQLStorage) RemTPData(table, tpid string, args ...string) error {
if len(table) == 0 { // Remove tpid out of all tables
tx := self.db.Begin()
for _, tblName := range []string{utils.TBL_TP_TIMINGS, utils.TBL_TP_DESTINATIONS, utils.TBL_TP_RATES, utils.TBL_TP_DESTINATION_RATES, utils.TBL_TP_RATING_PLANS, utils.TBL_TP_RATE_PROFILES,
utils.TBL_TP_SHARED_GROUPS, utils.TBL_TP_CDR_STATS, utils.TBL_TP_LCRS, utils.TBL_TP_ACTIONS, utils.TBL_TP_ACTION_PLANS, utils.TBL_TP_ACTION_TRIGGERS, utils.TBL_TP_ACCOUNT_ACTIONS, utils.TBL_TP_DERIVED_CHARGERS} {
if err := tx.Table(tblName).Where("tpid = ?", tpid).Delete(nil).Error; err != nil {
tx.Rollback()
return err
}
}
tx.Commit()
return nil
}
q := fmt.Sprintf("DELETE FROM %s WHERE tpid='%s' AND id='%s'", table, tpid, args[0])
switch table {
case utils.TBL_TP_RATE_PROFILES:

47
engine/storage_sql_local_test.go
View File

@@ -133,6 +133,53 @@ func TestRemoveData(t *testing.T) {
} else if err.Error() != "Record Not Found" {
t.Error(err.Error())
}
// Create again so we can test complete TP removal
if err := mysqlDb.SetTPTiming(TEST_SQL, tm); err != nil {
t.Error(err.Error())
}
if tmgs, err := mysqlDb.GetTpTimings(TEST_SQL, tm.Id); err != nil {
t.Error(err.Error())
} else if len(tmgs) == 0 {
t.Error("Could not store TPTiming")
}
// Create RatingProfile
if err := mysqlDb.SetTPRatingProfiles(TEST_SQL, map[string]*utils.TPRatingProfile{rp.KeyId(): rp}); err != nil {
t.Error(err.Error())
}
if rps, err := mysqlDb.GetTpRatingProfiles(rp); err != nil {
t.Error(err.Error())
} else if len(rps) == 0 {
t.Error("Could not store TPRatingProfile")
}
// Create AccountActions
if err := mysqlDb.SetTPAccountActions(aa.TPid, map[string]*utils.TPAccountActions{aa.KeyId(): aa}); err != nil {
t.Error(err.Error())
}
if aas, err := mysqlDb.GetTpAccountActions(aa); err != nil {
t.Error(err.Error())
} else if len(aas) == 0 {
t.Error("Could not create TPAccountActions")
}
// Remove TariffPlan completely
if err := mysqlDb.RemTPData("", TEST_SQL); err != nil {
t.Error(err.Error())
}
// Make sure we have removed it
if _, err := mysqlDb.GetTpTimings(TEST_SQL, tm.Id); err == nil {
t.Error("Should report error on querying here")
} else if err.Error() != "Record Not Found" {
t.Error(err.Error())
}
if _, err := mysqlDb.GetTpRatingProfiles(rp); err == nil {
t.Error("Should return error in case of record not found from ORM")
} else if err.Error() != "Record Not Found" {
t.Error(err.Error())
}
if _, err := mysqlDb.GetTpAccountActions(aa); err == nil {
t.Error("Should receive error in case of not found")
} else if err.Error() != "Record Not Found" {
t.Error(err.Error())
}
}
func TestSetCdr(t *testing.T) {