energylink/pkg/stores/rethinkdb_test.go

package stores

import (
	"testing"

	"github.com/dancannon/gorethink"
	. "github.com/smartystreets/goconvey/convey"
	"github.com/torkelo/grafana-pro/pkg/models"
)

func TestRethinkStore(t *testing.T) {
	store := NewRethinkStore(&RethinkCfg{DatabaseName: "tests"})
	defer gorethink.DbDrop("tests").Exec(store.session)

	Convey("Insert dashboard", t, func() {
		dashboard := models.NewDashboard("test")
		dashboard.AccountId = 1

		err := store.SaveDashboard(dashboard)
		So(err, ShouldBeNil)
		So(dashboard.Id, ShouldNotBeEmpty)

		read, err := store.GetDashboard("test", 1)
		So(err, ShouldBeNil)
		So(read, ShouldNotBeNil)
	})

	Convey("can get next account id", t, func() {
		id, err := store.getNextAccountId()
		So(err, ShouldBeNil)
		So(id, ShouldNotEqual, 0)

		id2, err := store.getNextAccountId()
		So(id2, ShouldEqual, id+1)
	})

	Convey("can create account", t, func() {
		account, err := store.createAccount()
		So(err, ShouldBeNil)
		So(account, ShouldNotBeNil)
		So(account.Id, ShouldNotEqual, 0)
	})

	Convey("can get next dashboard id", t, func() {
		account, err := store.createAccount()
		dashId, err := store.getNextDashboardNumber(account.Id)
		So(err, ShouldBeNil)
		So(dashId, ShouldEqual, 1)
	})

}