mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
257 lines
10 KiB
Go
257 lines
10 KiB
Go
/*
|
|
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 ers
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
|
|
"github.com/cgrates/cgrates/utils"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
func TestSQLSetURL(t *testing.T) {
|
|
sql := new(SQLEventReader)
|
|
expsql := &SQLEventReader{
|
|
connString: "cgrates:CGRateS.org@tcp(127.0.0.1:3306)/cgrates2?charset=utf8&loc=Local&parseTime=true&sql_mode='ALLOW_INVALID_DATES'",
|
|
connType: "mysql",
|
|
tableName: "cdrs2",
|
|
expConnString: "cgrates:CGRateS.org@tcp(127.0.0.1:3306)/cgrates3?charset=utf8&loc=Local&parseTime=true&sql_mode='ALLOW_INVALID_DATES'",
|
|
expConnType: "mysql",
|
|
expTableName: "cdrs3",
|
|
}
|
|
inURL := "*mysql://cgrates:CGRateS.org@127.0.0.1:3306"
|
|
outURL := "*mysql://cgrates:CGRateS.org@127.0.0.1:3306"
|
|
if err := sql.setURL(inURL, outURL, &config.EventReaderOpts{
|
|
SQLDBName: utils.StringPointer("cgrates2"),
|
|
SQLTableName: utils.StringPointer("cdrs2"),
|
|
PgSSLMode: utils.StringPointer("enabled"),
|
|
|
|
SQLDBNameProcessed: utils.StringPointer("cgrates3"),
|
|
SQLTableNameProcessed: utils.StringPointer("cdrs3"),
|
|
PgSSLModeProcessed: utils.StringPointer("enabled"),
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
} else if expsql.connString != sql.connString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connString, sql.connString)
|
|
} else if expsql.connType != sql.connType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connType, sql.connType)
|
|
} else if expsql.tableName != sql.tableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.tableName, sql.tableName)
|
|
} else if expsql.expConnString != sql.expConnString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnString, sql.expConnString)
|
|
} else if expsql.expConnType != sql.expConnType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnType, sql.expConnType)
|
|
} else if expsql.expTableName != sql.expTableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expTableName, sql.expTableName)
|
|
}
|
|
|
|
sql = new(SQLEventReader)
|
|
expsql = &SQLEventReader{
|
|
connString: "host=127.0.0.1 port=3306 dbname=cgrates2 user=cgrates password=CGRateS.org sslmode=enabled",
|
|
connType: "postgres",
|
|
tableName: "cdrs2",
|
|
expConnString: "host=127.0.0.1 port=3306 dbname=cgrates3 user=cgrates password=CGRateS.org sslmode=enabled",
|
|
expConnType: "postgres",
|
|
expTableName: "cdrs3",
|
|
}
|
|
inURL = "*postgres://cgrates:CGRateS.org@127.0.0.1:3306"
|
|
outURL = "*postgres://cgrates:CGRateS.org@127.0.0.1:3306"
|
|
if err := sql.setURL(inURL, outURL, &config.EventReaderOpts{
|
|
SQLDBName: utils.StringPointer("cgrates2"),
|
|
SQLTableName: utils.StringPointer("cdrs2"),
|
|
PgSSLMode: utils.StringPointer("enabled"),
|
|
|
|
SQLDBNameProcessed: utils.StringPointer("cgrates3"),
|
|
SQLTableNameProcessed: utils.StringPointer("cdrs3"),
|
|
PgSSLModeProcessed: utils.StringPointer("enabled"),
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
} else if expsql.connString != sql.connString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connString, sql.connString)
|
|
} else if expsql.connType != sql.connType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connType, sql.connType)
|
|
} else if expsql.tableName != sql.tableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.tableName, sql.tableName)
|
|
} else if expsql.expConnString != sql.expConnString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnString, sql.expConnString)
|
|
} else if expsql.expConnType != sql.expConnType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnType, sql.expConnType)
|
|
} else if expsql.expTableName != sql.expTableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expTableName, sql.expTableName)
|
|
}
|
|
|
|
sql = new(SQLEventReader)
|
|
expsql = &SQLEventReader{
|
|
connString: "host=127.0.0.1 port=3306 dbname=cgrates2 user=cgrates password=CGRateS.org sslmode=enabled",
|
|
connType: "postgres",
|
|
tableName: "cdrs2",
|
|
expConnString: "host=127.0.0.1 port=3306 dbname=cgrates2 user=cgrates password=CGRateS.org sslmode=enabled",
|
|
expConnType: "postgres",
|
|
expTableName: "cdrs2",
|
|
}
|
|
inURL = "*postgres://cgrates:CGRateS.org@127.0.0.1:3306"
|
|
outURL = ""
|
|
if err := sql.setURL(inURL, outURL, &config.EventReaderOpts{
|
|
SQLDBName: utils.StringPointer("cgrates2"),
|
|
SQLTableName: utils.StringPointer("cdrs2"),
|
|
PgSSLMode: utils.StringPointer("enabled"),
|
|
|
|
SQLDBNameProcessed: utils.StringPointer("cgrates2"),
|
|
SQLTableNameProcessed: utils.StringPointer("cdrs2"),
|
|
PgSSLModeProcessed: utils.StringPointer("enabled"),
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
} else if expsql.connString != sql.connString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connString, sql.connString)
|
|
} else if expsql.connType != sql.connType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.connType, sql.connType)
|
|
} else if expsql.tableName != sql.tableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.tableName, sql.tableName)
|
|
} else if expsql.expConnString != sql.expConnString {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnString, sql.expConnString)
|
|
} else if expsql.expConnType != sql.expConnType {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expConnType, sql.expConnType)
|
|
} else if expsql.expTableName != sql.expTableName {
|
|
t.Errorf("Expected: %q ,received: %q", expsql.expTableName, sql.expTableName)
|
|
}
|
|
|
|
inURL = "*postgres://cgrates:CGRateS.org@127.0.0.1:3306?dbName=cgrates2&tableName=cdrs2&sslmode=enabled"
|
|
outURL = "*postgres2://cgrates:CGRateS.org@127.0.0.1:3306?dbName=cgrates2&tableName=cdrs2&sslmode=enabled"
|
|
if err := sql.setURL(inURL, outURL, &config.EventReaderOpts{}); err == nil || err.Error() != "unknown db_type postgres2" {
|
|
t.Errorf("Expected error: 'unknown db_type postgres2' ,received: %v", err)
|
|
}
|
|
inURL = "*postgres2://cgrates:CGRateS.org@127.0.0.1:3306?dbName=cgrates2&tableName=cdrs2&sslmode=enabled"
|
|
outURL = "*postgres://cgrates:CGRateS.org@127.0.0.1:3306?dbName=cgrates2&tableName=cdrs2&sslmode=enabled"
|
|
if err := sql.setURL(inURL, outURL, &config.EventReaderOpts{}); err == nil || err.Error() != "unknown db_type postgres2" {
|
|
t.Errorf("Expected error: 'unknown db_type postgres2' ,received: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSQLPostCDR(t *testing.T) {
|
|
cfg := config.NewDefaultCGRConfig()
|
|
rdr := &SQLEventReader{
|
|
cgrCfg: cfg,
|
|
cfgIdx: 1,
|
|
fltrS: &engine.FilterS{},
|
|
connString: "testString",
|
|
connType: "testType",
|
|
tableName: "testName",
|
|
expConnString: "testExpConnString",
|
|
expConnType: "testExpConnType",
|
|
expTableName: "testExpTableName",
|
|
rdrEvents: nil,
|
|
rdrExit: nil,
|
|
rdrErr: nil,
|
|
cap: nil,
|
|
}
|
|
in := make([]interface{}, 2)
|
|
err := rdr.postCDR(in)
|
|
expected := "db type <testExpConnType> not supported"
|
|
if err == nil || err.Error() != expected {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
|
|
}
|
|
}
|
|
|
|
func TestSQLReaderServePostgresErr(t *testing.T) {
|
|
tmp := logger.Default
|
|
logger.Default = logger.Default.LogMode(logger.Silent)
|
|
rdr := &SQLEventReader{
|
|
connType: utils.Postgres,
|
|
connString: "host=127.0.0.1 port=9999 dbname=cdrs user=cgrates password=CGRateS.org sslmode=disabled",
|
|
}
|
|
expected := "cannot parse `host=127.0.0.1 port=9999 dbname=cdrs user=cgrates password=xxxxx sslmode=disabled`: failed to configure TLS (sslmode is invalid)"
|
|
err := rdr.Serve()
|
|
if err == nil || err.Error() != expected {
|
|
t.Errorf("\nExpected: <%+v>, \nreceived: <%+v>", expected, err)
|
|
}
|
|
logger.Default = tmp
|
|
}
|
|
|
|
func TestSQLReaderServeBadType(t *testing.T) {
|
|
tmp := logger.Default
|
|
logger.Default = logger.Default.LogMode(logger.Silent)
|
|
rdr := &SQLEventReader{
|
|
connType: utils.Postgres,
|
|
connString: "host=127.0.0.1 port=9999 dbname=cdrs user=cgrates password=CGRateS.org sslmode=disabled",
|
|
}
|
|
expected := "cannot parse `host=127.0.0.1 port=9999 dbname=cdrs user=cgrates password=xxxxx sslmode=disabled`: failed to configure TLS (sslmode is invalid)"
|
|
err := rdr.Serve()
|
|
if err == nil || err.Error() != expected {
|
|
t.Errorf("\nExpected: <%+v>, \nreceived: <%+v>", expected, err)
|
|
}
|
|
logger.Default = tmp
|
|
}
|
|
|
|
func TestSQLPostCDRMySQLError(t *testing.T) {
|
|
tmp := logger.Default
|
|
logger.Default = logger.Default.LogMode(logger.Silent)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
testSQLEventReader := &SQLEventReader{
|
|
cgrCfg: cfg,
|
|
cfgIdx: 0,
|
|
fltrS: nil,
|
|
connString: "",
|
|
connType: "",
|
|
tableName: "testName",
|
|
expConnString: "",
|
|
expConnType: utils.MySQL,
|
|
expTableName: "",
|
|
rdrEvents: nil,
|
|
rdrExit: nil,
|
|
rdrErr: nil,
|
|
cap: nil,
|
|
}
|
|
err := testSQLEventReader.postCDR([]interface{}{})
|
|
expected := "Error 1045: Access denied for user ''@'localhost' (using password: NO)"
|
|
if err == nil {
|
|
t.Errorf("\nExpected: <%+v>, \nreceived: <%+v>", expected, err)
|
|
}
|
|
logger.Default = tmp
|
|
}
|
|
|
|
func TestSQLPostCDRPostgresError(t *testing.T) {
|
|
tmp := logger.Default
|
|
logger.Default = logger.Default.LogMode(logger.Silent)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
testSQLEventReader := &SQLEventReader{
|
|
cgrCfg: cfg,
|
|
cfgIdx: 0,
|
|
fltrS: nil,
|
|
connString: "",
|
|
connType: "",
|
|
tableName: "testName",
|
|
expConnString: "",
|
|
expConnType: utils.Postgres,
|
|
expTableName: "",
|
|
rdrEvents: nil,
|
|
rdrExit: nil,
|
|
rdrErr: nil,
|
|
cap: nil,
|
|
}
|
|
err := testSQLEventReader.postCDR([]interface{}{})
|
|
expected := "Error 1045: Access denied for user ''@'localhost' (using password: NO)"
|
|
if err == nil {
|
|
t.Errorf("\nExpected: <%+v>, \nreceived: <%+v>", expected, err)
|
|
}
|
|
logger.Default = tmp
|
|
}
|