mirror of https://github.com/grafana/grafana
Alexander Akhmetov
5 months ago
parent
7acfdfa0af
commit
c9cdfc51a6
@ -0,0 +1,135 @@ |
|||||||
|
package fakes |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"fmt" |
||||||
|
"math/rand" |
||||||
|
"sync" |
||||||
|
"testing" |
||||||
|
|
||||||
|
"github.com/grafana/grafana/pkg/apimachinery/identity" |
||||||
|
"github.com/grafana/grafana/pkg/services/dashboards" |
||||||
|
"github.com/grafana/grafana/pkg/services/folder" |
||||||
|
"github.com/grafana/grafana/pkg/util" |
||||||
|
) |
||||||
|
|
||||||
|
type FakeAlertingFolderService struct { |
||||||
|
t *testing.T |
||||||
|
mtx sync.Mutex |
||||||
|
Hook func(cmd any) error // use Hook if you need to intercept some query and return an error
|
||||||
|
RecordedOps []any |
||||||
|
Folders map[int64][]*folder.Folder |
||||||
|
} |
||||||
|
|
||||||
|
func NewFakeAlertingFolderService(t *testing.T) *FakeAlertingFolderService { |
||||||
|
return &FakeAlertingFolderService{ |
||||||
|
t: t, |
||||||
|
Hook: func(any) error { |
||||||
|
return nil |
||||||
|
}, |
||||||
|
Folders: map[int64][]*folder.Folder{}, |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// GetRecordedCommands filters recorded commands using predicate function. Returns the subset of the recorded commands that meet the predicate
|
||||||
|
func (f *FakeAlertingFolderService) GetRecordedCommands(predicate func(cmd any) (any, bool)) []any { |
||||||
|
f.mtx.Lock() |
||||||
|
defer f.mtx.Unlock() |
||||||
|
|
||||||
|
result := make([]any, 0, len(f.RecordedOps)) |
||||||
|
for _, op := range f.RecordedOps { |
||||||
|
cmd, ok := predicate(op) |
||||||
|
if !ok { |
||||||
|
continue |
||||||
|
} |
||||||
|
result = append(result, cmd) |
||||||
|
} |
||||||
|
return result |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FakeAlertingFolderService) GetUserVisibleNamespaces(_ context.Context, orgID int64, _ identity.Requester) (map[string]*folder.Folder, error) { |
||||||
|
f.mtx.Lock() |
||||||
|
defer f.mtx.Unlock() |
||||||
|
|
||||||
|
namespacesMap := map[string]*folder.Folder{} |
||||||
|
|
||||||
|
for _, folder := range f.Folders[orgID] { |
||||||
|
namespacesMap[folder.UID] = folder |
||||||
|
} |
||||||
|
return namespacesMap, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FakeAlertingFolderService) GetNamespaceByUID(_ context.Context, uid string, orgID int64, user identity.Requester) (*folder.Folder, error) { |
||||||
|
q := GenericRecordedQuery{ |
||||||
|
Name: "GetNamespaceByUID", |
||||||
|
Params: []any{orgID, uid, user}, |
||||||
|
} |
||||||
|
defer func() { |
||||||
|
f.RecordedOps = append(f.RecordedOps, q) |
||||||
|
}() |
||||||
|
err := f.Hook(q) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
folders := f.Folders[orgID] |
||||||
|
for _, folder := range folders { |
||||||
|
if folder.UID == uid { |
||||||
|
return folder, nil |
||||||
|
} |
||||||
|
} |
||||||
|
return nil, fmt.Errorf("not found") |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FakeAlertingFolderService) GetOrCreateNamespaceByTitle(ctx context.Context, title string, orgID int64, user identity.Requester, parentUID string) (*folder.Folder, error) { |
||||||
|
f.mtx.Lock() |
||||||
|
defer f.mtx.Unlock() |
||||||
|
|
||||||
|
for _, folder := range f.Folders[orgID] { |
||||||
|
if folder.Title == title && folder.ParentUID == parentUID { |
||||||
|
return folder, nil |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
newFolder := &folder.Folder{ |
||||||
|
ID: rand.Int63(), // nolint:staticcheck
|
||||||
|
UID: util.GenerateShortUID(), |
||||||
|
Title: title, |
||||||
|
ParentUID: parentUID, |
||||||
|
Fullpath: "fullpath_" + title, |
||||||
|
} |
||||||
|
|
||||||
|
f.Folders[orgID] = append(f.Folders[orgID], newFolder) |
||||||
|
return newFolder, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FakeAlertingFolderService) GetNamespaceByTitle(ctx context.Context, title string, orgID int64, user identity.Requester, parentUID string) (*folder.Folder, error) { |
||||||
|
f.mtx.Lock() |
||||||
|
defer f.mtx.Unlock() |
||||||
|
|
||||||
|
for _, folder := range f.Folders[orgID] { |
||||||
|
if folder.Title == title && folder.ParentUID == parentUID { |
||||||
|
return folder, nil |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return nil, dashboards.ErrFolderNotFound |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FakeAlertingFolderService) GetNamespaceChildren(ctx context.Context, uid string, orgID int64, user identity.Requester) ([]*folder.Folder, error) { |
||||||
|
f.mtx.Lock() |
||||||
|
defer f.mtx.Unlock() |
||||||
|
|
||||||
|
result := []*folder.Folder{} |
||||||
|
|
||||||
|
for _, folder := range f.Folders[orgID] { |
||||||
|
if folder.ParentUID == uid { |
||||||
|
result = append(result, folder) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if len(result) == 0 { |
||||||
|
return nil, dashboards.ErrFolderNotFound |
||||||
|
} |
||||||
|
|
||||||
|
return result, nil |
||||||
|
} |
||||||
Loading…
Reference in new issue