mirror of
https://github.com/grafana/grafana.git
synced 2025-02-16 18:34:52 -06:00
67b2972052
* add/update sqlstore-related helper functions * add documentation & tests for InsertQuery and UpdateQuery, make generated SQL deterministic by sorting columns * remove old log line
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package dbtest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"xorm.io/core"
|
|
"xorm.io/xorm"
|
|
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore/migrator"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore/session"
|
|
tempuser "github.com/grafana/grafana/pkg/services/temp_user"
|
|
)
|
|
|
|
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
|
|
}
|
|
|
|
func (f *FakeDB) WithNewDbSession(ctx context.Context, callback sqlstore.DBTransactionFunc) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeDB) InTransaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeDB) GetDBType() core.DbType {
|
|
return ""
|
|
}
|
|
|
|
func (f *FakeDB) GetDialect() migrator.Dialect {
|
|
return nil
|
|
}
|
|
|
|
func (f *FakeDB) GetEngine() *xorm.Engine {
|
|
return nil
|
|
}
|
|
|
|
func (f *FakeDB) GetSqlxSession() *session.SessionDB {
|
|
return nil
|
|
}
|
|
|
|
func (f *FakeDB) Quote(value string) string {
|
|
return ""
|
|
}
|
|
|
|
func (f *FakeDB) RecursiveQueriesAreSupported() (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
// TODO: service-specific methods not yet split out ; to be removed
|
|
func (f *FakeDB) UpdateTempUserWithEmailSent(ctx context.Context, cmd *tempuser.UpdateTempUserWithEmailSentCommand) error {
|
|
return f.ExpectedError
|
|
}
|