From 307a1d4d54139d4eddb7c6d90fe3fd5ae05b16e1 Mon Sep 17 00:00:00 2001 From: Trial97 Date: Tue, 24 Mar 2020 17:29:25 +0200 Subject: [PATCH] Added MMS in the balances order for BalanceSummary --- engine/account.go | 2 +- engine/account_test.go | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/engine/account.go b/engine/account.go index 0c3d695e8..442984b55 100644 --- a/engine/account.go +++ b/engine/account.go @@ -1084,7 +1084,7 @@ func (acc *Account) AsAccountSummary() *AccountSummary { ad.ID = idSplt[1] } - for _, balanceType := range []string{utils.DATA, utils.SMS, utils.VOICE, utils.GENERIC, utils.MONETARY} { + for _, balanceType := range []string{utils.DATA, utils.SMS, utils.MMS, utils.VOICE, utils.GENERIC, utils.MONETARY} { balances, has := acc.BalanceMap[balanceType] if !has { continue diff --git a/engine/account_test.go b/engine/account_test.go index da4547d27..0d0567030 100644 --- a/engine/account_test.go +++ b/engine/account_test.go @@ -2157,6 +2157,7 @@ func TestAccountAsAccountDigest(t *testing.T) { AllowNegative: true, BalanceMap: map[string]Balances{ utils.SMS: Balances{&Balance{ID: "sms1", Value: 14}}, + utils.MMS: Balances{&Balance{ID: "mms1", Value: 140}}, utils.DATA: Balances{&Balance{ID: "data1", Value: 1204}}, utils.VOICE: Balances{ &Balance{ID: "voice1", Weight: 20, DestinationIDs: utils.StringMap{"NAT": true}, Value: 3600}, @@ -2169,6 +2170,7 @@ func TestAccountAsAccountDigest(t *testing.T) { ID: "account1", BalanceSummaries: []*BalanceSummary{ &BalanceSummary{ID: "sms1", Type: utils.SMS, Value: 14, Disabled: false}, + &BalanceSummary{ID: "mms1", Type: utils.SMS, Value: 140, Disabled: false}, &BalanceSummary{ID: "data1", Type: utils.DATA, Value: 1204, Disabled: false}, &BalanceSummary{ID: "voice1", Type: utils.VOICE, Value: 1204, Disabled: false}, &BalanceSummary{ID: "voice2", Type: utils.VOICE, Value: 1200, Disabled: false},