Updated monetary balance test

This commit is contained in:
Trial97
2021-09-28 08:30:43 +03:00
committed by Dan Christian Bogos
parent 358d0bbabf
commit 0abb2128fd

View File

@@ -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) {