Add a integration test for ReloadConfigFromJSON

2026-02-14 20:59:53 +05:00 · 2020-03-25 11:46:10 +02:00
parent 37cba7ae03
commit d1d401619e
1 changed files with 172 additions and 0 deletions
--- a/apier/v1/config_it_test.go
+++ b/apier/v1/config_it_test.go
@@ -0,0 +1,172 @@
+// +build integration
+
+/*
+Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
+Copyright (C) ITsysCOM GmbH
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>
+*/
+
+package v1
+
+import (
+	"net/rpc"
+	"path"
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/cgrates/cgrates/config"
+	"github.com/cgrates/cgrates/engine"
+	"github.com/cgrates/cgrates/utils"
+)
+
+var (
+	configCfgPath   string
+	configCfg       *config.CGRConfig
+	configRPC       *rpc.Client
+	configConfigDIR string //run tests for specific configuration
+
+	sTestsConfig = []func(t *testing.T){
+		testConfigSInitCfg,
+		testConfigSInitDataDb,
+		testConfigSResetStorDb,
+		testConfigSStartEngine,
+		testConfigSRPCConn,
+		testConfigSReloadConfigFromJSONSessionS,
+		testConfigSKillEngine,
+	}
+)
+
+//Test start here
+func TestConfigSIT(t *testing.T) {
+	switch *dbType {
+	case utils.MetaInternal:
+		configConfigDIR = "tutinternal"
+	case utils.MetaMySQL:
+		configConfigDIR = "tutmysql"
+	case utils.MetaMongo:
+		configConfigDIR = "tutmongo"
+	case utils.MetaPostgres:
+		t.SkipNow()
+	default:
+		t.Fatal("Unknown Database type")
+	}
+	for _, stest := range sTestsConfig {
+		t.Run(configConfigDIR, stest)
+	}
+}
+
+func testConfigSInitCfg(t *testing.T) {
+	var err error
+	configCfgPath = path.Join(*dataDir, "conf", "samples", configConfigDIR)
+	configCfg, err = config.NewCGRConfigFromPath(configCfgPath)
+	if err != nil {
+		t.Error(err)
+	}
+	configCfg.DataFolderPath = *dataDir
+	config.SetCgrConfig(configCfg)
+}
+
+func testConfigSInitDataDb(t *testing.T) {
+	if err := engine.InitDataDb(configCfg); err != nil {
+		t.Fatal(err)
+	}
+}
+
+// Wipe out the cdr database
+func testConfigSResetStorDb(t *testing.T) {
+	if err := engine.InitStorDb(configCfg); err != nil {
+		t.Fatal(err)
+	}
+}
+
+// Start CGR Engine
+func testConfigSStartEngine(t *testing.T) {
+	if _, err := engine.StopStartEngine(configCfgPath, *waitRater); err != nil {
+		t.Fatal(err)
+	}
+}
+
+// Connect rpc client to rater
+func testConfigSRPCConn(t *testing.T) {
+	var err error
+	configRPC, err = newRPCClient(configCfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+func testConfigSReloadConfigFromJSONSessionS(t *testing.T) {
+
+	var reply string
+	if err := configRPC.Call(utils.ConfigSv1ReloadConfigFromJSON, &config.JSONReloadWithArgDispatcher{
+		JSON: map[string]interface{}{
+			"sessions": map[string]interface{}{
+				"enabled":          true,
+				"resources_conns":  []string{"*localhost"},
+				"suppliers_conns":  []string{"*localhost"},
+				"attributes_conns": []string{"*localhost"},
+				"rals_conns":       []string{"*internal"},
+				"cdrs_conns":       []string{"*internal"},
+				"configs_conns":    []string{"*internal"},
+			},
+		},
+	}, &reply); err != nil {
+		t.Error(err)
+	} else if reply != utils.OK {
+		t.Errorf("Expected OK received: %s", reply)
+	}
+	exp := map[string]interface{}{
+		"Enabled":       true,
+		"ListenBijson":  "127.0.0.1:2014",
+		"ChargerSConns": []interface{}{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaChargers)},
+		"RALsConns":     []interface{}{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResponder)},
+		"ResSConns":     []interface{}{utils.MetaLocalHost},
+		"ThreshSConns":  []interface{}{},
+		"StatSConns":    []interface{}{},
+		"SupplSConns":   []interface{}{utils.MetaLocalHost},
+		"AttrSConns":    []interface{}{utils.MetaLocalHost},
+		"CDRsConns":     []interface{}{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaCDRs)},
+
+		"ReplicationConns":    []interface{}{},
+		"MaxCallDuration":     float64(3 * time.Hour),
+		"SessionIndexes":      map[string]interface{}{"OriginID": true},
+		"ClientProtocol":      1.,
+		"TerminateAttempts":   5.,
+		"ChannelSyncInterval": 0.,
+		"DebitInterval":       0.,
+		"MinCallDuration":     0.,
+		"SessionTTL":          0.,
+		"SessionTTLLastUsed":  nil,
+		"SessionTTLMaxDelay":  nil,
+		"SessionTTLUsage":     nil,
+		"StoreSCosts":         false,
+		"AlterableFields":     map[string]interface{}{},
+	}
+
+	var rpl map[string]interface{}
+	if err := configRPC.Call(utils.ConfigSv1GetJSONSection, &config.StringWithArgDispatcher{
+		Section: config.SessionSJson,
+	}, &rpl); err != nil {
+		t.Error(err)
+	} else if !reflect.DeepEqual(exp, rpl) {
+		t.Errorf("Expected %+v , received: %+v ", exp, rpl)
+	}
+}
+
+func testConfigSKillEngine(t *testing.T) {
+	if err := engine.KillEngine(100); err != nil {
+		t.Error(err)
+	}
+}