The open and composable observability and data visualization platform. Visualize metrics, logs, and traces from multiple sources like Prometheus, Loki, Elasticsearch, InfluxDB, Postgres and many more.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
grafana/pkg/tsdb/mssql/mssql_test.go

123 lines
4.0 KiB

package mssql
import (
"testing"
"time"
"strings"
_ "github.com/denisenkom/go-mssqldb"
"github.com/go-xorm/xorm"
"github.com/grafana/grafana/pkg/components/simplejson"
"github.com/grafana/grafana/pkg/log"
"github.com/grafana/grafana/pkg/services/sqlstore/sqlutil"
"github.com/grafana/grafana/pkg/tsdb"
. "github.com/smartystreets/goconvey/convey"
)
// To run this test, remove the Skip from SkipConvey
// and set up a MSSQL db named grafana_tests and a user/password grafana/password
// and set the variable below to the IP address of the database
var serverIP string = "10.20.30.40"
func TestMSSQL(t *testing.T) {
//SkipConvey("MSSQL", t, func() {
SkipConvey("MSSQL", t, func() {
x := InitMSSQLTestDB(t)
endpoint := &MssqlQueryEndpoint{
sqlEngine: &tsdb.DefaultSqlEngine{
MacroEngine: NewMssqlMacroEngine(),
XormEngine: x,
},
log: log.New("tsdb.mssql"),
}
sess := x.NewSession()
defer sess.Close()
sql := "IF OBJECT_ID('dbo.[mssql_types]', 'U') IS NOT NULL"
sql += " DROP TABLE dbo.[mssql_types];"
sql += "CREATE TABLE [mssql_types] ( "
sql += "abit bit, "
sql += "atinyint tinyint, "
sql += "asmallint smallint, "
sql += "aint int, "
sql += "abigint bigint, "
sql += "avarchar varchar(3), "
sql += "achar char(3), "
sql += "anewvarchar varchar(14), "
sql += "anewchar char(14), "
sql += "areal real, "
sql += "anewdecimal decimal(10,2), "
sql += "afloat float, "
sql += "adatetime datetime, "
sql += "adate date, "
sql += "atime time) "
_, err := sess.Exec(sql)
So(err, ShouldBeNil)
sql = "INSERT INTO [mssql_types] "
sql += "(abit, atinyint, asmallint, aint, abigint, "
sql += "avarchar, achar, anewvarchar, anewchar, "
sql += "areal, anewdecimal, afloat, "
sql += "adatetime, adate, atime ) "
sql += "VALUES(1, 5, 20020, 980300, 1420070400, "
sql += "'abc', 'def', 'hi varchar', 'I am only char', "
sql += "1.11, 2.22, 3.33, "
sql += "GETUTCDATE(), CAST(GETUTCDATE() AS DATE), CAST(GETUTCDATE() AS TIME) );"
_, err = sess.Exec(sql)
So(err, ShouldBeNil)
Convey("Query with Table format should map MSSQL column types to Go types", func() {
query := &tsdb.TsdbQuery{
Queries: []*tsdb.Query{
{
Model: simplejson.NewFromAny(map[string]interface{}{
"rawSql": "SELECT * FROM mssql_types",
"format": "table",
}),
RefId: "A",
},
},
}
resp, err := endpoint.Query(nil, nil, query)
queryResult := resp.Results["A"]
So(err, ShouldBeNil)
column := queryResult.Tables[0].Rows[0]
So(column[0].(bool), ShouldEqual, true)
So(column[1].(int64), ShouldEqual, 5)
So(column[2].(int64), ShouldEqual, 20020)
So(column[3].(int64), ShouldEqual, 980300)
So(column[4].(int64), ShouldEqual, 1420070400)
So(column[5].(string), ShouldEqual, "abc")
So(column[6].(string), ShouldEqual, "def")
So(column[7].(string), ShouldEqual, "hi varchar")
So(column[8].(string), ShouldEqual, "I am only char")
So(column[9].(float64), ShouldEqual, 1.1100000143051147) // MSSQL dose not have precision for "real" datatype
// fiix me: MSSQL driver puts the decimal inside an array of chars. and the test fails despite the values are correct.
//So(column[10].([]uint8), ShouldEqual, []uint8{'2', '.', '2', '2'})
So(column[11].(float64), ShouldEqual, 3.33)
So(column[12].(time.Time), ShouldHappenWithin, time.Duration(15*time.Second), time.Now().UTC() )
So(column[13].(time.Time), ShouldHappenWithin, time.Duration(15*time.Second), time.Now().UTC().Truncate(24*time.Hour) )
So(column[14].(time.Time), ShouldHappenWithin, time.Duration(15*time.Second), time.Date( 1, time.January, 1, time.Now().UTC().Hour(), time.Now().UTC().Minute(), time.Now().UTC().Second(), 0, time.UTC) )
})
})
}
func InitMSSQLTestDB(t *testing.T) *xorm.Engine {
x, err := xorm.NewEngine(sqlutil.TestDB_Mssql.DriverName, strings.Replace(sqlutil.TestDB_Mssql.ConnStr, "localhost", serverIP, 1) )
// x.ShowSQL()
if err != nil {
t.Fatalf("Failed to init mssql db %v", err)
}
sqlutil.CleanDB(x)
return x
}