energylink/pkg/tsdb/mssql/mssql_test.go

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
}