From e17d57b74dedf6cd5fbeccdec385f9ccb9714136 Mon Sep 17 00:00:00 2001
From: ionutboangiu <ionut.boangiu@itsyscom.com>
Date: Mon, 26 Apr 2021 12:47:37 +0300
Subject: [PATCH] Cover the remainder of chargers.go

---
 engine/chargers_test.go | 36 ++++++++++++++++++++++++++++++++----
 1 file changed, 32 insertions(+), 4 deletions(-)

diff --git a/engine/chargers_test.go b/engine/chargers_test.go
index 2be99fad1..c3553ddab 100644
--- a/engine/chargers_test.go
+++ b/engine/chargers_test.go
@@ -18,7 +18,11 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>
 package engine
 
 import (
+	"bytes"
+	"log"
+	"os"
 	"reflect"
+	"strings"
 	"testing"
 	"time"
 
@@ -667,7 +671,31 @@ func TestChargersV1ProcessEventMissingArgs(t *testing.T) {
 	}
 }
 
-// func TestChargersShutdown(t *testing.T) {
-// 	cS := &ChargerService{}
-// 	cS.Shutdown()
-// }
+func TestChargersShutdown(t *testing.T) {
+	cS := &ChargerService{}
+
+	utils.Logger.SetLogLevel(6)
+	utils.Logger.SetSyslog(nil)
+
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	exp := []string{
+		"CGRateS <> [INFO] <ChargerS> shutdown initialized",
+		"CGRateS <> [INFO] <ChargerS> shutdown complete",
+	}
+	cS.Shutdown()
+	rcv := strings.Split(buf.String(), "\n")
+
+	for i := 0; i < 2; i++ {
+		rcv[i] = rcv[i][20:]
+		if rcv[i] != exp[i] {
+			t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp[i], rcv[i])
+		}
+	}
+
+	utils.Logger.SetLogLevel(0)
+}