From 77976ea81089549241893213388cd684d90ba99e Mon Sep 17 00:00:00 2001 From: idafurjes <36131195+idafurjes@users.noreply.github.com> Date: Wed, 23 Mar 2022 15:06:47 +0100 Subject: [PATCH] Add fakes for db interface (#46885) --- pkg/services/sqlstore/db/dbtest/dbtest.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 pkg/services/sqlstore/db/dbtest/dbtest.go diff --git a/pkg/services/sqlstore/db/dbtest/dbtest.go b/pkg/services/sqlstore/db/dbtest/dbtest.go new file mode 100644 index 00000000000..c8013cb4f49 --- /dev/null +++ b/pkg/services/sqlstore/db/dbtest/dbtest.go @@ -0,0 +1,23 @@ +package dbtest + +import ( + "context" + + "github.com/grafana/grafana/pkg/services/sqlstore" +) + +type FakeDB struct { + ExpectedError error +} + +func NewFakeDB() *FakeDB { + return &FakeDB{} +} + +func (f *FakeDB) WithTransactionalDbSession(ctx context.Context, callback sqlstore.DBTransactionFunc) error { + return f.ExpectedError +} + +func (f *FakeDB) WithDbSession(ctx context.Context, callback sqlstore.DBTransactionFunc) error { + return f.ExpectedError +}