mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
added it tests for ErSv1.RunReader API
This commit is contained in:
committed by
Dan Christian Bogos
parent
1a9313be68
commit
b0aa272f75
@@ -22,11 +22,13 @@ import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/cgrates/birpc/context"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/ers"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
@@ -169,3 +171,81 @@ cgrates.org,call,1001,2014-01-14T00:00:00Z,RP_ANY,`,
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestErsRunReader(t *testing.T) {
|
||||
switch *utils.DBType {
|
||||
case utils.MetaInternal:
|
||||
case utils.MetaMySQL, utils.MetaMongo, utils.MetaPostgres:
|
||||
t.SkipNow()
|
||||
default:
|
||||
t.Fatal("unsupported dbtype value")
|
||||
}
|
||||
csvcontent := `csvfile1,1001,1303535,1727779754,1727779754,60`
|
||||
csvFd, procFd := t.TempDir(), t.TempDir()
|
||||
filePath := filepath.Join(csvFd, fmt.Sprintf("file1%s", utils.CSVSuffix))
|
||||
if err := os.WriteFile(filePath, []byte(csvcontent), 0644); err != nil {
|
||||
t.Fatalf("could not write to file %s: %v", filePath, err)
|
||||
}
|
||||
content := fmt.Sprintf(`{
|
||||
"general": {
|
||||
"log_level": 7
|
||||
},
|
||||
"data_db": {
|
||||
"db_type": "*internal"
|
||||
},
|
||||
"stor_db": {
|
||||
"db_type": "*internal"
|
||||
},
|
||||
"apiers":{
|
||||
"enabled":true,
|
||||
},
|
||||
"ers": {
|
||||
"enabled": true,
|
||||
"sessions_conns": ["*localhost"],
|
||||
"readers": [
|
||||
{
|
||||
"id": "file_csv_reader",
|
||||
"run_delay": "0",
|
||||
"start_delay":"1s",
|
||||
"type": "*file_csv",
|
||||
"source_path": "%s",
|
||||
"flags": ["*log","*none"],
|
||||
"processed_path": "%s",
|
||||
"fields":[
|
||||
{"tag": "ToR", "path": "*cgreq.ToR", "type": "*constant", "value": "*voice"},
|
||||
{"tag": "OriginID", "path": "*cgreq.OriginID", "type": "*variable", "value": "~*req.0", "mandatory": true},
|
||||
{"tag": "RequestType", "path": "*cgreq.RequestType", "type": "*constant", "value": "*rated", "mandatory": true},
|
||||
{"tag":"Category","path":"*cgreq.Category","type":"*constant","value":"call"},
|
||||
{"tag":"Subject","path":"*cgreq.Subject","type":"*variable","value":"~*req.1"},
|
||||
{"tag":"Destination","path":"*cgreq.Destination","type":"*variable","value":"~*req.2"},
|
||||
{"tag": "SetupTime", "path": "*cgreq.SetupTime", "type": "*variable", "value": "~*req.3"},
|
||||
{"tag": "AnswerTime", "path": "*cgreq.AnswerTime", "type": "*variable", "value": "~*req.4"},
|
||||
{"tag": "Usage", "path": "*cgreq.Usage", "filters": ["*notempty:~*req.5:"],"type": "*variable", "value": "~*req.5;s", "mandatory": true},
|
||||
]
|
||||
},
|
||||
]
|
||||
}
|
||||
}`, csvFd, procFd)
|
||||
var buf bytes.Buffer
|
||||
ng := engine.TestEngine{
|
||||
ConfigJSON: content,
|
||||
LogBuffer: &buf,
|
||||
}
|
||||
client, _ := ng.Run(t)
|
||||
t.Run("TestRunReader", func(t *testing.T) {
|
||||
runReader := ers.V1RunReaderParams{
|
||||
Tenant: "cgrates.org",
|
||||
ReaderID: "file_csv_reader",
|
||||
}
|
||||
var reply string
|
||||
if err := client.Call(context.Background(), utils.ErSv1RunReader, runReader, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.OK {
|
||||
t.Error("expected reply to be OK")
|
||||
}
|
||||
})
|
||||
time.Sleep(400 * time.Millisecond)
|
||||
if has := strings.Contains(buf.String(), "<ERs> LOG, reader: <file_csv_reader>"); !has {
|
||||
t.Error("expected reader to process files")
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user