grafana/pkg/services/sqlstore/transactions_test.go

package sqlstore

import (
	"context"
	"errors"
	"testing"

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

	"github.com/grafana/grafana/pkg/models"
)

var ErrProvokedError = errors.New("testing error")

func TestTransaction(t *testing.T) {
	ss := InitTestDB(t)

	Convey("InTransaction", t, func() {
		cmd := &models.AddApiKeyCommand{Key: "secret-key", Name: "key", OrgId: 1}

		err := AddApiKey(cmd)
		So(err, ShouldBeNil)

		deleteApiKeyCmd := &models.DeleteApiKeyCommand{Id: cmd.Result.Id, OrgId: 1}

		Convey("can update key", func() {
			err := ss.InTransaction(context.Background(), func(ctx context.Context) error {
				return DeleteApiKeyCtx(ctx, deleteApiKeyCmd)
			})

			So(err, ShouldBeNil)

			query := &models.GetApiKeyByIdQuery{ApiKeyId: cmd.Result.Id}
			err = GetApiKeyById(query)
			So(err, ShouldEqual, models.ErrInvalidApiKey)
		})

		Convey("won't update if one handler fails", func() {
			err := ss.InTransaction(context.Background(), func(ctx context.Context) error {
				err := DeleteApiKeyCtx(ctx, deleteApiKeyCmd)
				if err != nil {
					return err
				}

				return ErrProvokedError
			})

			So(err, ShouldEqual, ErrProvokedError)

			query := &models.GetApiKeyByIdQuery{ApiKeyId: cmd.Result.Id}
			err = GetApiKeyById(query)
			So(err, ShouldBeNil)
			So(query.Result.Id, ShouldEqual, cmd.Result.Id)
		})
	})
}