From 8e6f2c360c089d8fdab05c617b939c6c91c335d3 Mon Sep 17 00:00:00 2001
From: ionutboangiu <ionut.boangiu@itsyscom.com>
Date: Thu, 6 Jan 2022 17:52:31 +0200
Subject: [PATCH] Add test for paginate function

---
 utils/apitpdata_test.go | 63 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)

diff --git a/utils/apitpdata_test.go b/utils/apitpdata_test.go
index d49f23d3b..3bbf8bc23 100644
--- a/utils/apitpdata_test.go
+++ b/utils/apitpdata_test.go
@@ -620,3 +620,66 @@ func TestNewAttrReloadCacheWithOptsFromMap(t *testing.T) {
 		t.Errorf("Expected %+v \n, received %+v", ToJSON(mp), ToJSON(rplyM))
 	}
 }
+
+func TestAPITPDataPaginate(t *testing.T) {
+	in := []string{"FLTR_1", "FLTR_2", "FLTR_3", "FLTR_4", "FLTR_5", "FLTR_6", "FLTR_7",
+		"FLTR_8", "FLTR_9", "FLTR_10", "FLTR_11", "FLTR_12", "FLTR_13", "FLTR_14",
+		"FLTR_15", "FLTR_16", "FLTR_17", "FLTR_18", "FLTR_19", "FLTR_20"}
+	args := &ArgsItemIDs{
+		Tenant: "cgrates.org",
+		APIOpts: map[string]interface{}{
+			PageLimitOpt:    2,
+			PageOffsetOpt:   6,
+			PageMaxItemsOpt: 9,
+		},
+	}
+
+	exp := []string{"FLTR_7", "FLTR_8"}
+	if rcv, err := Paginate(in, args.APIOpts); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(rcv, exp) {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", exp, rcv)
+	}
+
+	args.APIOpts[PageLimitOpt] = 0
+	args.APIOpts[PageOffsetOpt] = 18
+	args.APIOpts[PageMaxItemsOpt] = 50
+	exp = []string{"FLTR_19", "FLTR_20"}
+	if rcv, err := Paginate(in, args.APIOpts); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(rcv, exp) {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", exp, rcv)
+	}
+
+	args.APIOpts[PageOffsetOpt] = 0
+	if rcv, err := Paginate(in, args.APIOpts); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(rcv, in) {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", in, rcv)
+	}
+
+	args.APIOpts[PageLimitOpt] = 25
+	args.APIOpts[PageOffsetOpt] = 18
+	if rcv, err := Paginate(in, args.APIOpts); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(rcv, exp) {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", exp, rcv)
+	}
+
+	args.APIOpts[PageLimitOpt] = 2
+	args.APIOpts[PageOffsetOpt] = 22
+	var expOut []string
+	if rcv, err := Paginate(in, args.APIOpts); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(rcv, expOut) {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", expOut, rcv)
+	}
+
+	args.APIOpts[PageLimitOpt] = 2
+	args.APIOpts[PageOffsetOpt] = 4
+	args.APIOpts[PageMaxItemsOpt] = 5
+	experr := `SERVER_ERROR: maximum number of items exceeded`
+	if _, err := Paginate(in, args.APIOpts); err == nil || err.Error() != experr {
+		t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
+	}
+}