From 7f3a9879dedcb02f713b86fddf81d184e5326bca Mon Sep 17 00:00:00 2001
From: andronache <andrei.andronache@itsyscom.com>
Date: Thu, 20 May 2021 15:03:26 +0300
Subject: [PATCH] Fixed tests

---
 accounts/accounts_test.go       | 2 +-
 accounts/libaccounts_test.go    | 2 +-
 actions/actions_test.go         | 2 +-
 dispatchers/dispatchers_test.go | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/accounts/accounts_test.go b/accounts/accounts_test.go
index 17a46d59c..de9d1203d 100644
--- a/accounts/accounts_test.go
+++ b/accounts/accounts_test.go
@@ -72,7 +72,7 @@ func TestShutDownCoverage(t *testing.T) {
 }
 
 type dataDBMockErrorNotFound struct {
-	*engine.DataDBMock
+	engine.DataDBMock
 }
 
 func (dB *dataDBMockErrorNotFound) GetAccountDrv(*context.Context, string, string) (*utils.Account, error) {
diff --git a/accounts/libaccounts_test.go b/accounts/libaccounts_test.go
index 6a622d82a..7520f375d 100644
--- a/accounts/libaccounts_test.go
+++ b/accounts/libaccounts_test.go
@@ -546,7 +546,7 @@ func TestRestoreAccount(t *testing.T) { //coverage purpose
 }
 
 type dataDBMockError struct {
-	*engine.DataDBMock
+	engine.DataDBMock
 }
 
 func TestRestoreAccount2(t *testing.T) { //coverage purpose
diff --git a/actions/actions_test.go b/actions/actions_test.go
index 677905b0e..0166b5d3b 100644
--- a/actions/actions_test.go
+++ b/actions/actions_test.go
@@ -496,7 +496,7 @@ func TestActionShutDown(t *testing.T) {
 }
 
 type dataDBMockError struct {
-	*engine.DataDBMock
+	engine.DataDBMock
 }
 
 func (dbM *dataDBMockError) GetActionProfileDrv(*context.Context, string, string) (*engine.ActionProfile, error) {
diff --git a/dispatchers/dispatchers_test.go b/dispatchers/dispatchers_test.go
index c97d4f28b..d5d0f8895 100644
--- a/dispatchers/dispatchers_test.go
+++ b/dispatchers/dispatchers_test.go
@@ -844,7 +844,7 @@ func TestDispatcherServiceDispatcherProfileForEventNotNotFound(t *testing.T) {
 	var cnt int
 
 	dm := engine.NewDataManager(&engine.DataDBMock{
-		GetIndexesDrvF: func(idxItmType, tntCtx, idxKey string) (indexes map[string]utils.StringSet, err error) {
+		GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey string) (indexes map[string]utils.StringSet, err error) {
 			if cnt == 0 {
 				cnt++
 				return map[string]utils.StringSet{