mirror of
https://github.com/grafana/grafana.git
synced 2025-07-31 22:02:17 +08:00
125 lines
4.2 KiB
Go
125 lines
4.2 KiB
Go
package mssql
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
_ "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() {
|
|
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, "
|
|
sql += "adatetimeoffset datetimeoffset) "
|
|
_, 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, adatetimeoffset ) "
|
|
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), SWITCHOFFSET(SYSDATETIMEOFFSET(), '-07:00') );"
|
|
_, 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
|
|
// fix 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(10*time.Second), time.Now().UTC())
|
|
So(column[13].(time.Time), ShouldHappenWithin, time.Duration(10*time.Second), time.Now().UTC().Truncate(24*time.Hour)) // ShouldEqual dose not work here !!?
|
|
So(column[14].(time.Time), ShouldHappenWithin, time.Duration(10*time.Second), time.Date(1, time.January, 1, time.Now().UTC().Hour(), time.Now().UTC().Minute(), time.Now().UTC().Second(), 0, time.UTC))
|
|
So(column[15].(time.Time), ShouldHappenWithin, time.Duration(10*time.Second), time.Now().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
|
|
}
|