mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
1487 lines
40 KiB
Go
1487 lines
40 KiB
Go
/*
|
|
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 (
|
|
"bytes"
|
|
"errors"
|
|
"reflect"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/birpc"
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
"github.com/cgrates/ltcache"
|
|
)
|
|
|
|
func TestCacheSSetWithReplicateTrue(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgCacheReplicateSet{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemID",
|
|
Value: &utils.CachedRPCResponse{
|
|
Result: "reply",
|
|
Error: nil},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.CacheCfg().ReplicationConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Replicate: true,
|
|
},
|
|
}
|
|
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1ReplicateSet: func(_ *context.Context, args, reply any) error {
|
|
argCache, canCast := args.(*utils.ArgCacheReplicateSet)
|
|
if !canCast {
|
|
return errors.New("cannot cast")
|
|
}
|
|
Cache.Set(nil, argCache.CacheID, argCache.ItemID, argCache.Value, nil, true, utils.EmptyString)
|
|
*reply.(*string) = utils.OK
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg), utils.CacheSv1, clientconn)
|
|
|
|
stopchan := make(chan struct{}, 1)
|
|
close(stopchan)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.SetWithReplicate(context.Background(), args); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
expectedVal := &utils.CachedRPCResponse{
|
|
Result: "reply",
|
|
Error: nil,
|
|
}
|
|
if val, ok := Cache.Get(utils.CacheAccounts, "itemID"); !ok {
|
|
t.Errorf("Expected value")
|
|
} else {
|
|
valConverted, canCast := val.(*utils.CachedRPCResponse)
|
|
if !canCast {
|
|
t.Error("Should cast")
|
|
}
|
|
if valConverted.Error != nil {
|
|
t.Errorf("Expected error <%v>, Received error <%v>", expectedVal.Error, valConverted.Error)
|
|
}
|
|
if !reflect.DeepEqual(expectedVal.Result, valConverted.Result) {
|
|
t.Errorf("Expected %v, received %v", utils.ToJSON(expectedVal), utils.ToJSON(valConverted))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCacheSSetWithReplicateFalse(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgCacheReplicateSet{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemID",
|
|
Value: &utils.CachedRPCResponse{Result: "reply", Error: nil},
|
|
GroupIDs: []string{"groupId", "groupId"},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.CacheCfg().ReplicationConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Replicate: false,
|
|
},
|
|
}
|
|
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
connMgr := NewConnManager(cfg)
|
|
|
|
stopchan := make(chan struct{}, 1)
|
|
close(stopchan)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.SetWithReplicate(context.Background(), args); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestCacheSGetWithRemote(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Remote: true,
|
|
},
|
|
}
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1GetItem: func(_ *context.Context, args, reply any) error {
|
|
var valBack string = "test_value_was_set"
|
|
*reply.(*any) = valBack
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg), utils.CacheSv1GetItem, clientconn)
|
|
|
|
stopchan := make(chan struct{}, 1)
|
|
close(stopchan)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
// first we have to set the value in order to get it from our mock
|
|
cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "test_value_was_set", []string{}, true, utils.NonTransactional)
|
|
var reply any
|
|
expected := "test_value_was_set"
|
|
if err := cacheS.V1GetItemWithRemote(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else {
|
|
strVal, canCast := reply.(string)
|
|
if !canCast {
|
|
t.Error("must be a string")
|
|
}
|
|
if strVal != expected {
|
|
t.Errorf("Expected %v, received %v", expected, strVal)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCacheSGetWithRemoteFalse(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Remote: false,
|
|
},
|
|
}
|
|
|
|
connMgr := NewConnManager(cfg)
|
|
|
|
stopchan := make(chan struct{}, 1)
|
|
close(stopchan)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
var reply any = utils.OK
|
|
if err := cacheS.V1GetItemWithRemote(context.Background(), args, &reply); err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("Expected error <%v>, received error <%v>", utils.ErrNotFound, err)
|
|
}
|
|
}
|
|
func TestRemoveWithoutReplicate(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
connMgr := NewConnManager(cfg)
|
|
chS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
chS.tCache.Set(utils.CacheAccounts, "itemId", "value", nil, true, utils.NonTransactional)
|
|
|
|
chS.RemoveWithoutReplicate(utils.CacheAccounts, "itemId", true, utils.NonTransactional)
|
|
if _, has := chS.tCache.Get(utils.CacheAccounts, "itemId"); has {
|
|
t.Error("This itemId shouldn't exist")
|
|
}
|
|
|
|
}
|
|
|
|
func TestV1GetItemExpiryTimeFromCacheErr(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
var reply time.Time
|
|
if err := cacheS.V1GetItemExpiryTime(context.Background(), args, &reply); err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("Expected error <%v>, Received error <%v>", utils.ErrNotFound, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestV1GetItemErr(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
var reply any
|
|
if err := cacheS.V1GetItem(context.Background(), args, &reply); err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("Expected error <%v>, Received error <%v>", utils.ErrNotFound, err)
|
|
}
|
|
|
|
}
|
|
func TestV1GetItemIDsErr(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemIDsWithAPIOpts{
|
|
ArgsGetCacheItemIDs: utils.ArgsGetCacheItemIDs{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemIDPrefix: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
var reply []string
|
|
if err := cacheS.V1GetItemIDs(context.Background(), args, &reply); err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("Expected error <%v>, Received error <%v>", utils.ErrNotFound, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSGetWithRemoteQueryErr(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
|
|
Remote: true,
|
|
},
|
|
}
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1GetItem: func(_ *context.Context, args, reply any) error {
|
|
return utils.ErrNotFound
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg), utils.CacheSv1GetItem, clientconn)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expErr := utils.ErrNotFound
|
|
if _, err := cacheS.GetWithRemote(context.Background(), args); err == nil || err != expErr {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestCacheSGetWithRemoteTCacheGet(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.AccountsStr,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().RemoteConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg)}
|
|
|
|
var customRply any = utils.ArgsGetCacheItem{
|
|
CacheID: utils.AccountsStr,
|
|
ItemID: "itemId",
|
|
}
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1GetItem: func(_ *context.Context, args, reply any) error {
|
|
*reply.(*any) = customRply
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.RemoteConnsCfg), utils.CacheSv1GetItem, clientconn)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
exp := "expected value"
|
|
cacheS.tCache.Set(utils.AccountsStr, "itemId", exp, []string{}, true, utils.EmptyString)
|
|
|
|
if rcv, err := cacheS.GetWithRemote(context.Background(), args); err != nil {
|
|
t.Error(err)
|
|
} else if rcv != exp {
|
|
t.Errorf("Expected <%v>, received <%v>", exp, rcv)
|
|
}
|
|
}
|
|
|
|
func TestCacheSV1ReplicateRemove(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
args := &utils.ArgCacheReplicateRemove{
|
|
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
Tenant: utils.CGRateSorg,
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().ReplicationConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg)}
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1ReplicateRemove: func(_ *context.Context, args, reply any) error {
|
|
var valBack string = utils.OK
|
|
*reply.(*any) = valBack
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg), utils.CacheSv1, clientconn)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
var reply string
|
|
if err := cacheS.V1ReplicateRemove(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Expected reply <%v>, Received <%v>", utils.OK, reply)
|
|
}
|
|
}
|
|
|
|
func TestCacheSReplicateRemove(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cfg.CacheCfg().ReplicationConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg)}
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
utils.CacheAccounts: {
|
|
Replicate: true,
|
|
},
|
|
}
|
|
clientconn := make(chan birpc.ClientConnector, 1)
|
|
clientconn <- &ccMock{
|
|
calls: map[string]func(_ *context.Context, args any, reply any) error{
|
|
utils.CacheSv1ReplicateRemove: func(_ *context.Context, args, reply any) error {
|
|
if err := Cache.Remove(context.Background(), utils.CacheAccounts, "itemId", true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
connMgr := NewConnManager(cfg)
|
|
connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicationConnsCfg), utils.CacheSv1, clientconn)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := Cache.Set(context.Background(), utils.CacheAccounts, "itemId", "val", nil, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := cacheS.ReplicateRemove(context.Background(), utils.CacheAccounts, "itemId"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAccounts, "itemId"); ok != false || rcv != nil {
|
|
t.Errorf("Expected rcv <nil>, Received <%v>, OK <%v>", rcv, ok)
|
|
}
|
|
}
|
|
|
|
func TestCacheSV1ReplicateSet(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
args := &utils.ArgCacheReplicateSet{
|
|
Tenant: utils.CGRateSorg,
|
|
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
Value: "valinterface",
|
|
GroupIDs: []string{},
|
|
}
|
|
|
|
if err := Cache.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := utils.OK
|
|
var reply string
|
|
if err := cacheS.V1ReplicateSet(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if exp != reply {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
getExp := "valinterface"
|
|
if rcv, ok := Cache.Get(utils.CacheAccounts, "itemId"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if rcv != getExp {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", getExp, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1ReplicateSetErr(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
fltr := &Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "fltr1",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaRegex,
|
|
Element: "~*req.Account",
|
|
Values: []string{"^(?!On.*On\\s.+?wrote:)(On\\s(.+?)wrote:)$"},
|
|
},
|
|
},
|
|
}
|
|
args := &utils.ArgCacheReplicateSet{
|
|
Tenant: utils.CGRateSorg,
|
|
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
Value: fltr,
|
|
GroupIDs: []string{},
|
|
}
|
|
|
|
expErr := "error parsing regexp: invalid or unsupported Perl syntax: `(?!`"
|
|
var reply string
|
|
if err := cacheS.V1ReplicateSet(context.Background(), args, &reply); err == nil || err.Error() != expErr {
|
|
t.Errorf("Expected error <%v>, received error <%v>", expErr, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSCacheDataFromDB(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
attrs := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
atrPrfl := &utils.AttributeProfile{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "TEST_ATTRIBUTES_TEST",
|
|
Attributes: []*utils.Attribute{
|
|
{
|
|
Path: "*opts.RateSProfile",
|
|
Type: utils.MetaConstant,
|
|
Value: utils.NewRSRParsersMustCompile("RP_2", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blockers: utils.DynamicBlockers{
|
|
{
|
|
Blocker: false,
|
|
},
|
|
},
|
|
}
|
|
if err := dm.SetAttributeProfile(context.Background(), atrPrfl, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if _, err := dm.GetAttributeProfile(context.Background(), utils.CGRateSorg, "TEST_ATTRIBUTES_TEST", true, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := utils.OK
|
|
var reply string
|
|
if err := cacheS.cacheDataFromDB(context.Background(), attrs, &reply, false); err != nil {
|
|
t.Error(err)
|
|
} else if exp != reply {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if !reflect.DeepEqual(rcv, atrPrfl) {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", atrPrfl, rcv)
|
|
}
|
|
}
|
|
|
|
func TestCacheScacheDataFromDBErrCacheDataFromDB(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cacheS := NewCacheS(cfg, nil, connMgr, nil)
|
|
|
|
attrs := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
|
|
expErr := utils.ErrNoDatabaseConn
|
|
var reply string
|
|
|
|
if err := cacheS.cacheDataFromDB(context.Background(), attrs, &reply, true); err == nil || err != expErr {
|
|
t.Errorf("Expected error <%v>, received error <%v>", expErr, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheScacheDataFromDBErrGetItemLoadIDs(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
attrs := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
|
|
dm.dataDB = &DataDBMock{
|
|
GetItemLoadIDsDrvF: func(ctx *context.Context, itemIDPrefix string) (loadIDs map[string]int64, err error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
|
|
expErr := utils.ErrNotImplemented
|
|
var reply string
|
|
|
|
if err := cacheS.cacheDataFromDB(context.Background(), attrs, &reply, true); err == nil || err != expErr {
|
|
t.Errorf("Expected error <%v>, received error <%v>", expErr, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1LoadCache(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
attrs := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
atrPrfl := &utils.AttributeProfile{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "TEST_ATTRIBUTES_TEST",
|
|
Attributes: []*utils.Attribute{
|
|
{
|
|
Path: "*opts.RateSProfile",
|
|
Type: utils.MetaConstant,
|
|
Value: utils.NewRSRParsersMustCompile("RP_2", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blockers: utils.DynamicBlockers{
|
|
{
|
|
Blocker: false,
|
|
},
|
|
},
|
|
}
|
|
if err := dm.SetAttributeProfile(context.Background(), atrPrfl, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := utils.OK
|
|
var reply string
|
|
if err := cacheS.V1LoadCache(context.Background(), attrs, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if exp != reply {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if !reflect.DeepEqual(rcv, atrPrfl) {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", atrPrfl, rcv)
|
|
}
|
|
}
|
|
|
|
func TestCacheSV1ReloadCache(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
attrs := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
|
|
atrPrfl := &utils.AttributeProfile{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "TEST_ATTRIBUTES_TEST",
|
|
Attributes: []*utils.Attribute{
|
|
{
|
|
Path: "*opts.RateSProfile",
|
|
Type: utils.MetaConstant,
|
|
Value: utils.NewRSRParsersMustCompile("RP_2", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blockers: utils.DynamicBlockers{
|
|
{
|
|
Blocker: false,
|
|
},
|
|
},
|
|
}
|
|
if err := dm.SetAttributeProfile(context.Background(), atrPrfl, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if _, err := dm.GetAttributeProfile(context.Background(), utils.CGRateSorg, "TEST_ATTRIBUTES_TEST", true, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := utils.OK
|
|
var reply string
|
|
if err := cacheS.V1ReloadCache(context.Background(), attrs, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if exp != reply {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if !reflect.DeepEqual(rcv, atrPrfl) {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", atrPrfl, rcv)
|
|
}
|
|
}
|
|
|
|
func TestCacheSV1RemoveGroup(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
args := &utils.ArgsGetGroupWithAPIOpts{
|
|
Tenant: "cgrates.org",
|
|
APIOpts: map[string]any{},
|
|
ArgsGetGroup: utils.ArgsGetGroup{
|
|
CacheID: utils.CacheAccounts,
|
|
GroupID: "Group",
|
|
},
|
|
}
|
|
|
|
if err := Cache.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{"Group", "group2"}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
exp := utils.OK
|
|
var reply string
|
|
if err := cacheS.V1RemoveGroup(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if exp != reply {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
var hasRply bool
|
|
if err := cacheS.V1HasGroup(context.Background(), args, &hasRply); err != nil {
|
|
t.Error(err)
|
|
} else if hasRply {
|
|
t.Error("There are groups in cacheS")
|
|
}
|
|
|
|
}
|
|
|
|
func TestV1GetCacheStats(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
args := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{"cacheId1"},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
|
|
if err := Cache.Set(context.Background(), "cacheId1", "itemId", "valinterface", []string{"GroupId"}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := cacheS.tCache.GetCacheStats(args.CacheIDs)
|
|
var reply map[string]*ltcache.CacheStats
|
|
if err := cacheS.V1GetCacheStats(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if !reflect.DeepEqual(reply, exp) {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1Clear(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
args := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{"cacheId1", "cacheId2", "cacheId3", "cacheId4"},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
|
|
if err := Cache.Set(context.Background(), "cacheId1", "itemId", "valinterface", []string{"GroupId"}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
var reply string
|
|
if err := cacheS.V1Clear(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Expected <%v>, Received <%v>", utils.OK, reply)
|
|
}
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAccounts, "cacheId1"); ok {
|
|
t.Errorf("Cache.Get ok shouldnt be true, received <%v>", rcv)
|
|
} else if rcv != nil {
|
|
t.Errorf("Expected <%v>, Received <%v>", nil, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1RemoveItems(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
args := &utils.AttrReloadCacheWithAPIOpts{
|
|
Tenant: utils.CGRateSorg,
|
|
AttributeProfileIDs: []string{"cgrates.org:TEST_ATTRIBUTES_TEST"},
|
|
}
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
expNotRemovedGet := "valinterface"
|
|
if rcv, ok := cacheS.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if rcv != expNotRemovedGet {
|
|
t.Errorf("Expected <%v>, Received <%v>", expNotRemovedGet, rcv)
|
|
}
|
|
var reply string
|
|
if err := cacheS.V1RemoveItems(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Expected <%v>, Received <%v>", utils.OK, reply)
|
|
}
|
|
|
|
if rcv, ok := cacheS.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); ok {
|
|
t.Errorf("Cache.Get shouldnt receive ok, received <%v>", rcv)
|
|
} else if rcv != nil {
|
|
t.Errorf("Expected <%v>, Received <%v>", nil, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1RemoveSingular(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
expNotRemovedGet := "valinterface"
|
|
if rcv, ok := cacheS.Get(utils.CacheAccounts, "itemId"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if rcv != expNotRemovedGet {
|
|
t.Errorf("Expected <%v>, Received <%v>", expNotRemovedGet, rcv)
|
|
}
|
|
var reply string
|
|
if err := cacheS.V1RemoveItem(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Expected <%v>, Received <%v>", utils.OK, reply)
|
|
}
|
|
|
|
if rcv, ok := cacheS.Get(utils.CacheAccounts, "itemId"); ok {
|
|
t.Errorf("Cache.Get shouldnt receive ok, received <%v>", rcv)
|
|
} else if rcv != nil {
|
|
t.Errorf("Expected <%v>, Received <%v>", nil, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1GetItemExpiryTime(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Limit: 1,
|
|
TTL: 5 * time.Second,
|
|
},
|
|
}
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := time.Now().Year()
|
|
var reply time.Time
|
|
if err := cacheS.V1GetItemExpiryTime(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply.Year() != exp {
|
|
t.Errorf("Expected <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestV1GetItemSingular(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := any("valinterface")
|
|
var reply any
|
|
if err := cacheS.V1GetItem(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != exp {
|
|
t.Errorf("Expected <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1HasItem(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
args := &utils.ArgsGetCacheItemWithAPIOpts{
|
|
ArgsGetCacheItem: utils.ArgsGetCacheItem{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemID: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := true
|
|
var reply bool
|
|
if err := cacheS.V1HasItem(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != exp {
|
|
t.Errorf("Expected <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1GetItemIDs(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
args := &utils.ArgsGetCacheItemIDsWithAPIOpts{
|
|
ArgsGetCacheItemIDs: utils.ArgsGetCacheItemIDs{
|
|
CacheID: utils.CacheAccounts,
|
|
ItemIDPrefix: "itemId",
|
|
},
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := []string{"itemId"}
|
|
var reply []string
|
|
if err := cacheS.V1GetItemIDs(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if !reflect.DeepEqual(exp, reply) {
|
|
t.Errorf("Expected <%v>, Received <%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
type ccCloner struct {
|
|
mckField string
|
|
}
|
|
|
|
func (cc *ccCloner) Clone() (any, error) {
|
|
cc.mckField = "value"
|
|
return cc, nil
|
|
}
|
|
|
|
func TestCacheSGetPrecacheChannel(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := cacheS.Set(context.Background(), utils.MetaAccounts, "itemId", "valinterface", nil, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
exp := cacheS.pcItems[utils.MetaAccounts]
|
|
|
|
if rcv := cacheS.GetPrecacheChannel(utils.MetaAccounts); exp != rcv {
|
|
t.Errorf("Expected <%v>, Received <%v>", exp, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1PrecacheStatusDefault(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
args := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expArgs := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
expArgs.CacheIDs = utils.CachePartitions.AsSlice()
|
|
pCacheStatus := make(map[string]string)
|
|
for _, cacheID := range expArgs.CacheIDs {
|
|
pCacheStatus[cacheID] = utils.MetaPrecaching
|
|
}
|
|
|
|
exp := pCacheStatus
|
|
|
|
var reply map[string]string
|
|
if err := cacheS.V1PrecacheStatus(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if !reflect.DeepEqual(reply, exp) {
|
|
t.Errorf("Expected \n<%v>,\nReceived \n<%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1PrecacheStatusErrUnknownCacheID(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
args := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{"Inproper ID"},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expErr := "unknown cacheID: Inproper ID"
|
|
var reply map[string]string
|
|
if err := cacheS.V1PrecacheStatus(context.Background(), args, &reply); err == nil || err.Error() != expErr {
|
|
t.Errorf("Expected error <%v>, error received <%v>", expErr, err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSV1PrecacheStatusMetaReady(t *testing.T) {
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
args := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaSubsys: utils.MetaChargers,
|
|
},
|
|
CacheIDs: []string{utils.MetaAccounts},
|
|
Tenant: "cgrates.org",
|
|
}
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
if err := Cache.Set(context.Background(), utils.MetaAccounts, "itemId", "valinterface", nil, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
go func() {
|
|
cacheS.GetPrecacheChannel(utils.MetaAccounts) <- struct{}{}
|
|
}()
|
|
time.Sleep(10 * time.Millisecond)
|
|
|
|
pCacheStatus := make(map[string]string)
|
|
pCacheStatus[utils.MetaAccounts] = utils.MetaReady
|
|
|
|
exp := pCacheStatus
|
|
var reply map[string]string
|
|
if err := cacheS.V1PrecacheStatus(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if !reflect.DeepEqual(reply, exp) {
|
|
t.Errorf("Expected \n<%v>,\nReceived \n<%v>", exp, reply)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSPrecachePartitions(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
utils.CacheAccounts: {
|
|
Precache: false,
|
|
},
|
|
utils.MetaAttributeProfiles: {
|
|
Precache: true,
|
|
},
|
|
}
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
atrPrfl := &utils.AttributeProfile{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "TEST_ATTRIBUTES_TEST",
|
|
Attributes: []*utils.Attribute{
|
|
{
|
|
Path: "*opts.RateSProfile",
|
|
Type: utils.MetaConstant,
|
|
Value: utils.NewRSRParsersMustCompile("RP_2", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blockers: utils.DynamicBlockers{
|
|
{
|
|
Blocker: false,
|
|
},
|
|
},
|
|
}
|
|
if err := dm.SetAttributeProfile(context.Background(), atrPrfl, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if _, err := dm.GetAttributeProfile(context.Background(), utils.CGRateSorg, "TEST_ATTRIBUTES_TEST", true, true, utils.NonTransactional); err != nil {
|
|
t.Error(err)
|
|
}
|
|
cacheS.Precache(utils.NewSyncedChan())
|
|
time.Sleep(10 * time.Millisecond)
|
|
|
|
if rcv, ok := Cache.Get(utils.CacheAttributeProfiles, "cgrates.org:TEST_ATTRIBUTES_TEST"); !ok {
|
|
t.Errorf("Cache.Get did not receive ok, received <%v>", rcv)
|
|
} else if !reflect.DeepEqual(rcv, atrPrfl) {
|
|
t.Errorf("Expected rcv <%v>, Received <%v>", atrPrfl, rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSPrecacheErr(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
tmpLog := utils.Logger
|
|
defer func() {
|
|
Cache = tmp
|
|
utils.Logger = tmpLog
|
|
}()
|
|
|
|
Cache.Clear(nil)
|
|
|
|
buf := new(bytes.Buffer)
|
|
utils.Logger = utils.NewStdLoggerWithWriter(buf, "", 7)
|
|
|
|
args := &utils.ArgCacheReplicateSet{
|
|
CacheID: utils.CacheAccounts,
|
|
}
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.CacheCfg().Partitions = map[string]*config.CacheParamCfg{
|
|
args.CacheID: {
|
|
Precache: true,
|
|
},
|
|
}
|
|
|
|
cacheS := NewCacheS(cfg, nil, connMgr, nil)
|
|
|
|
cacheS.Precache(utils.NewSyncedChan())
|
|
time.Sleep(10 * time.Millisecond)
|
|
expErr := "<CacheS> precaching cacheID <*accounts>, got error: NO_DATABASE_CONNECTION"
|
|
|
|
if rcvTxt := buf.String(); !strings.Contains(rcvTxt, expErr) {
|
|
t.Errorf("Expected <%v>, Received <%v>", expErr, rcvTxt)
|
|
}
|
|
|
|
buf.Reset()
|
|
|
|
}
|
|
|
|
func TestCacheSBeginTransaction(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expFormat := `........-....-....-....-............`
|
|
rcv := cacheS.BeginTransaction()
|
|
if matched, err := regexp.Match(expFormat, []byte(rcv)); err != nil {
|
|
t.Error(err)
|
|
} else if !matched {
|
|
t.Errorf("Unexpected transaction format, Received <%v>", rcv)
|
|
}
|
|
|
|
}
|
|
|
|
func TestCacheSRollbackTransaction(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expFormat := `........-....-....-....-............`
|
|
tranId := cacheS.BeginTransaction()
|
|
if matched, err := regexp.Match(expFormat, []byte(tranId)); err != nil {
|
|
t.Error(err)
|
|
} else if !matched {
|
|
t.Errorf("Unexpected transaction format, Received <%v>", tranId)
|
|
}
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, tranId); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if rcv, ok := cacheS.Get(utils.CacheAccounts, "itemId"); !ok {
|
|
t.Errorf("Cache.Get should receive ok, received <%v>", rcv)
|
|
} else if rcv != "valinterface" {
|
|
t.Errorf("Expected <%v>, Received <%v>", "valinterface", rcv)
|
|
}
|
|
|
|
// destroys a transaction from transactions buffer
|
|
cacheS.RollbackTransaction(tranId)
|
|
|
|
}
|
|
|
|
func TestCacheSCommitTransaction(t *testing.T) {
|
|
|
|
tmp := Cache
|
|
defer func() {
|
|
Cache = tmp
|
|
}()
|
|
Cache.Clear(nil)
|
|
|
|
cfg := config.NewDefaultCGRConfig()
|
|
db, _ := NewInternalDB(nil, nil, nil, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg, nil)
|
|
|
|
cacheS := NewCacheS(cfg, dm, connMgr, nil)
|
|
|
|
expFormat := `........-....-....-....-............`
|
|
tranId := cacheS.BeginTransaction()
|
|
if matched, err := regexp.Match(expFormat, []byte(tranId)); err != nil {
|
|
t.Error(err)
|
|
} else if !matched {
|
|
t.Errorf("Unexpected transaction format, Received <%v>", tranId)
|
|
}
|
|
|
|
if err := cacheS.Set(context.Background(), utils.CacheAccounts, "itemId", "valinterface", []string{}, true, tranId); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if rcv, ok := cacheS.Get(utils.CacheAccounts, "itemId"); !ok {
|
|
t.Errorf("Cache.Get should receive ok, received <%v>", rcv)
|
|
} else if rcv != "valinterface" {
|
|
t.Errorf("Expected <%v>, Received <%v>", "valinterface", rcv)
|
|
}
|
|
|
|
// executes the actions in a transaction buffer
|
|
cacheS.CommitTransaction(tranId)
|
|
|
|
}
|