mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
improving tests at engine
This commit is contained in:
gezimbll
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
6e024ae79a
commit
e708cbc3b7
@@ -1162,18 +1162,70 @@ func TestV2StoreSessionCostSet(t *testing.T) {
|
||||
OriginID: "originid",
|
||||
CostSource: "cgrates",
|
||||
CostDetails: &EventCost{
|
||||
CGRID: "evcgrid",
|
||||
RunID: "evrunid",
|
||||
StartTime: time.Date(2021, 11, 1, 2, 0, 0, 0, time.UTC),
|
||||
Usage: utils.DurationPointer(122),
|
||||
Cost: utils.Float64Pointer(134),
|
||||
Charges: []*ChargingInterval{},
|
||||
AccountSummary: &AccountSummary{},
|
||||
Rating: Rating{},
|
||||
Accounting: Accounting{},
|
||||
RatingFilters: RatingFilters{},
|
||||
Rates: ChargedRates{},
|
||||
Timings: ChargedTimings{},
|
||||
CGRID: "evcgrid",
|
||||
RunID: "evrunid",
|
||||
StartTime: time.Date(2021, 11, 1, 2, 0, 0, 0, time.UTC),
|
||||
Usage: utils.DurationPointer(122),
|
||||
Cost: utils.Float64Pointer(134),
|
||||
Charges: []*ChargingInterval{
|
||||
{
|
||||
RatingID: "rating1",
|
||||
Increments: []*ChargingIncrement{
|
||||
{
|
||||
Usage: 5 * time.Minute,
|
||||
Cost: 23,
|
||||
AccountingID: "acc_id",
|
||||
CompressFactor: 5,
|
||||
},
|
||||
{
|
||||
Usage: 5 * time.Minute,
|
||||
Cost: 23,
|
||||
AccountingID: "acc_id",
|
||||
CompressFactor: 5,
|
||||
},
|
||||
},
|
||||
CompressFactor: 3,
|
||||
usage: utils.DurationPointer(10 * time.Minute),
|
||||
ecUsageIdx: utils.DurationPointer(4 * time.Minute),
|
||||
cost: utils.Float64Pointer(38),
|
||||
},
|
||||
{
|
||||
RatingID: "rating2",
|
||||
Increments: []*ChargingIncrement{
|
||||
{
|
||||
Usage: 5 * time.Minute,
|
||||
Cost: 23,
|
||||
AccountingID: "acc_id",
|
||||
CompressFactor: 5,
|
||||
},
|
||||
{
|
||||
Usage: 5 * time.Minute,
|
||||
Cost: 23,
|
||||
AccountingID: "acc_id",
|
||||
CompressFactor: 5,
|
||||
},
|
||||
},
|
||||
CompressFactor: 3,
|
||||
usage: utils.DurationPointer(10 * time.Minute),
|
||||
ecUsageIdx: utils.DurationPointer(4 * time.Minute),
|
||||
cost: utils.Float64Pointer(38),
|
||||
},
|
||||
},
|
||||
AccountSummary: &AccountSummary{
|
||||
Tenant: "Tenant",
|
||||
ID: "acc_id",
|
||||
BalanceSummaries: BalanceSummaries{},
|
||||
AllowNegative: false,
|
||||
Disabled: true,
|
||||
},
|
||||
Rating: Rating{
|
||||
"rating1": &RatingUnit{},
|
||||
"rating2": &RatingUnit{},
|
||||
},
|
||||
Accounting: Accounting{},
|
||||
RatingFilters: RatingFilters{},
|
||||
Rates: ChargedRates{},
|
||||
Timings: ChargedTimings{},
|
||||
},
|
||||
},
|
||||
}
|
||||
@@ -1217,14 +1269,7 @@ func TestV1RateCDRS(t *testing.T) {
|
||||
}
|
||||
arg := &ArgRateCDRs{
|
||||
Flags: []string{utils.MetaStore, utils.MetaExport, utils.MetaThresholds, utils.MetaStats, utils.MetaChargers, utils.MetaAttributes},
|
||||
RPCCDRsFilter: utils.RPCCDRsFilter{
|
||||
CGRIDs: []string{"id", "cgr"},
|
||||
NotRequestTypes: []string{"noreq"},
|
||||
NotCGRIDs: []string{"cgrid"},
|
||||
RunIDs: []string{"runid"},
|
||||
OriginIDs: []string{"o_id"},
|
||||
NotOriginIDs: []string{"noid"},
|
||||
},
|
||||
|
||||
Tenant: "cgrates.org",
|
||||
APIOpts: map[string]interface{}{},
|
||||
}
|
||||
@@ -1235,3 +1280,97 @@ func TestV1RateCDRS(t *testing.T) {
|
||||
}
|
||||
|
||||
}
|
||||
func TestGetCostFromRater2(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CdrsCfg().RaterConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RateSConnsCfg)}
|
||||
cfg.CdrsCfg().SchedulerConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.SchedulerConnsCfg)}
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
ccMock := &ccMock{
|
||||
calls: map[string]func(args interface{}, reply interface{}) error{
|
||||
utils.ResponderDebit: func(args, reply interface{}) error {
|
||||
|
||||
return utils.ErrAccountNotFound
|
||||
},
|
||||
utils.SchedulerSv1ExecuteActionPlans: func(args, reply interface{}) error {
|
||||
rpl := "reply"
|
||||
*reply.(*string) = rpl
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
clientconn := make(chan rpcclient.ClientConnector, 1)
|
||||
clientconn <- ccMock
|
||||
connMgr := NewConnManager(cfg, map[string]chan rpcclient.ClientConnector{
|
||||
utils.ConcatenatedKey(utils.MetaInternal, utils.RateSConnsCfg): clientconn,
|
||||
utils.ConcatenatedKey(utils.MetaInternal, utils.SchedulerConnsCfg): clientconn,
|
||||
})
|
||||
cdrS := &CDRServer{
|
||||
cgrCfg: cfg,
|
||||
cdrDb: db,
|
||||
dm: dm,
|
||||
connMgr: connMgr,
|
||||
}
|
||||
cdr := &CDRWithAPIOpts{
|
||||
|
||||
CDR: &CDR{
|
||||
ToR: "tor",
|
||||
Tenant: "tenant",
|
||||
Category: "cdr",
|
||||
Subject: "cdrsubj",
|
||||
Account: "acc_cdr",
|
||||
Destination: "acc_dest",
|
||||
RequestType: utils.MetaDynaprepaid,
|
||||
Usage: 1 * time.Minute,
|
||||
},
|
||||
APIOpts: map[string]interface{}{},
|
||||
}
|
||||
|
||||
if _, err := cdrS.getCostFromRater(cdr); err == nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetCostFromRater3(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CdrsCfg().RaterConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RateSConnsCfg)}
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
ccMock := &ccMock{
|
||||
calls: map[string]func(args interface{}, reply interface{}) error{
|
||||
utils.ResponderGetCost: func(args, reply interface{}) error {
|
||||
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
clientconn := make(chan rpcclient.ClientConnector, 1)
|
||||
clientconn <- ccMock
|
||||
connMgr := NewConnManager(cfg, map[string]chan rpcclient.ClientConnector{
|
||||
utils.ConcatenatedKey(utils.MetaInternal, utils.RateSConnsCfg): clientconn,
|
||||
})
|
||||
cdrS := &CDRServer{
|
||||
cgrCfg: cfg,
|
||||
cdrDb: db,
|
||||
dm: dm,
|
||||
connMgr: connMgr,
|
||||
}
|
||||
cdr := &CDRWithAPIOpts{
|
||||
|
||||
CDR: &CDR{
|
||||
ToR: "tor",
|
||||
Tenant: "tenant",
|
||||
Category: "cdr",
|
||||
Subject: "cdrsubj",
|
||||
Account: "acc_cdr",
|
||||
Destination: "acc_dest",
|
||||
RequestType: "default",
|
||||
Usage: 1 * time.Minute,
|
||||
},
|
||||
APIOpts: map[string]interface{}{},
|
||||
}
|
||||
|
||||
if _, err := cdrS.getCostFromRater(cdr); err == nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -630,3 +630,519 @@ func TestResponderDebit(t *testing.T) {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetCostOnRatingPlans(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
arg := &utils.GetCostOnRatingPlansArgs{
|
||||
Account: "account",
|
||||
Subject: "subj",
|
||||
Destination: "destination",
|
||||
Tenant: "cgrates.org",
|
||||
SetupTime: time.Date(2021, 12, 24, 8, 0, 0, 0, time.UTC),
|
||||
Usage: 10 * time.Minute,
|
||||
RatingPlanIDs: []string{"rplan1", "rplan2", "rplan3"},
|
||||
APIOpts: map[string]interface{}{
|
||||
"apiopts": "opt",
|
||||
},
|
||||
}
|
||||
reply := &map[string]interface{}{}
|
||||
rs := &Responder{
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
},
|
||||
}
|
||||
if err := rs.GetCostOnRatingPlans(arg, reply); err == nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetMaxComputedUsage(t *testing.T) {
|
||||
|
||||
rs := &Responder{
|
||||
Timeout: 10 * time.Minute,
|
||||
Timezone: "UTC",
|
||||
}
|
||||
|
||||
mx := map[string]time.Duration{
|
||||
"usage1": 2 * time.Minute,
|
||||
"usage2": 4 * time.Minute,
|
||||
}
|
||||
rs.SetMaxComputedUsage(mx)
|
||||
if !reflect.DeepEqual(rs.MaxComputedUsage, mx) {
|
||||
t.Errorf("expected %v,received %v", mx, rs.MaxComputedUsage)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResponderDebitSet(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &CallCost{
|
||||
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
}
|
||||
key := utils.ConcatenatedKey(utils.ResponderDebit, arg.CgrID)
|
||||
Cache.Set(utils.CacheRPCResponses, key,
|
||||
&utils.CachedRPCResponse{Result: reply, Error: nil},
|
||||
nil, true, utils.NonTransactional)
|
||||
|
||||
if err := rs.Debit(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{Result: reply, Error: nil}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, key)
|
||||
|
||||
if !has {
|
||||
t.Error("has no values")
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %+v,received %+v", exp, rcv)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResponderMaxDebit(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &CallCost{
|
||||
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
}
|
||||
if err := rs.MaxDebit(arg, reply); err == nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderMaxDebit, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
}
|
||||
|
||||
func TestResponderMaxDebitSet(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &CallCost{
|
||||
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
}
|
||||
Cache.Set(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderMaxDebit, arg.CgrID),
|
||||
&utils.CachedRPCResponse{Result: reply, Error: nil},
|
||||
nil, true, utils.NonTransactional)
|
||||
if err := rs.MaxDebit(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderMaxDebit, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
}
|
||||
|
||||
func TestResponderRefundIncrements(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &Account{
|
||||
ID: "acc_id",
|
||||
BalanceMap: map[string]Balances{},
|
||||
UnitCounters: UnitCounters{},
|
||||
ActionTriggers: ActionTriggers{},
|
||||
AllowNegative: false,
|
||||
Disabled: false,
|
||||
UpdateTime: time.Date(2021, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
executingTriggers: false,
|
||||
}
|
||||
if err := rs.RefundIncrements(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundIncrements, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
|
||||
}
|
||||
func TestResponderRefundIncrementsSet(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &Account{
|
||||
ID: "acc_id",
|
||||
BalanceMap: map[string]Balances{},
|
||||
UnitCounters: UnitCounters{},
|
||||
ActionTriggers: ActionTriggers{},
|
||||
AllowNegative: false,
|
||||
Disabled: false,
|
||||
UpdateTime: time.Date(2021, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
executingTriggers: false,
|
||||
}
|
||||
|
||||
Cache.Set(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundIncrements, arg.CgrID), &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}, nil, true, utils.NonTransactional)
|
||||
if err := rs.RefundIncrements(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundIncrements, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestResponderRefundRounding(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &Account{
|
||||
ID: "acc_id",
|
||||
BalanceMap: map[string]Balances{},
|
||||
UnitCounters: UnitCounters{},
|
||||
ActionTriggers: ActionTriggers{},
|
||||
AllowNegative: false,
|
||||
Disabled: false,
|
||||
UpdateTime: time.Date(2021, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
executingTriggers: false,
|
||||
}
|
||||
if err := rs.RefundRounding(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundRounding, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
|
||||
}
|
||||
func TestResponderRefundRoundingSet(t *testing.T) {
|
||||
Cache.Clear(nil)
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
Cache = NewCacheS(cfg, dm, nil)
|
||||
config.SetCgrConfig(cfg)
|
||||
rs := &Responder{
|
||||
Timezone: "UTC",
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
MaxComputedUsage: map[string]time.Duration{},
|
||||
}
|
||||
arg := &CallDescriptorWithAPIOpts{
|
||||
|
||||
CallDescriptor: &CallDescriptor{
|
||||
CgrID: "cgrid",
|
||||
Category: "category",
|
||||
Tenant: "tenant",
|
||||
Subject: "subject",
|
||||
Account: "acount",
|
||||
Destination: "uk",
|
||||
ToR: "tor",
|
||||
TimeStart: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
TimeEnd: time.Date(2022, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
},
|
||||
APIOpts: map[string]interface{}{
|
||||
"tor": 30 * time.Minute,
|
||||
},
|
||||
}
|
||||
reply := &Account{
|
||||
ID: "acc_id",
|
||||
BalanceMap: map[string]Balances{},
|
||||
UnitCounters: UnitCounters{},
|
||||
ActionTriggers: ActionTriggers{},
|
||||
AllowNegative: false,
|
||||
Disabled: false,
|
||||
UpdateTime: time.Date(2021, 12, 1, 12, 0, 0, 0, time.UTC),
|
||||
executingTriggers: false,
|
||||
}
|
||||
Cache.Set(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundRounding, arg.CgrID),
|
||||
&utils.CachedRPCResponse{Result: reply, Error: err},
|
||||
nil, true, utils.NonTransactional)
|
||||
|
||||
if err := rs.RefundRounding(arg, reply); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
exp := &utils.CachedRPCResponse{
|
||||
Result: reply,
|
||||
Error: nil,
|
||||
}
|
||||
rcv, has := Cache.Get(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderRefundRounding, arg.CgrID))
|
||||
|
||||
if !has {
|
||||
t.Error("has no value")
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(rcv, exp) {
|
||||
t.Errorf("expected %v,received %v", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestGetMaxSessionTimeOnAccounts(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
||||
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
||||
arg := &utils.GetMaxSessionTimeOnAccountsArgs{
|
||||
Subject: "subject",
|
||||
Tenant: "",
|
||||
Destination: "destination",
|
||||
AccountIDs: []string{"acc_id1", "acc_id2"},
|
||||
Usage: 10 * time.Minute,
|
||||
SetupTime: time.Date(2022, 12, 1, 1, 0, 0, 0, time.UTC),
|
||||
APIOpts: map[string]interface{}{},
|
||||
}
|
||||
|
||||
reply := &map[string]interface{}{}
|
||||
rs := &Responder{
|
||||
FilterS: &FilterS{
|
||||
cfg: cfg,
|
||||
dm: dm,
|
||||
connMgr: nil,
|
||||
},
|
||||
}
|
||||
if err := rs.GetMaxSessionTimeOnAccounts(arg, reply); err == nil || err != utils.ErrAccountNotFound {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user