Add coverage tests on engine

engine/filterhelpers_test.go

@@ -19,6 +19,7 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>
 package engine
 
 import (
+	"errors"
 	"reflect"
 	"testing"
 	"time"
@@ -135,3 +136,61 @@ func TestMatchingItemIDsForEventFilterIndexTypeNotNone(t *testing.T) {
 	}
 
 }
+
+func TestSentrypeerGetTokenErrorResponse(t *testing.T) {
+	tokenUrl := "Url"
+	clientID := "ID"
+	clientSecret := "clientSecret"
+	audience := "audience"
+	grantType := "grantType"
+
+	token, err := sentrypeerGetToken(tokenUrl, clientID, clientSecret, audience, grantType)
+	if err == nil {
+		t.Fatal("Expected error, got nil")
+	}
+	if token != "" {
+		t.Errorf("Expected empty token, got %v", token)
+	}
+}
+
+func TestExtractUrlFromType(t *testing.T) {
+	tests := []struct {
+		name     string
+		httpType string
+		wantURL  string
+		wantErr  error
+	}{
+		{
+			name:     "Valid input",
+			httpType: "http#cgrates.com",
+			wantURL:  "cgrates.com",
+			wantErr:  nil,
+		},
+		{
+			name:     "Incorrect format",
+			httpType: "http",
+			wantURL:  "",
+			wantErr:  errors.New("url is not specified"),
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotURL, err := extractUrlFromType(tt.httpType)
+
+			if tt.wantErr != nil {
+				if err == nil || err.Error() != tt.wantErr.Error() {
+					t.Errorf("Expected error %v, got %v", tt.wantErr, err)
+				}
+			} else {
+				if err != nil {
+					t.Fatalf("Expected no error, got %v", err)
+				}
+			}
+
+			if gotURL != tt.wantURL {
+				t.Errorf("Expected URL %q, got %q", tt.wantURL, gotURL)
+			}
+		})
+	}
+}

engine/storage_csv_test.go

@@ -0,0 +1,160 @@
+/*
+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 (
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/cgrates/cgrates/utils"
+)
+
+func TestNewFileCSVStorageSuccess(t *testing.T) {
+	tmpDir := t.TempDir()
+
+	subDirs := []string{"folder1", "folder2"}
+	for _, dir := range subDirs {
+		err := os.Mkdir(filepath.Join(tmpDir, dir), 0755)
+		if err != nil {
+			t.Fatalf("Could not create test directory %q: %v", dir, err)
+		}
+	}
+
+	storage, err := NewFileCSVStorage(',', tmpDir)
+	if err != nil {
+		t.Fatalf("Expected no error, got %v", err)
+	}
+
+	if storage == nil {
+		t.Fatalf("Expected CSVStorage, got nil")
+	}
+
+	var filteredResProfilesFn []string
+	for _, path := range storage.resProfilesFn {
+		if strings.Contains(path, "folder1") || strings.Contains(path, "folder2") {
+			filteredResProfilesFn = append(filteredResProfilesFn, path)
+		}
+	}
+
+	if len(filteredResProfilesFn) != 2 {
+		t.Fatalf("Expected 2 resource profile paths, got %d", len(filteredResProfilesFn))
+	}
+
+	expectedPaths := []string{
+		filepath.Join(tmpDir, "folder1", "Resources.csv"),
+		filepath.Join(tmpDir, "folder2", "Resources.csv"),
+	}
+
+	for i, path := range filteredResProfilesFn {
+		if !strings.EqualFold(path, expectedPaths[i]) {
+			t.Errorf("Expected %v, got %v", expectedPaths[i], path)
+		}
+	}
+}
+
+func TestNewCsvFile(t *testing.T) {
+	csv := NewCsvFile()
+
+	if csv == nil {
+		t.Fatalf("Expected a non-nil csvReaderCloser, got nil")
+	}
+
+	_, ok := csv.(*csvFile)
+	if !ok {
+		t.Fatalf("Expected type *csvFile, got %T", csv)
+	}
+}
+
+func TestGetTpTableIds(t *testing.T) {
+	csvStorage := &CSVStorage{}
+
+	tpid := "TPID"
+	table := "Table"
+	distinct := []string{"col1", "col2"}
+	filters := map[string]string{"key": "value"}
+	paginator := &utils.PaginatorWithSearch{}
+
+	ids, err := csvStorage.GetTpTableIds(tpid, table, distinct, filters, paginator)
+
+	if ids != nil {
+		t.Errorf("Expected ids to be nil, got %v", ids)
+	}
+
+	if err != utils.ErrNotImplemented {
+		t.Errorf("Expected error to be %v, got %v", utils.ErrNotImplemented, err)
+	}
+}
+
+func TestGetTpIds(t *testing.T) {
+	csvStorage := &CSVStorage{}
+
+	colName := "Column"
+
+	ids, err := csvStorage.GetTpIds(colName)
+
+	if ids != nil {
+		t.Errorf("Expected ids to be nil, got %v", ids)
+	}
+
+	if err != utils.ErrNotImplemented {
+		t.Errorf("Expected error to be %v, got %v", utils.ErrNotImplemented, err)
+	}
+}
+
+func TestJoinURL(t *testing.T) {
+	tests := []struct {
+		baseURL string
+		fn      string
+		want    string
+	}{
+		{"http://cgrates.org", "path/to/resource", "http://cgrates.org/path/to/resource"},
+		{"http://cgrates.org/", "path/to/resource", "http://cgrates.org/path/to/resource"},
+		{"http://cgrates.org/path/", "to/resource", "http://cgrates.org/path/to/resource"},
+		{"invalid-url", "path/to/resource", "invalid-url/path/to/resource"},
+		{"http://cgrates.org", "", "http://cgrates.org"},
+		{"http://cgrates.org/", "", "http://cgrates.org/"},
+		{"http://cgrates.org/path", "", "http://cgrates.org/path"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.baseURL+tt.fn, func(t *testing.T) {
+			got := joinURL(tt.baseURL, tt.fn)
+			if got != tt.want {
+				t.Errorf("joinURL(%q, %q) = %q; want %q", tt.baseURL, tt.fn, got, tt.want)
+			}
+		})
+	}
+}
+
+func TestNewURLCSVStorage(t *testing.T) {
+	sep := ','
+	dataPath := "http://cgrates.org/data1,http://cgrates.org/data2"
+
+	csvStorage := NewURLCSVStorage(sep, dataPath)
+
+	if csvStorage == nil {
+		t.Fatalf("Expected CSVStorage to be initialized, got nil")
+	}
+
+	if csvStorage.generator == nil {
+		t.Fatal("Expected generator to be set, got nil")
+	}
+}