uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-14 12:49:54 +05:00
Code Issues Packages Projects Releases Wiki Activity

Auto-configure loader in test suite if needed

Browse Source
This commit is contained in:
ionutboangiu
2024-11-05 16:31:46 +02:00
committed by Dan Christian Bogos
parent 5bf8f1756c
commit 4c8cf69f70
2 changed files with 47 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
engine/libtest.go
View File

@@ -362,6 +362,12 @@ func (ng TestEngine) Run(t testing.TB, extraFlags ...string) (*birpc.Client, *co
t.Helper()
cfg := parseCfg(t, ng.ConfigPath, ng.ConfigJSON, ng.DBCfg)
FlushDBs(t, cfg, !ng.PreserveDataDB, !ng.PreserveStorDB)
if ng.TpPath != "" || len(ng.TpFiles) != 0 {
if ng.TpPath == "" {
ng.TpPath = t.TempDir()
}
setupLoader(t, ng.TpPath, cfg.ConfigPath)
}
if ng.PreStartHook != nil {
ng.PreStartHook(t, cfg)
}
@@ -454,9 +460,33 @@ func parseCfg(t testing.TB, cfgPath, cfgJSON string, dbCfg DBCfg) (cfg *config.C
return
}
// loadCSVs loads tariff plan data from CSV files. The CSV files are created based on the csvFiles map, where
// the key represents the file name and the value the contains its contents. Assumes the data is loaded
// automatically (RunDelay != 0)
// setupLoader configures the *default loader to automatically load from the
// specified path.
func setupLoader(t testing.TB, tpPath, cfgPath string) {
t.Helper()
loadersJSON := fmt.Sprintf(`{
"loaders": [{
"id": "*default",
"enabled": true,
"run_delay": "-1",
"tp_in_dir": "%s",
"tp_out_dir": "",
"action": "*store",
"opts": {
"*stopOnError": true
}
}]
}`, tpPath)
filePath := filepath.Join(cfgPath, "zzz_dynamic_loader.json")
if err := os.WriteFile(filePath, []byte(loadersJSON), 0644); err != nil {
t.Fatal(err)
}
}
// loadCSVs loads tariff plan data from CSV files. The CSV files are created
// based on the csvFiles map, where the key represents the file name and the
// value the contains its contents. Assumes the data is loaded automatically
// (RunDelay != 0)
func loadCSVs(t testing.TB, tpPath string, csvFiles map[string]string) {
t.Helper()
if tpPath != "" {
@@ -469,7 +499,8 @@ func loadCSVs(t testing.TB, tpPath string, csvFiles map[string]string) {
}
}
// FlushDBs resets the databases specified in the configuration if the corresponding flags are true.
// FlushDBs resets the databases specified in the configuration if the
// corresponding flags are true.
func FlushDBs(t testing.TB, cfg *config.CGRConfig, flushDataDB, flushStorDB bool) {
t.Helper()
if flushDataDB {