energylink/pkg/stores/sqlstore/sqlstore_test.go

package sqlstore

import (
	"testing"

	"github.com/go-xorm/xorm"

	. "github.com/smartystreets/goconvey/convey"

	m "github.com/torkelo/grafana-pro/pkg/models"
)

func InitTestDB(t *testing.T) {
	x, err := xorm.NewEngine("sqlite3", ":memory:")

	if err != nil {
		t.Fatalf("Failed to init in memory sqllite3 db %v", err)
	}

	SetEngine(x, false)
}

type Test struct {
	Id   int64
	Name string
}

func TestDataAccess(t *testing.T) {

	Convey("Testing DB", t, func() {
		InitTestDB(t)

		Convey("Can add datasource", func() {

			err := AddDataSource(&m.AddDataSourceCommand{
				AccountId: 10,
				Type:      m.DS_GRAPHITE,
				Access:    m.DS_ACCESS_DIRECT,
				Url:       "http://test",
			})

			So(err, ShouldBeNil)

			query := m.GetDataSourcesQuery{AccountId: 10}
			err = GetDataSources(&query)
			So(err, ShouldBeNil)

			So(len(query.Resp), ShouldEqual, 1)

		})

	})

}