From 27b55e32e5408ca050f7d53ef3826e847bf542fa Mon Sep 17 00:00:00 2001
From: TeoV <teofil.voivozeanu@itsyscom.com>
Date: Tue, 17 Dec 2019 10:35:38 -0500
Subject: [PATCH] Instead of shutdown the US_SITE send the UpdateSession1 only
 to AU_SITE

---
 general_tests/gocs_it_test.go | 34 ++++++++--------------------------
 1 file changed, 8 insertions(+), 26 deletions(-)

diff --git a/general_tests/gocs_it_test.go b/general_tests/gocs_it_test.go
index b4a2aaa22..308a6918b 100644
--- a/general_tests/gocs_it_test.go
+++ b/general_tests/gocs_it_test.go
@@ -51,9 +51,7 @@ var (
 		testGOCSLoadData,
 		testGOCSAuthSession,
 		testGOCSInitSession,
-		testGOCSKillUSEngine,
 		testGOCSUpdateSession,
-		testGOCSStartUSEngine,
 		testGOCSVerifyAccountsAfterStart,
 		testGOCSUpdateSession2,
 		testGOCSTerminateSession,
@@ -329,13 +327,6 @@ func testGOCSInitSession(t *testing.T) {
 
 }
 
-func testGOCSKillUSEngine(t *testing.T) {
-	if err := usEngine.Process.Kill(); err != nil {
-		t.Error(err)
-	}
-	time.Sleep(10 * time.Millisecond)
-}
-
 func testGOCSUpdateSession(t *testing.T) {
 	reqUsage := 5 * time.Minute
 	args := &sessions.V1UpdateSessionArgs{
@@ -362,7 +353,7 @@ func testGOCSUpdateSession(t *testing.T) {
 
 	// right now dispatcher receive utils.ErrPartiallyExecuted
 	// in case of of engines fails
-	if err := dspRPC.Call(utils.SessionSv1UpdateSession, args, &rply); err == nil || err.Error() != utils.ErrPartiallyExecuted.Error() {
+	if err := auRPC.Call(utils.SessionSv1UpdateSession, args, &rply); err != nil {
 		t.Errorf("Expecting : %+v, received: %+v", utils.ErrPartiallyExecuted, err)
 	}
 
@@ -393,15 +384,6 @@ func testGOCSUpdateSession(t *testing.T) {
 
 }
 
-func testGOCSStartUSEngine(t *testing.T) {
-	if usEngine, err = engine.StartEngine(usCfgPath, *waitRater); err != nil {
-		t.Fatal(err)
-	}
-	if usRPC, err = newRPCClient(usCfg.ListenCfg()); err != nil {
-		t.Fatal(err)
-	}
-}
-
 func testGOCSVerifyAccountsAfterStart(t *testing.T) {
 	var acnt *engine.Account
 	attrAcc := &utils.AttrGetAccount{
@@ -471,7 +453,7 @@ func testGOCSUpdateSession2(t *testing.T) {
 		t.Errorf("wrong active sessions: %s \n , and len(aSessions) %+v", utils.ToJSON(aSessions), len(aSessions))
 	} else if aSessions[0].NodeID != "US_SITE" {
 		t.Errorf("Expecting : %+v, received: %+v", "US_SITE", aSessions[0].NodeID)
-	} else if aSessions[0].Usage != time.Duration(5*time.Minute) {
+	} else if aSessions[0].Usage != time.Duration(10*time.Minute) {
 		t.Errorf("Expecting : %+v, received: %+v", time.Duration(5*time.Minute), aSessions[0].Usage)
 	}
 
@@ -543,14 +525,14 @@ func testGOCSTerminateSession(t *testing.T) {
 
 	if err := auRPC.Call(utils.ApierV2GetAccount, attrAcc, &acnt); err != nil {
 		t.Error(err)
-	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2340000000000.0 {
-		t.Errorf("Expecting : %+v, received: %+v", 2340000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
+	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2640000000000.0 {
+		t.Errorf("Expecting : %+v, received: %+v", 2640000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
 	}
 
 	if err := usRPC.Call(utils.ApierV2GetAccount, attrAcc, &acnt); err != nil {
 		t.Error(err)
-	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2340000000000.0 {
-		t.Errorf("Expecting : %+v, received: %+v", 2340000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
+	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2640000000000.0 {
+		t.Errorf("Expecting : %+v, received: %+v", 2640000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
 	}
 }
 
@@ -593,13 +575,13 @@ func testGOCSProcessCDR(t *testing.T) {
 
 	if err := auRPC.Call(utils.ApierV2GetAccount, attrAcc, &acnt); err != nil {
 		t.Error(err)
-	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2340000000000.0 {
+	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2640000000000.0 {
 		t.Errorf("Expecting : %+v, received: %+v", 2640000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
 	}
 
 	if err := usRPC.Call(utils.ApierV2GetAccount, attrAcc, &acnt); err != nil {
 		t.Error(err)
-	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2340000000000.0 {
+	} else if acnt.BalanceMap[utils.VOICE].GetTotalValue() != 2640000000000.0 {
 		t.Errorf("Expecting : %+v, received: %+v", 2640000000000.0, acnt.BalanceMap[utils.VOICE].GetTotalValue())
 	}
 }