mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
add tests for resourceS
This commit is contained in:
armirveliaj
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
1c2aac5547
commit
a106f7eb6f
@@ -3747,3 +3747,252 @@ func TestResourceMatchingResourcesForEventWeightFromDynamicsErr(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
func TestStoreMatchedResourcesStoreIntervalZero(t *testing.T) {
|
||||||
|
cfg := config.NewDefaultCGRConfig()
|
||||||
|
cfg.ResourceSCfg().StoreInterval = 0
|
||||||
|
|
||||||
|
data, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
||||||
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: data}, cfg.DbCfg())
|
||||||
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
||||||
|
|
||||||
|
rS := &ResourceS{
|
||||||
|
dm: dm,
|
||||||
|
cfg: cfg,
|
||||||
|
storedResources: utils.NewStringSet(nil),
|
||||||
|
}
|
||||||
|
|
||||||
|
dirty := true
|
||||||
|
resources := Resources{
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{},
|
||||||
|
},
|
||||||
|
dirty: &dirty,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rS.storeMatchedResources(context.Background(), resources)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected nil error, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !*resources[0].dirty {
|
||||||
|
t.Error("Expected dirty flag to remain true when StoreInterval is 0")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreMatchedResourcesStoreIntervalPositive(t *testing.T) {
|
||||||
|
cfg := config.NewDefaultCGRConfig()
|
||||||
|
cfg.ResourceSCfg().StoreInterval = 10 * time.Second
|
||||||
|
|
||||||
|
data, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
||||||
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: data}, cfg.DbCfg())
|
||||||
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
||||||
|
|
||||||
|
rS := &ResourceS{
|
||||||
|
dm: dm,
|
||||||
|
cfg: cfg,
|
||||||
|
storedResources: utils.NewStringSet(nil),
|
||||||
|
}
|
||||||
|
|
||||||
|
dirty := true
|
||||||
|
resources := Resources{
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{},
|
||||||
|
},
|
||||||
|
dirty: &dirty,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES2",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{},
|
||||||
|
},
|
||||||
|
dirty: &dirty,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rS.storeMatchedResources(context.Background(), resources)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected nil error, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !*resources[0].dirty {
|
||||||
|
t.Error("Expected dirty flag to be true for RES1")
|
||||||
|
}
|
||||||
|
if !*resources[1].dirty {
|
||||||
|
t.Error("Expected dirty flag to be true for RES2")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !rS.storedResources.Has("cgrates.org:RES1") {
|
||||||
|
t.Error("Expected RES1 to be in storedResources set")
|
||||||
|
}
|
||||||
|
if !rS.storedResources.Has("cgrates.org:RES2") {
|
||||||
|
t.Error("Expected RES2 to be in storedResources set")
|
||||||
|
}
|
||||||
|
|
||||||
|
if rS.storedResources.Size() != 2 {
|
||||||
|
t.Errorf("Expected 2 resources in set, got: %d", rS.storedResources.Size())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreMatchedResourcesStoreIntervalNegative(t *testing.T) {
|
||||||
|
cfg := config.NewDefaultCGRConfig()
|
||||||
|
cfg.ResourceSCfg().StoreInterval = -1
|
||||||
|
|
||||||
|
data, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
||||||
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: data}, cfg.DbCfg())
|
||||||
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
||||||
|
|
||||||
|
rS := &ResourceS{
|
||||||
|
dm: dm,
|
||||||
|
cfg: cfg,
|
||||||
|
storedResources: utils.NewStringSet(nil),
|
||||||
|
}
|
||||||
|
|
||||||
|
dirty := true
|
||||||
|
resources := Resources{
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{},
|
||||||
|
},
|
||||||
|
dirty: &dirty,
|
||||||
|
rPrf: &resourceProfile{
|
||||||
|
ResourceProfile: &utils.ResourceProfile{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Limit: 10,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rS.storeMatchedResources(context.Background(), resources)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected nil error, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
savedRes, err := dm.GetResource(context.Background(), "cgrates.org", "RES1", false, false, "")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected resource to be stored in database, got error: %v", err)
|
||||||
|
}
|
||||||
|
if savedRes == nil {
|
||||||
|
t.Error("Expected resource to be stored in database")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreMatchedResourcesNoDirtyFlag(t *testing.T) {
|
||||||
|
cfg := config.NewDefaultCGRConfig()
|
||||||
|
cfg.ResourceSCfg().StoreInterval = -1
|
||||||
|
|
||||||
|
data, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
||||||
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: data}, cfg.DbCfg())
|
||||||
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
||||||
|
|
||||||
|
rS := &ResourceS{
|
||||||
|
dm: dm,
|
||||||
|
cfg: cfg,
|
||||||
|
storedResources: utils.NewStringSet(nil),
|
||||||
|
}
|
||||||
|
|
||||||
|
resources := Resources{
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{},
|
||||||
|
},
|
||||||
|
dirty: nil,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rS.storeMatchedResources(context.Background(), resources)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected nil error, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = dm.GetResource(context.Background(), "cgrates.org", "RES1", false, false, "")
|
||||||
|
if err == nil {
|
||||||
|
t.Error("Expected resource with nil dirty flag not to be stored")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStoreMatchedResourcesWithUsages(t *testing.T) {
|
||||||
|
cfg := config.NewDefaultCGRConfig()
|
||||||
|
cfg.ResourceSCfg().StoreInterval = -1
|
||||||
|
|
||||||
|
data, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
||||||
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: data}, cfg.DbCfg())
|
||||||
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
||||||
|
|
||||||
|
rS := &ResourceS{
|
||||||
|
dm: dm,
|
||||||
|
cfg: cfg,
|
||||||
|
storedResources: utils.NewStringSet(nil),
|
||||||
|
}
|
||||||
|
|
||||||
|
dirty := true
|
||||||
|
resources := Resources{
|
||||||
|
{
|
||||||
|
Resource: &utils.Resource{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Usages: map[string]*utils.ResourceUsage{
|
||||||
|
"USAGE1": {
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "USAGE1",
|
||||||
|
ExpiryTime: time.Now().Add(1 * time.Hour),
|
||||||
|
Units: 5,
|
||||||
|
},
|
||||||
|
"USAGE2": {
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "USAGE2",
|
||||||
|
ExpiryTime: time.Now().Add(2 * time.Hour),
|
||||||
|
Units: 3,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
TTLIdx: []string{"USAGE1", "USAGE2"},
|
||||||
|
},
|
||||||
|
dirty: &dirty,
|
||||||
|
rPrf: &resourceProfile{
|
||||||
|
ResourceProfile: &utils.ResourceProfile{
|
||||||
|
Tenant: "cgrates.org",
|
||||||
|
ID: "RES1",
|
||||||
|
Limit: 10,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := rS.storeMatchedResources(context.Background(), resources)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Expected nil error, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
savedRes, err := dm.GetResource(context.Background(), "cgrates.org", "RES1", false, false, "")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Expected resource to be stored, got error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(savedRes.Usages) != 2 {
|
||||||
|
t.Errorf("Expected 2 usages, got: %d", len(savedRes.Usages))
|
||||||
|
}
|
||||||
|
if _, exists := savedRes.Usages["USAGE1"]; !exists {
|
||||||
|
t.Error("Expected USAGE1 to be stored")
|
||||||
|
}
|
||||||
|
if _, exists := savedRes.Usages["USAGE2"]; !exists {
|
||||||
|
t.Error("Expected USAGE2 to be stored")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(savedRes.TTLIdx) != 2 {
|
||||||
|
t.Errorf("Expected 2 TTL indexes, got: %d", len(savedRes.TTLIdx))
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user