From 8ea4ba4df93e7dabea94a585ce99de396cb3d464 Mon Sep 17 00:00:00 2001 From: ionutboangiu Date: Fri, 7 May 2021 17:16:07 +0300 Subject: [PATCH] Add missing test after resolving the conflict --- engine/z_resources_test.go | 53 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) diff --git a/engine/z_resources_test.go b/engine/z_resources_test.go index 1ef946243..6ccd2b053 100644 --- a/engine/z_resources_test.go +++ b/engine/z_resources_test.go @@ -3224,3 +3224,56 @@ func TestResourcesAllocateResourceDryRun(t *testing.T) { t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, rcv) } } + +func TestResourceAllocateResourceOtherDB(t *testing.T) { + rProf := &ResourceProfile{ + Tenant: "cgrates.org", + ID: "RL_DB", + FilterIDs: []string{"*string:~*opts.Resource:RL_DB"}, + Weight: 100, + Limit: 2, + ThresholdIDs: []string{utils.MetaNone}, + UsageTTL: -time.Nanosecond, + } + + Cache.Clear(nil) + cfg := config.NewDefaultCGRConfig() + dm := NewDataManager(NewInternalDB(nil, nil, true), cfg.CacheCfg(), nil) + fltS := NewFilterS(cfg, nil, dm) + rs := NewResourceService(dm, cfg, fltS, nil) + if err := dm.SetResourceProfile(rProf, true); err != nil { + t.Fatal(err) + } + if err := dm.SetResource(&Resource{ + Tenant: "cgrates.org", + ID: "RL_DB", + Usages: map[string]*ResourceUsage{ + "RU1": { // the resource in DB is expired (should be cleaned when the next allocate is called) + Tenant: "cgrates.org", + ID: "RU1", + ExpiryTime: time.Date(2014, 7, 3, 13, 43, 0, 1, time.UTC), + Units: 1, + }, + }, + TTLIdx: []string{"RU1"}, + }, nil, 2, true); err != nil { // simulate how the resource is stored in redis or mongo(non-exported fields are not populated) + t.Fatal(err) + } + var reply string + exp := rProf.ID + if err := rs.V1AllocateResources(utils.ArgRSv1ResourceUsage{ + CGREvent: &utils.CGREvent{ + Tenant: "cgrates.org", + ID: "ef0f554", + Event: map[string]interface{}{"": ""}, + APIOpts: map[string]interface{}{"Resource": "RL_DB"}, + }, + UsageID: "56156434-2e44-4f16-a766-086f10b413cd", + Units: 1, + }, &reply); err != nil { + t.Fatal(err) + } else if reply != exp { + t.Errorf("Expected: %q, received: %q", exp, reply) + } + +}