diff --git a/general_tests/session_money_rating_subject_it_test.go b/general_tests/session_money_rating_subject_it_test.go index 02eca02aa..21715ad3e 100644 --- a/general_tests/session_money_rating_subject_it_test.go +++ b/general_tests/session_money_rating_subject_it_test.go @@ -146,7 +146,7 @@ func testSesMRSItAddVoiceBalance(t *testing.T) { Tenant: sesMRSTenant, Account: sesMRSAccount, BalanceType: utils.MetaMonetary, - Value: float64(time.Hour), + Value: float64(time.Hour - 5*time.Second), Balance: map[string]interface{}{ utils.ID: "TestSesBal1", utils.RatingSubject: "*zero1s", @@ -159,6 +159,23 @@ func testSesMRSItAddVoiceBalance(t *testing.T) { t.Errorf("Received: %s", reply) } + attrSetBalance = utils.AttrSetBalance{ + Tenant: sesMRSTenant, + Account: sesMRSAccount, + BalanceType: utils.MetaVoice, + Value: float64(5 * time.Second), + Balance: map[string]interface{}{ + utils.ID: "TestSesBal2", + utils.RatingSubject: "*zero1s", + utils.Weight: 10, + }, + } + if err := sesMRSRPC.Call(utils.APIerSv2SetBalance, attrSetBalance, &reply); err != nil { + t.Error(err) + } else if reply != utils.OK { + t.Errorf("Received: %s", reply) + } + var acnt engine.Account if err := sesMRSRPC.Call(utils.APIerSv2GetAccount, &utils.AttrGetAccount{ @@ -167,10 +184,14 @@ func testSesMRSItAddVoiceBalance(t *testing.T) { }, &acnt); err != nil { t.Fatal(err) } - expected := float64(time.Hour) + expected := float64(time.Hour - 5*time.Second) if rply := acnt.BalanceMap[utils.MetaMonetary].GetTotalValue(); rply != expected { t.Errorf("Expected: %v, received: %v", expected, rply) } + expected = float64(5 * time.Second) + if rply := acnt.BalanceMap[utils.MetaVoice].GetTotalValue(); rply != expected { + t.Errorf("Expected: %v, received: %v", expected, rply) + } } func testSesMRSItInitSession(t *testing.T) { @@ -220,6 +241,10 @@ func testSesMRSItTerminateSession(t *testing.T) { if rply := acnt.BalanceMap[utils.MetaMonetary].GetTotalValue(); rply != expected { t.Errorf("Expected: %v, received: %v", utils.ToJSON(expected), utils.ToJSON(rply)) } + expected = 0 + if rply := acnt.BalanceMap[utils.MetaVoice].GetTotalValue(); rply != expected { + t.Errorf("Expected: %v, received: %v", expected, rply) + } } func testSesMRSItStopCgrEngine(t *testing.T) {