diff --git a/engine/balances_test.go b/engine/balances_test.go index 995ba9c9a..8b327c468 100644 --- a/engine/balances_test.go +++ b/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) + } + } +} diff --git a/engine/version_test.go b/engine/version_test.go index b7ca3c4be..1f33b02d2 100644 --- a/engine/version_test.go +++ b/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: ", 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") + } +}