Add coverage tests for engine/chargers.go

This commit is contained in:
ionutboangiu
2021-06-02 13:45:59 +03:00
committed by Dan Christian Bogos
parent 33cab3f80a
commit 04390e6fb3
3 changed files with 812 additions and 2 deletions

View File

@@ -18,7 +18,11 @@ along with this program. If not, see <http://www.gnu.org/licenses/>
package engine
import (
"bytes"
"log"
"os"
"reflect"
"strings"
"testing"
"time"
@@ -462,3 +466,212 @@ func TestChargerProcessEvent(t *testing.T) {
t.Errorf("Expecting: %+v, received: %+v ", utils.ToJSON(rpl[0]), utils.ToJSON(rcv[0]))
}
}
func TestChargersmatchingChargerProfilesForEventChargerProfileNotFound(t *testing.T) {
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().StringIndexedFields = &[]string{
"string",
}
defaultCfg.ChargerSCfg().PrefixIndexedFields = &[]string{"prefix"}
defaultCfg.ChargerSCfg().SuffixIndexedFields = &[]string{"suffix"}
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().NestedFields = false
dataDB := NewInternalDB(nil, nil, true)
dmCharger := NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
cS := &ChargerService{
dm: dmCharger,
filterS: &FilterS{
dm: dmCharger,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
cgrEv := &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "cgrEvID",
Event: map[string]interface{}{
"Charger": "ChargerProfile1",
utils.AnswerTime: time.Date(2021, 4, 1, 10, 0, 0, 0, time.UTC),
"UsageInterval": "1s",
utils.Weight: "10.0",
},
APIOpts: map[string]interface{}{
utils.Subsys: utils.MetaChargers,
},
}
experr := utils.ErrNotFound
rcv, err := cS.matchingChargerProfilesForEvent("tnt", cgrEv)
if err == nil || err != experr {
t.Fatalf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestChargersmatchingChargerProfilesForEventDoesNotPass(t *testing.T) {
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().StringIndexedFields = &[]string{
"string",
}
defaultCfg.ChargerSCfg().PrefixIndexedFields = &[]string{"prefix"}
defaultCfg.ChargerSCfg().SuffixIndexedFields = &[]string{"suffix"}
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().NestedFields = false
dataDB := NewInternalDB(nil, nil, true)
dmCharger := NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
cS := &ChargerService{
dm: dmCharger,
filterS: &FilterS{
dm: dmCharger,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
cgrEv := &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "cgrEvID",
Event: map[string]interface{}{
"Charger": "ChargerProfile1",
utils.AnswerTime: time.Date(2021, 4, 1, 10, 0, 0, 0, time.UTC),
"UsageInterval": "1s",
utils.Weight: "10.0",
},
APIOpts: map[string]interface{}{
utils.Subsys: utils.MetaChargers,
},
}
experr := utils.ErrNotFound
rcv, err := cS.matchingChargerProfilesForEvent(cgrEv.Tenant, cgrEv)
if err == nil || err != experr {
t.Fatalf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestChargersmatchingChargerProfilesForEventErrGetChPrf(t *testing.T) {
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().StringIndexedFields = &[]string{
"string",
}
defaultCfg.ChargerSCfg().PrefixIndexedFields = &[]string{"prefix"}
defaultCfg.ChargerSCfg().SuffixIndexedFields = &[]string{"suffix"}
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().NestedFields = false
dbm := &DataDBMock{
GetKeysForPrefixF: func(ctx *context.Context, s string) ([]string, error) {
return []string{":"}, nil
},
}
dmCharger := NewDataManager(dbm, defaultCfg.CacheCfg(), nil)
cS := &ChargerService{
dm: dmCharger,
filterS: &FilterS{
dm: dmCharger,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
cgrEv := &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "cgrEvID",
Event: map[string]interface{}{
"Charger": "ChargerProfile1",
utils.AnswerTime: time.Date(2021, 4, 1, 10, 0, 0, 0, time.UTC),
"UsageInterval": "1s",
utils.Weight: "10.0",
},
APIOpts: map[string]interface{}{
utils.Subsys: utils.MetaChargers,
},
}
experr := utils.ErrNotImplemented
rcv, err := cS.matchingChargerProfilesForEvent(cgrEv.Tenant, cgrEv)
if err == nil || err != experr {
t.Fatalf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestChargersprocessEvent(t *testing.T) {
defaultCfg := config.NewDefaultCGRConfig()
cS := &ChargerService{
cfg: defaultCfg,
}
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
APIOpts: map[string]interface{}{
utils.OptsAttributesProcessRuns: 2,
},
}
experr := "NO_DATABASE_CONNECTION"
rcv, err := cS.processEvent(cgrEv.Tenant, cgrEv)
if err == nil || err.Error() != experr {
t.Fatalf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestChargersV1ProcessEventMissingArgs(t *testing.T) {
cS := &ChargerService{}
args := &utils.CGREvent{}
var reply *[]*ChrgSProcessEventReply
experr := "MANDATORY_IE_MISSING: [Event]"
err := cS.V1ProcessEvent(args, reply)
if err == nil || err.Error() != experr {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
}
func TestChargersShutdown(t *testing.T) {
cS := &ChargerService{}
utils.Logger.SetLogLevel(6)
utils.Logger.SetSyslog(nil)
var buf bytes.Buffer
log.SetOutput(&buf)
defer func() {
log.SetOutput(os.Stderr)
}()
exp := []string{
"CGRateS <> [INFO] <ChargerS> shutdown initialized",
"CGRateS <> [INFO] <ChargerS> shutdown complete",
}
cS.Shutdown()
rcv := strings.Split(buf.String(), "\n")
for i := 0; i < 2; i++ {
rcv[i] = rcv[i][20:]
if rcv[i] != exp[i] {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp[i], rcv[i])
}
}
utils.Logger.SetLogLevel(0)
}

View File

@@ -35,7 +35,7 @@ type DataDBMock struct {
RemoveAttributeProfileDrvF func(ctx *context.Context, str1 string, str2 string) error
SetLoadIDsDrvF func(ctx *context.Context, loadIDs map[string]int64) error
GetFilterDrvF func(ctx *context.Context, str1 string, str2 string) (*Filter, error)
GetChargerProfileDrvF func(*context.Context, string, string) (*ChargerProfile, error)
GetChargerProfileDrvF func(tnt, id string) (*ChargerProfile, error)
GetThresholdProfileDrvF func(ctx *context.Context, tenant, id string) (tp *ThresholdProfile, err error)
SetThresholdProfileDrvF func(ctx *context.Context, tp *ThresholdProfile) (err error)
RemThresholdProfileDrvF func(ctx *context.Context, tenant, id string) (err error)
@@ -294,7 +294,10 @@ func (dbM *DataDBMock) RemoveAttributeProfileDrv(ctx *context.Context, str1 stri
return utils.ErrNotImplemented
}
func (dbM *DataDBMock) GetChargerProfileDrv(string, string) (*ChargerProfile, error) {
func (dbM *DataDBMock) GetChargerProfileDrv(tnt, id string) (*ChargerProfile, error) {
if dbM.GetChargerProfileDrvF != nil {
return dbM.GetChargerProfileDrvF(tnt, id)
}
return nil, utils.ErrNotImplemented
}

594
engine/z_chargers_test.go Normal file
View File

@@ -0,0 +1,594 @@
/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
package engine
import (
"fmt"
"reflect"
"testing"
"time"
"github.com/cgrates/birpc"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
"github.com/cgrates/rpcclient"
)
func TestChargersmatchingChargerProfilesForEventErrPass(t *testing.T) {
Cache.Clear(nil)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
dbm := &DataDBMock{
GetChargerProfileDrvF: func(s1, s2 string) (*ChargerProfile, error) {
return &ChargerProfile{
Tenant: s1,
ID: s2,
RunID: utils.MetaDefault,
FilterIDs: []string{"fltr1"},
}, nil
},
GetKeysForPrefixF: func(ctx *context.Context, s string) ([]string, error) {
return []string{s + "cgrates.org:chr1"}, nil
},
GetFilterDrvF: func(ctx *context.Context, s1, s2 string) (*Filter, error) {
return nil, utils.ErrNotImplemented
},
}
dmFilter := NewDataManager(dbm, defaultCfg.CacheCfg(), nil)
cS := &ChargerService{
dm: dmFilter,
filterS: &FilterS{
dm: dmFilter,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
cgrEv := &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "cgrEvID",
Event: map[string]interface{}{
"Charger": "ChargerProfile1",
utils.AnswerTime: time.Date(2021, 4, 19, 12, 0, 0, 0, time.UTC),
"UsageInterval": "10s",
utils.Weight: "10.0",
},
APIOpts: map[string]interface{}{
utils.Subsys: utils.MetaChargers,
},
}
experr := utils.ErrNotImplemented
rcv, err := cS.matchingChargerProfilesForEvent(cgrEv.Tenant, cgrEv)
if err == nil || err != experr {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestChargersprocessEventCallNilErr(t *testing.T) {
Cache.Clear(nil)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
data := NewInternalDB(nil, nil, true)
dm := NewDataManager(data, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.AttributeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
rply := AttrSProcessEventReply{
AlteredFields: []string{utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
}
*reply.(*AttrSProcessEventReply) = rply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
connMgr: NewConnManager(defaultCfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): rpcInternal,
}),
}
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
exp := []*ChrgSProcessEventReply{
{
ChargerSProfile: "1001",
AlteredFields: []string{utils.MetaReqRunID, utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
},
}
rcv, err := cS.processEvent(cgrEv.Tenant, cgrEv)
if err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
}
if !reflect.DeepEqual(exp, rcv) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(rcv))
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersprocessEventCallErr(t *testing.T) {
Cache.Clear(nil)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
data := NewInternalDB(nil, nil, true)
dm := NewDataManager(data, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.AttributeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
return utils.ErrNotFound
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
connMgr: NewConnManager(defaultCfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): rpcInternal,
}),
}
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
exp := []*ChrgSProcessEventReply{
{
ChargerSProfile: "1001",
AlteredFields: []string{utils.MetaReqRunID},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]interface{}{
utils.Subsys: utils.MetaChargers,
utils.OptsContext: utils.MetaChargers,
},
},
},
}
rcv, err := cS.processEvent(cgrEv.Tenant, cgrEv)
if err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
}
if !reflect.DeepEqual(exp, rcv) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(rcv))
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersV1ProcessEventErrNotFound(t *testing.T) {
Cache.Clear(nil)
dataDB := NewInternalDB(nil, nil, true)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := NewDataManager(dataDB, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.AttributeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
rply := AttrSProcessEventReply{
AlteredFields: []string{utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
}
*reply.(*AttrSProcessEventReply) = rply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
connMgr: NewConnManager(defaultCfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): rpcInternal,
}),
}
args := &utils.CGREvent{
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1002",
},
}
reply := &[]*ChrgSProcessEventReply{}
experr := utils.ErrNotFound
err := cS.V1ProcessEvent(args, reply)
if err == nil || err != experr {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersV1ProcessEventErrOther(t *testing.T) {
Cache.Clear(nil)
dataDB := NewInternalDB(nil, nil, true)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := NewDataManager(dataDB, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
"invalidMethod": func(ctx *context.Context, args, reply interface{}) error {
rply := AttrSProcessEventReply{
AlteredFields: []string{utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
}
*reply.(*AttrSProcessEventReply) = rply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
connMgr: NewConnManager(defaultCfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): rpcInternal,
}),
}
args := &utils.CGREvent{
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
reply := &[]*ChrgSProcessEventReply{}
exp := &[]*ChrgSProcessEventReply{}
experr := fmt.Sprintf("SERVER_ERROR: %s", rpcclient.ErrUnsupporteServiceMethod)
err := cS.V1ProcessEvent(args, reply)
if err == nil || err.Error() != experr {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if !reflect.DeepEqual(reply, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(reply))
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersV1ProcessEvent(t *testing.T) {
Cache.Clear(nil)
dataDB := NewInternalDB(nil, nil, true)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := NewDataManager(dataDB, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.AttributeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
rply := AttrSProcessEventReply{
AlteredFields: []string{utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
}
*reply.(*AttrSProcessEventReply) = rply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
connMgr: NewConnManager(defaultCfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): rpcInternal,
}),
}
args := &utils.CGREvent{
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
reply := &[]*ChrgSProcessEventReply{}
exp := &[]*ChrgSProcessEventReply{
{
ChargerSProfile: "1001",
AlteredFields: []string{utils.MetaReqRunID, utils.AccountField},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
},
}
err := cS.V1ProcessEvent(args, reply)
if err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
}
if !reflect.DeepEqual(reply, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(reply))
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersV1GetChargersForEventNilErr(t *testing.T) {
Cache.Clear(nil)
dataDB := NewInternalDB(nil, nil, true)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
defaultCfg.ChargerSCfg().AttributeSConns = []string{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := NewDataManager(dataDB, defaultCfg.CacheCfg(), nil)
cP := &ChargerProfile{
Tenant: "cgrates.org",
ID: "1001",
RunID: utils.MetaDefault,
FilterIDs: []string{"*string:~*req.Account:1001"},
}
if err := dm.SetChargerProfile(cP, true); err != nil {
t.Fatal(err)
}
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
args := &utils.CGREvent{
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
reply := &ChargerProfiles{}
exp := &ChargerProfiles{
{
Tenant: "cgrates.org",
ID: "1001",
FilterIDs: []string{"*string:~*req.Account:1001"},
RunID: "*default",
},
}
err := cS.V1GetChargersForEvent(args, reply)
if err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
}
if !reflect.DeepEqual(reply, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(reply))
}
if err := dm.DataDB().Flush(""); err != nil {
t.Error(err)
}
}
func TestChargersV1GetChargersForEventErr(t *testing.T) {
Cache.Clear(nil)
defaultCfg := config.NewDefaultCGRConfig()
defaultCfg.ChargerSCfg().IndexedSelects = false
dbm := &DataDBMock{
GetKeysForPrefixF: func(ctx *context.Context, s string) ([]string, error) {
return []string{":"}, nil
},
}
dm := NewDataManager(dbm, defaultCfg.CacheCfg(), nil)
cS := &ChargerService{
dm: dm,
filterS: &FilterS{
dm: dm,
cfg: defaultCfg,
},
cfg: defaultCfg,
}
args := &utils.CGREvent{
ID: "cgrEvID",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
}
reply := &ChargerProfiles{}
exp := &ChargerProfiles{}
experr := fmt.Sprintf("SERVER_ERROR: %s", utils.ErrNotImplemented)
err := cS.V1GetChargersForEvent(args, reply)
if err == nil || err.Error() != experr {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
}
if !reflect.DeepEqual(reply, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(reply))
}
}