From b4910419f74604906ba086ca65cfc2822734d221 Mon Sep 17 00:00:00 2001
From: TeoV <teofil.voivozeanu@itsyscom.com>
Date: Sun, 17 Mar 2019 13:21:05 +0200
Subject: [PATCH] Add test for caching methods for Responder

---
 general_tests/sessions_rpccaching_it_test.go | 98 ++++++++++++++++++++
 1 file changed, 98 insertions(+)

diff --git a/general_tests/sessions_rpccaching_it_test.go b/general_tests/sessions_rpccaching_it_test.go
index 6efe92594..0be1c9a39 100644
--- a/general_tests/sessions_rpccaching_it_test.go
+++ b/general_tests/sessions_rpccaching_it_test.go
@@ -60,6 +60,12 @@ var sTestsRPCMethods = []func(t *testing.T){
 	testRPCMethodsResetStorDb,
 	testRPCMethodsCdrsProcessCDR,
 	testRPCMethodsCdrsStoreSessionCost,
+	//reset the storDB and dataDB
+	testRPCMethodsInitDataDb,
+	testRPCMethodsResetStorDb,
+	testRPCMethodsLoadData,
+	testRPCMethodsResponderDebit,
+	testRPCMethodsResponderMaxDebit,
 	testRPCMethodsStopEngine,
 }
 
@@ -784,6 +790,98 @@ func testRPCMethodsCdrsStoreSessionCost(t *testing.T) {
 	}
 }
 
+// Load the tariff plan, creating accounts and their balances
+func testRPCMethodsLoadData(t *testing.T) {
+	attrs := &utils.AttrLoadTpFromFolder{FolderPath: path.Join(*dataDir, "tariffplans", "testtp")}
+	if err := rpcRpc.Call("ApierV2.LoadTariffPlanFromFolder", attrs, &tpLoadInst); err != nil {
+		t.Error(err)
+	}
+	time.Sleep(time.Duration(*waitRater) * time.Millisecond) // Give time for scheduler to execute topups
+}
+
+func testRPCMethodsResponderDebit(t *testing.T) {
+	tStart := time.Date(2016, 3, 31, 0, 0, 0, 0, time.UTC)
+	cd := engine.CallDescriptor{
+		CgrID:         "testRPCMethodsResponderDebit",
+		Category:      "call",
+		Tenant:        "cgrates.org",
+		Subject:       "1001",
+		Destination:   "+49",
+		DurationIndex: 0,
+		TimeStart:     tStart,
+		TimeEnd:       tStart.Add(time.Duration(15) * time.Second),
+	}
+	var cc engine.CallCost
+	//cache the response
+	if err := rpcRpc.Call(utils.ResponderDebit, cd, &cc); err != nil {
+		t.Error(err)
+	} else if cc.GetDuration() != 15*time.Second {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			15*time.Second, cc.GetDuration())
+	} else if cc.Cost != 15 {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			15, cc.Cost)
+	}
+	cd2 := engine.CallDescriptor{
+		CgrID: "testRPCMethodsResponderDebit",
+	}
+	var ccCache engine.CallCost
+	//cache the response
+	if err := rpcRpc.Call(utils.ResponderDebit, cd2, &ccCache); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(ccCache, cc) {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			utils.ToJSON(cc), utils.ToJSON(ccCache))
+	}
+	//give time to CGRateS to delete the response from cache
+	time.Sleep(1*time.Second + 500*time.Millisecond)
+	if err := rpcRpc.Call(utils.ResponderDebit, cd2, &cc); err == nil || err.Error() != "ACCOUNT_NOT_FOUND" {
+		t.Error("Unexpected error returned", err)
+	}
+}
+
+func testRPCMethodsResponderMaxDebit(t *testing.T) {
+	tStart := time.Date(2016, 3, 31, 0, 0, 0, 0, time.UTC)
+	cd := engine.CallDescriptor{
+		CgrID:         "testRPCMethodsResponderMaxDebit",
+		Category:      "call",
+		Tenant:        "cgrates.org",
+		Account:       "1001",
+		Subject:       "free",
+		Destination:   "+49",
+		DurationIndex: 0,
+		TimeStart:     tStart,
+		TimeEnd:       tStart.Add(time.Duration(15) * time.Second),
+	}
+	var cc engine.CallCost
+	//cache the response
+	if err := rpcRpc.Call(utils.ResponderMaxDebit, cd, &cc); err != nil {
+		t.Error(err)
+	} else if cc.GetDuration() != 15*time.Second {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			15*time.Second, cc.GetDuration())
+	} else if cc.Cost != 0 {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			0, cc.Cost)
+	}
+	cd2 := engine.CallDescriptor{
+		CgrID: "testRPCMethodsResponderMaxDebit",
+	}
+	var ccCache engine.CallCost
+	//cache the response
+	if err := rpcRpc.Call(utils.ResponderMaxDebit, cd2, &ccCache); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(ccCache, cc) {
+		t.Errorf("Expecting: %+v, \n received: %+v",
+			utils.ToJSON(cc), utils.ToJSON(ccCache))
+	}
+	//give time to CGRateS to delete the response from cache
+	time.Sleep(1*time.Second + 500*time.Millisecond)
+	if err := rpcRpc.Call(utils.ResponderMaxDebit, cd2, &cc); err == nil || err.Error() != "ACCOUNT_NOT_FOUND" {
+		t.Error("Unexpected error returned", err)
+	}
+}
+
 func testRPCMethodsStopEngine(t *testing.T) {
 	if err := engine.KillEngine(100); err != nil {
 		t.Error(err)