From d7e5ff23acec7fba4e1f5f752ff1e1ce26a6608f Mon Sep 17 00:00:00 2001
From: porosnicuadrian <adrian.porosnicu@itsyscom.com>
Date: Mon, 7 Feb 2022 16:18:42 +0200
Subject: [PATCH] Tested rates count/list apis

---
 apis/rates_it_test.go | 43 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/apis/rates_it_test.go b/apis/rates_it_test.go
index bada156fa..ca91b041d 100644
--- a/apis/rates_it_test.go
+++ b/apis/rates_it_test.go
@@ -49,7 +49,7 @@ var (
 		testRateSResetStorDb,
 		testRateSStartEngine,
 		testRateSRPCConn,
-		/* testGetRateProfileBeforeSet,
+		testGetRateProfileBeforeSet,
 		testGetRateProfilesBeforeSet,
 		testRateSetRateProfile,
 		testRateGetRateProfileIDs,
@@ -67,13 +67,15 @@ var (
 		testRateSetAttributeProfileBrokenReference,
 		testRateRemoveRateProfileRates,
 		testRateSetRateProfileRates,
-		testRateSetRateProfilesWithPrefix, */
+		testRateSetRateProfilesWithPrefix,
 		// here we will tests better the create,read,update and delte for the rates inside of a RateProfile
 		testRateProfileWithMultipleRates,
+		testRateProfileRateIDsAndCount,
 		testRateProfileUpdateRates,
 		testRateProfileRemoveMultipleRates,
 		testRateProfileSetMultipleRatesInProfile,
 		testRateProfileUpdateProfile,
+
 		testRateSKillEngine,
 	}
 )
@@ -1265,6 +1267,43 @@ func testRateProfileWithMultipleRates(t *testing.T) {
 	}
 }
 
+func testRateProfileRateIDsAndCount(t *testing.T) {
+	// we will get all the rates from MultipleRates rate profile with the prefix RT_T
+	expected := []string{"RT_THUESDAY", "RT_THURSDAY"}
+	var result []string
+	if err := rateSRPC.Call(context.Background(), utils.AdminSv1GetRateProfileRateIDs,
+		&utils.ArgsSubItemIDs{
+			Tenant:      "cgrates.org",
+			ProfileID:   "MultipleRates",
+			ItemsPrefix: "RT_T",
+		}, &result); err != nil {
+		t.Error(err)
+	} else {
+		sort.Slice(expected, func(i, j int) bool {
+			return expected[i] < expected[j]
+		})
+		sort.Slice(expected, func(i, j int) bool {
+			return result[i] < result[j]
+		})
+		if !reflect.DeepEqual(expected, result) {
+			t.Errorf("Expected %v, received %v", expected, result)
+		}
+
+	}
+
+	// we will count all the rates from MultipleRates rate profile. The prefix is missing, so it will count all the rates
+	var reply int
+	if err := rateSRPC.Call(context.Background(), utils.AdminSv1GetRateProfileRateCount,
+		&utils.ArgsSubItemIDs{
+			Tenant:    "cgrates.org",
+			ProfileID: "MultipleRates",
+		}, &reply); err != nil {
+		t.Error(err)
+	} else if reply != 5 {
+		t.Errorf("Unexpected reply returned: %v", reply)
+	}
+}
+
 func testRateProfileUpdateRates(t *testing.T) {
 	argsRate := &utils.APIRateProfile{
 		RateProfile: &utils.RateProfile{