From 98473b930a4b952d75b8c71c55fc5e8974051601 Mon Sep 17 00:00:00 2001 From: DanB Date: Mon, 28 Dec 2015 10:31:16 +0100 Subject: [PATCH] TestSMCosts --- engine/storage_cdrs_it_test.go | 33 ++++++++++++++++++++++++++++++++- engine/storage_sql.go | 2 +- 2 files changed, 33 insertions(+), 2 deletions(-) diff --git a/engine/storage_cdrs_it_test.go b/engine/storage_cdrs_it_test.go index 466ea2812..8f295664c 100644 --- a/engine/storage_cdrs_it_test.go +++ b/engine/storage_cdrs_it_test.go @@ -22,7 +22,7 @@ import ( "flag" "fmt" "path" - //"reflect" + "reflect" "testing" "time" @@ -51,6 +51,9 @@ func TestITCDRsMySQL(t *testing.T) { if err := testSetCDR(mysqlDb); err != nil { t.Error(err) } + if err := testSMCosts(mysqlDb); err != nil { + t.Error(err) + } } func TestITCDRsPSQL(t *testing.T) { @@ -72,6 +75,9 @@ func TestITCDRsPSQL(t *testing.T) { if err := testSetCDR(psqlDb); err != nil { t.Error(err) } + if err := testSMCosts(psqlDb); err != nil { + t.Error(err) + } } // helper function to populate CDRs and check if they were stored in storDb @@ -175,3 +181,28 @@ func testSetCDR(cdrStorage CdrStorage) error { } return nil } + +func testSMCosts(cdrStorage CdrStorage) error { + cc := &CallCost{ + Direction: utils.OUT, + Destination: "+4986517174963", + Timespans: []*TimeSpan{ + &TimeSpan{ + TimeStart: time.Date(2015, 12, 28, 8, 53, 0, 0, time.UTC), + TimeEnd: time.Date(2015, 12, 28, 8, 54, 40, 0, time.UTC), + DurationIndex: 0, + RateInterval: &RateInterval{Rating: &RIRate{Rates: RateGroups{&Rate{GroupIntervalStart: 0, Value: 100, RateIncrement: 10 * time.Second, RateUnit: time.Second}}}}, + }, + }, + TOR: utils.VOICE, + } + if err := cdrStorage.LogCallCost("164b0422fdc6a5117031b427439482c6a4f90e41", utils.META_DEFAULT, utils.UNIT_TEST, cc); err != nil { + return err + } + if rcvCC, err := cdrStorage.GetCallCostLog("164b0422fdc6a5117031b427439482c6a4f90e41", utils.META_DEFAULT); err != nil { + return err + } else if !reflect.DeepEqual(cc, rcvCC) { + return fmt.Errorf("Expecting: %+v, received: %+v", cc, rcvCC) + } + return nil +} diff --git a/engine/storage_sql.go b/engine/storage_sql.go index 25129812a..b953d66e1 100644 --- a/engine/storage_sql.go +++ b/engine/storage_sql.go @@ -569,7 +569,7 @@ func (self *SQLStorage) SetTpAccountActions(aas []TpAccountAction) error { return nil } -func (self *SQLStorage) LogCallCost(cgrid, runid, source string, cc *CallCost) (err error) { // +func (self *SQLStorage) LogCallCost(cgrid, runid, source string, cc *CallCost) error { if cc == nil { return nil }