uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-11 18:16:24 +05:00
Code Issues Packages Projects Releases Wiki Activity

Add integration test for *sql exporter

Browse Source
This commit is contained in:
TeoV
2020-12-18 18:55:50 +02:00
committed by Dan Christian Bogos
parent 07159d8d6a
commit d955fbc959
5 changed files with 143 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
ees/sql_it_test.go
View File

@@ -48,11 +48,12 @@ var (
testSqlEeCreateTable,
testSqlEeLoadConfig,
testSqlEeResetDataDB,
testSqlEeResetStorDb,
testSqlEeStartEngine,
testSqlEeRPCConn,
testSqlEeExportEvent,
testSqlEeExportEventFull,
testSqlEeVerifyExportedEvent,
testSqlEeExportEventPartial,
testSqlEeVerifyExportedEvent2,
testStopCgrEngine,
testCleanDirectory,
}
@@ -67,14 +68,10 @@ func TestSqlEeExport(t *testing.T) {
// create a struct serve as model for *sql exporter
type testModelSql struct {
ID int64
Cgrid string
AnswerTime time.Time
Usage int64
Cost float64
CreatedAt time.Time
UpdatedAt time.Time
DeletedAt *time.Time
}
func (_ *testModelSql) TableName() string {
@@ -88,22 +85,29 @@ func (nopLogger) Print(values ...interface{}) {}
func testSqlEeCreateTable(t *testing.T) {
var err error
if db2, err = gorm.Open("mysql", fmt.Sprintf(dbConnString, "exportedDatabase")); err != nil {
if db2, err = gorm.Open("mysql", fmt.Sprintf(dbConnString, "cgrates")); err != nil {
t.Fatal(err)
}
db2.SetLogger(new(nopLogger))
if _, err = db2.DB().Exec(`CREATE DATABASE IF NOT EXISTS exportedDatabase;`); err != nil {
t.Fatal(err)
}
if db2, err = gorm.Open("mysql", fmt.Sprintf(dbConnString, "exportedDatabase")); err != nil {
t.Fatal(err)
}
tx := db2.Begin()
if !tx.HasTable("expTable") {
tx = tx.CreateTable(new(testModelSql))
if tx.HasTable("expTable") {
tx = tx.DropTable(new(testModelSql))
if err = tx.Error; err != nil {
tx.Rollback()
t.Fatal(err)
}
}
tx = tx.CreateTable(new(testModelSql))
if err = tx.Error; err != nil {
tx.Rollback()
t.Fatal(err)
}
tx.Commit()
}
@@ -121,12 +125,6 @@ func testSqlEeResetDataDB(t *testing.T) {
}
}
func testSqlEeResetStorDb(t *testing.T) {
if err := engine.InitStorDb(sqlEeCfg); err != nil {
t.Fatal(err)
}
}
func testSqlEeStartEngine(t *testing.T) {
if _, err := engine.StopStartEngine(sqlEeCfgPath, *waitRater); err != nil {
t.Fatal(err)
@@ -141,9 +139,9 @@ func testSqlEeRPCConn(t *testing.T) {
}
}
func testSqlEeExportEvent(t *testing.T) {
func testSqlEeExportEventFull(t *testing.T) {
eventVoice := &utils.CGREventWithEeIDs{
EeIDs: []string{"SQLExporter"},
EeIDs: []string{"SQLExporterFull"},
CGREventWithOpts: &utils.CGREventWithOpts{
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
@@ -179,6 +177,44 @@ func testSqlEeExportEvent(t *testing.T) {
time.Sleep(10 * time.Millisecond)
}
func testSqlEeExportEventPartial(t *testing.T) {
eventVoice := &utils.CGREventWithEeIDs{
EeIDs: []string{"SQLExporterPartial"},
CGREventWithOpts: &utils.CGREventWithOpts{
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "voiceEvent",
Time: utils.TimePointer(time.Now()),
Event: map[string]interface{}{
utils.CGRID: utils.Sha1("asd", time.Unix(1383813745, 0).UTC().String()),
utils.ToR: utils.VOICE,
utils.OriginID: "dsafdsaf",
utils.OriginHost: "192.168.1.1",
utils.RequestType: utils.META_RATED,
utils.Tenant: "cgrates.org",
utils.Category: "call",
utils.Account: "1001",
utils.Subject: "1001",
utils.Destination: "1002",
utils.SetupTime: time.Unix(1383813745, 0).UTC(),
utils.AnswerTime: time.Unix(1383813746, 0).UTC(),
utils.Usage: 10 * time.Second,
utils.RunID: utils.MetaDefault,
utils.Cost: 123,
"ExtraFields": map[string]string{"extra1": "val_extra1",
"extra2": "val_extra2", "extra3": "val_extra3"},
},
},
},
}
var reply map[string]utils.MapStorage
if err := sqlEeRpc.Call(utils.EeSv1ProcessEvent, eventVoice, &reply); err != nil {
t.Error(err)
}
time.Sleep(10 * time.Millisecond)
}
func testSqlEeVerifyExportedEvent(t *testing.T) {
var result int64
db2.Table("expTable").Count(&result)
@@ -186,3 +222,11 @@ func testSqlEeVerifyExportedEvent(t *testing.T) {
t.Fatal("Expected table to have only one result ", result)
}
}
func testSqlEeVerifyExportedEvent2(t *testing.T) {
var result int64
db2.Table("expTable").Count(&result)
if result != 2 {
t.Fatal("Expected table to have only one result ", result)
}
}