From ed0beba2c1ed768cd6cf5f377dd42e9035b9665a Mon Sep 17 00:00:00 2001
From: andronache <andrei.andronache@itsyscom.com>
Date: Wed, 21 Apr 2021 13:46:29 +0300
Subject: [PATCH] Test for authorize in dispatchers

---
 dispatchers/dispatchers_test.go | 76 ++++++++++++++++++++++++++++++---
 1 file changed, 71 insertions(+), 5 deletions(-)

diff --git a/dispatchers/dispatchers_test.go b/dispatchers/dispatchers_test.go
index 938b44375..d6631aa1f 100644
--- a/dispatchers/dispatchers_test.go
+++ b/dispatchers/dispatchers_test.go
@@ -19,7 +19,6 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>
 package dispatchers
 
 import (
-	"fmt"
 	"testing"
 	"time"
 
@@ -280,7 +279,6 @@ func TestDispatcherServiceAuthorizeEventError3(t *testing.T) {
 type mockTypeCon3 struct{}
 
 func (*mockTypeCon3) Call(serviceMethod string, args, reply interface{}) error {
-	fmt.Println("yayCall")
 	eVreply := &engine.AttrSProcessEventReply{
 		CGREvent: &utils.CGREvent{
 			Tenant: "testTenant",
@@ -326,18 +324,34 @@ func TestDispatcherServiceAuthorizeError(t *testing.T) {
 	err := dsp.authorize(utils.APIMethods, "testTenant", "apikey", &time.Time{})
 	expected := "UNAUTHORIZED_API"
 	if err == nil || err.Error() != expected {
-		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
+		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
 	}
 	engine.Cache = cacheInit
 }
 
+type mockTypeCon4 struct{}
+
+func (*mockTypeCon4) Call(serviceMethod string, args, reply interface{}) error {
+	eVreply := &engine.AttrSProcessEventReply{
+		CGREvent: &utils.CGREvent{
+			Tenant:  "testTenant",
+			ID:      "testID",
+			Time:    nil,
+			Event:   map[string]interface{}{},
+			APIOpts: nil,
+		},
+	}
+	*reply.(*engine.AttrSProcessEventReply) = *eVreply
+	return nil
+}
+
 func TestDispatcherServiceAuthorizeError2(t *testing.T) {
 	cacheInit := engine.Cache
 	cfg := config.NewDefaultCGRConfig()
 	cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
 	dm := engine.NewDataManager(nil, nil, nil)
 	chanRPC := make(chan rpcclient.ClientConnector, 1)
-	chanRPC <- new(mockTypeCon3)
+	chanRPC <- new(mockTypeCon4)
 	rpcInt := map[string]chan rpcclient.ClientConnector{
 		utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
 	}
@@ -359,8 +373,60 @@ func TestDispatcherServiceAuthorizeError2(t *testing.T) {
 	engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
 		value, nil, true, utils.NonTransactional)
 	err := dsp.authorize(utils.APIMethods, "testTenant", "apikey", &time.Time{})
-	expected := "UNAUTHORIZED_API"
+	expected := "NOT_FOUND"
 	if err == nil || err.Error() != expected {
+		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
+	}
+	engine.Cache = cacheInit
+}
+
+type mockTypeCon5 struct{}
+
+func (*mockTypeCon5) Call(serviceMethod string, args, reply interface{}) error {
+	eVreply := &engine.AttrSProcessEventReply{
+		CGREvent: &utils.CGREvent{
+			Tenant: "testTenant",
+			ID:     "testID",
+			Time:   nil,
+			Event: map[string]interface{}{
+				utils.APIMethods: "testMethod",
+			},
+			APIOpts: nil,
+		},
+	}
+	*reply.(*engine.AttrSProcessEventReply) = *eVreply
+	return nil
+}
+
+func TestDispatcherServiceAuthorizeError3(t *testing.T) {
+	cacheInit := engine.Cache
+	cfg := config.NewDefaultCGRConfig()
+	cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
+	dm := engine.NewDataManager(nil, nil, nil)
+	chanRPC := make(chan rpcclient.ClientConnector, 1)
+	chanRPC <- new(mockTypeCon5)
+	rpcInt := map[string]chan rpcclient.ClientConnector{
+		utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
+	}
+	connMgr := engine.NewConnManager(cfg, rpcInt)
+
+	dsp := NewDispatcherService(dm, cfg, nil, connMgr)
+	value := &engine.DispatcherHost{
+		Tenant: "testTenant",
+		RemoteHost: &config.RemoteHost{
+			ID:          "testID",
+			Address:     rpcclient.InternalRPC,
+			Transport:   utils.MetaInternal,
+			Synchronous: false,
+			TLS:         false,
+		},
+	}
+	newCache := engine.NewCacheS(cfg, dm, nil)
+	engine.Cache = newCache
+	engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
+		value, nil, true, utils.NonTransactional)
+	err := dsp.authorize("testMethod", "testTenant", "apikey", &time.Time{})
+	if err != nil {
 		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
 	}
 	engine.Cache = cacheInit