Add tests for concureqs.go

utils/concureqs_test.go

@@ -0,0 +1,133 @@
+/*
+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 utils
+
+import (
+	"testing"
+)
+
+func TestConcureqsNewConReqs(t *testing.T) {
+	expected := &ConcReqs{
+		limit:    5,
+		strategy: MetaBusy,
+		aReqs:    make(chan struct{}, 5),
+	}
+	for i := 0; i < 5; i++ {
+		expected.aReqs <- struct{}{}
+	}
+	rcv := NewConReqs(5, MetaBusy)
+
+	if rcv.limit != expected.limit || rcv.strategy != expected.strategy || len(rcv.aReqs) != len(expected.aReqs) {
+		t.Errorf("expected: <%+v>, \nrecived: <%+v>", expected, rcv)
+	}
+}
+
+func TestConcureqsAllocate(t *testing.T) {
+
+	cases := []struct {
+		name         string
+		limit        int
+		rcv          int
+		emptyChannel bool
+		err          error
+	}{
+		{
+			name: "if limit is equal to 0",
+			limit: 0,
+			rcv: 0,
+			emptyChannel: false,
+			err: nil,
+		},
+		{
+			name: "limit more than 0",
+			limit: 5,
+			rcv: 4,
+			emptyChannel: false,
+			err: nil,
+		},
+		{
+			name: "limit more than 0 but channel is empty",
+			limit: 5,
+			rcv: 0,
+			emptyChannel: true,
+			err: errDeny,
+		},
+	}
+
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			cr := NewConReqs(c.limit, MetaBusy)
+
+			if c.emptyChannel {
+				for i := 0; i < c.limit; i++ {
+					<-cr.aReqs
+				}
+			}
+
+			err := cr.Allocate()
+
+			if err != c.err {
+				t.Fatal(err)
+			}
+
+			if len(cr.aReqs) != c.rcv {
+				t.Errorf("expected: <%+v>, recived: <%+v>", len(cr.aReqs), c.rcv)
+			}
+		})
+	}
+}
+
+func TestConcureqsDeallocate(t *testing.T) {
+
+	cases := []struct {
+		name     string
+		limit    int
+	}{
+		{
+			name: "limit is 0",
+			limit: 0,
+		},
+		{
+			name: "limit is not 0",
+			limit: 5,
+		},
+	}
+
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			cr := NewConReqs(c.limit, MetaBusy)
+
+			if c.limit != 0 {
+				go cr.Deallocate()
+				_, ok := <-cr.aReqs
+
+				if !ok {
+					t.Error("didn't recive from the channel")
+				}
+			} else {
+				cr.Deallocate()
+
+				if len(cr.aReqs) != c.limit {
+					t.Errorf("expected: <%+v>, recived: <%+v>", len(cr.aReqs), c.limit)
+				}
+			}
+
+		})
+	}
+}