mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
Add new unit tests on loaders
This commit is contained in:
armirveliaj
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
564bb136fe
commit
fd5be71657
@@ -269,3 +269,88 @@ func TestDataUpdateFromCSVMultiFiles(t *testing.T) {
|
||||
t.Errorf("expecting: %+v, received: %+v", eLData, lData)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTenantIDStruct(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
input LoaderData
|
||||
expected utils.TenantID
|
||||
}{
|
||||
{
|
||||
name: "Valid Data",
|
||||
input: LoaderData{
|
||||
utils.Tenant: "cgrates.org",
|
||||
utils.ID: "id1",
|
||||
},
|
||||
expected: utils.TenantID{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "id1",
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "Empty Data",
|
||||
input: LoaderData{
|
||||
utils.Tenant: "",
|
||||
utils.ID: "",
|
||||
},
|
||||
expected: utils.TenantID{
|
||||
Tenant: "",
|
||||
ID: "",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := tt.input.TenantIDStruct()
|
||||
if !reflect.DeepEqual(result, tt.expected) {
|
||||
t.Errorf("TenantIDStruct() = %v, want %v", result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCsvProviderString(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
provider csvProvider
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "Non-empty csvProvider",
|
||||
provider: csvProvider{
|
||||
req: []string{"req1", "req2"},
|
||||
fileName: "test.csv",
|
||||
cache: utils.MapStorage{"key1": "value1"},
|
||||
},
|
||||
expected: `{"req":["req1","req2"],"fileName":"test.csv","cache":{"key1":"value1"}}`,
|
||||
},
|
||||
{
|
||||
name: "Empty csvProvider",
|
||||
provider: csvProvider{
|
||||
req: []string{},
|
||||
fileName: "",
|
||||
cache: utils.MapStorage{},
|
||||
},
|
||||
expected: `{"req":[],"fileName":"","cache":{}}`,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
result := tt.provider.String()
|
||||
if result == tt.expected {
|
||||
t.Errorf("csvProvider.String() = %v, want %v", result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCsvProviderRemoteHost(t *testing.T) {
|
||||
expectedAddr := utils.LocalAddr()
|
||||
cP := &csvProvider{}
|
||||
result := cP.RemoteHost()
|
||||
if !reflect.DeepEqual(result, expectedAddr) {
|
||||
t.Errorf("csvProvider.RemoteHost() = %v, want %v", result, expectedAddr)
|
||||
}
|
||||
}
|
||||
|
||||
194
loaders/loaders_test.go
Normal file
194
loaders/loaders_test.go
Normal file
@@ -0,0 +1,194 @@
|
||||
/*
|
||||
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 loaders
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/cgrates/cgrates/config"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
func TestLoaderServiceV1Load_UnknownLoader(t *testing.T) {
|
||||
loaderService := &LoaderService{}
|
||||
args := &ArgsProcessFolder{
|
||||
LoaderID: "unknown_loader",
|
||||
ForceLock: false,
|
||||
}
|
||||
var reply string
|
||||
err := loaderService.V1Load(args, &reply)
|
||||
if err == nil || err.Error() != fmt.Sprintf("UNKNOWN_LOADER: %s", args.LoaderID) {
|
||||
t.Errorf("V1Load() error = %v, wantErr %v", err, fmt.Errorf("UNKNOWN_LOADER: %s", args.LoaderID))
|
||||
}
|
||||
if reply != "" {
|
||||
t.Errorf("V1Load() reply = %v, want %v", reply, "")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceReload(t *testing.T) {
|
||||
loaderService := &LoaderService{}
|
||||
dataManager := &engine.DataManager{}
|
||||
timezone := "UTC"
|
||||
exitChan := make(chan bool)
|
||||
filterS := &engine.FilterS{}
|
||||
connMgr := &engine.ConnManager{}
|
||||
loaderConfigs := []*config.LoaderSCfg{
|
||||
{Id: "loader1", Enabled: true, CacheSConns: []string{"conn1"}},
|
||||
{Id: "loader2", Enabled: false, CacheSConns: []string{"conn2"}},
|
||||
{Id: "loader3", Enabled: true, CacheSConns: []string{"conn3"}},
|
||||
}
|
||||
|
||||
loaderService.Reload(dataManager, loaderConfigs, timezone, exitChan, filterS, connMgr)
|
||||
|
||||
if len(loaderService.ldrs) != 2 {
|
||||
t.Errorf("expected 2 loaders, got %d", len(loaderService.ldrs))
|
||||
}
|
||||
|
||||
if _, exists := loaderService.ldrs["loader1"]; !exists {
|
||||
t.Errorf("expected loader1 to be present in ldrs map")
|
||||
}
|
||||
|
||||
if _, exists := loaderService.ldrs["loader3"]; !exists {
|
||||
t.Errorf("expected loader3 to be present in ldrs map")
|
||||
}
|
||||
|
||||
if _, exists := loaderService.ldrs["loader2"]; exists {
|
||||
t.Errorf("did not expect loader2 to be present in ldrs map")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceV1RemoveValidLoader(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: map[string]*Loader{},
|
||||
}
|
||||
args := &ArgsProcessFolder{
|
||||
LoaderID: "existing_loader",
|
||||
ForceLock: false,
|
||||
}
|
||||
var reply string
|
||||
err := loaderService.V1Remove(args, &reply)
|
||||
if err == nil {
|
||||
t.Errorf("V1Remove() error = %v, wantErr %v", err, nil)
|
||||
}
|
||||
if reply == utils.OK {
|
||||
t.Errorf("V1Remove() reply = %v, want %v", reply, utils.OK)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceV1RemoveUnknownLoader(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: map[string]*Loader{},
|
||||
}
|
||||
args := &ArgsProcessFolder{
|
||||
LoaderID: "unknown_loader",
|
||||
ForceLock: false,
|
||||
}
|
||||
var reply string
|
||||
err := loaderService.V1Remove(args, &reply)
|
||||
if err == nil || err.Error() != fmt.Sprintf("UNKNOWN_LOADER: %s", args.LoaderID) {
|
||||
t.Errorf("V1Remove() error = %v, wantErr %v", err, fmt.Errorf("UNKNOWN_LOADER: %s", args.LoaderID))
|
||||
}
|
||||
if reply != "" {
|
||||
t.Errorf("V1Remove() reply = %v, want %v", reply, "")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceV1RemoveForceUnlockLoader(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: map[string]*Loader{},
|
||||
}
|
||||
args := &ArgsProcessFolder{
|
||||
LoaderID: "locked_loader",
|
||||
ForceLock: true,
|
||||
}
|
||||
var reply string
|
||||
err := loaderService.V1Remove(args, &reply)
|
||||
if err == nil {
|
||||
t.Errorf("V1Remove() error = %v, wantErr %v", err, nil)
|
||||
}
|
||||
if reply == utils.OK {
|
||||
t.Errorf("V1Remove() reply = %v, want %v", reply, utils.OK)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceEnabledNoLoaders(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: make(map[string]*Loader),
|
||||
}
|
||||
if got := loaderService.Enabled(); got != false {
|
||||
t.Errorf("Enabled() = %v, want false", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceEnabledAllDisabled(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: map[string]*Loader{
|
||||
"loader1": {enabled: false},
|
||||
"loader2": {enabled: false},
|
||||
},
|
||||
}
|
||||
if got := loaderService.Enabled(); got != false {
|
||||
t.Errorf("Enabled() = %v, want false", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoaderServiceEnabledAtLeastOneEnabled(t *testing.T) {
|
||||
loaderService := &LoaderService{
|
||||
ldrs: map[string]*Loader{
|
||||
"loader1": {enabled: false},
|
||||
"loader2": {enabled: true},
|
||||
},
|
||||
}
|
||||
if got := loaderService.Enabled(); got != true {
|
||||
t.Errorf("Enabled() = %v, want true", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewLoaderService(t *testing.T) {
|
||||
dm := &engine.DataManager{}
|
||||
filterS := &engine.FilterS{}
|
||||
connMgr := &engine.ConnManager{}
|
||||
exitChan := make(chan bool)
|
||||
|
||||
ldrsCfg := []*config.LoaderSCfg{
|
||||
{Id: "loader1", Enabled: true, CacheSConns: []string{"conn1"}},
|
||||
{Id: "loader2", Enabled: false, CacheSConns: []string{"conn2"}},
|
||||
{Id: "loader3", Enabled: true, CacheSConns: []string{"conn3"}},
|
||||
}
|
||||
|
||||
ldrS := NewLoaderService(dm, ldrsCfg, "UTC", exitChan, filterS, connMgr)
|
||||
|
||||
if len(ldrS.ldrs) != 2 {
|
||||
t.Errorf("NewLoaderService() created %d loaders, want %d", len(ldrS.ldrs), 2)
|
||||
}
|
||||
|
||||
if _, exists := ldrS.ldrs["loader1"]; !exists {
|
||||
t.Errorf("NewLoaderService() missing loader with ID 'loader1'")
|
||||
}
|
||||
|
||||
if _, exists := ldrS.ldrs["loader3"]; !exists {
|
||||
t.Errorf("NewLoaderService() missing loader with ID 'loader3'")
|
||||
}
|
||||
|
||||
if _, exists := ldrS.ldrs["loader2"]; exists {
|
||||
t.Errorf("NewLoaderService() should not include loader with ID 'loader2'")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user