Add unit tests on engine

engine/balances_test.go

@@ -631,3 +631,38 @@ func TestBalancesSetInitialValue(t *testing.T) {
 		}
 	})
 }
+
+func TestBalancesValueFactorsGetValue(t *testing.T) {
+	testCases := []struct {
+		factors  ValueFactors
+		category string
+		expected float64
+	}{
+		{
+			factors:  nil,
+			category: "category1",
+			expected: 1.0,
+		},
+		{
+			factors:  ValueFactors{"category2": 2.0, "category3": 3.0},
+			category: "category1",
+			expected: 1.0,
+		},
+		{
+			factors:  ValueFactors{"category1": 1.5, "category2": 2.0, "category3": 3.0},
+			category: "category1",
+			expected: 1.5,
+		},
+		{
+			factors:  ValueFactors{"category1": 0.0, "category2": 2.0, "category3": 3.0},
+			category: "category1",
+			expected: 0.0,
+		},
+	}
+	for _, tc := range testCases {
+		result := tc.factors.GetValue(tc.category)
+		if result != tc.expected {
+			t.Errorf("Test failed for category '%s': expected %.2f, got %.2f", tc.category, tc.expected, result)
+		}
+	}
+}

engine/version_test.go

@@ -24,6 +24,7 @@ import (
 	"testing"
 
 	"github.com/cgrates/cgrates/utils"
+	"github.com/google/go-cmp/cmp"
 )
 
 func TestVersionCompare(t *testing.T) {
@@ -309,3 +310,27 @@ func TestVersionCheckVersionsCompareNonNilMsg(t *testing.T) {
 		t.Errorf("expected error to be of type: <Migration needed: please backup cgr data and run : ..., \n received: <%+v>", err)
 	}
 }
+
+func TestVersionCurrentAllDBVersions(t *testing.T) {
+	expectedDataVersions := CurrentDataDBVersions()
+	expectedStorVersions := CurrentStorDBVersions()
+	allVersions := CurrentAllDBVersions()
+	for key, value := range expectedDataVersions {
+		if allVersions[key] != value {
+			t.Errorf("Data version mismatch for key %s. Expected: %d, Got: %d", key, value, allVersions[key])
+		}
+	}
+	for key, value := range expectedStorVersions {
+		if allVersions[key] != value {
+			t.Errorf("Storage version mismatch for key %s. Expected: %d, Got: %d", key, value, allVersions[key])
+		}
+	}
+}
+
+func TestVersionCurrentDBVersionsInternal(t *testing.T) {
+	internalVersions := CurrentDBVersions(utils.MetaInternal, true)
+	expectedInternalVersions := CurrentAllDBVersions()
+	if !cmp.Equal(internalVersions, expectedInternalVersions) {
+		t.Errorf("Internal: Versions mismatch")
+	}
+}