40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package dashvertest
|
|
|
|
import (
|
|
"context"
|
|
|
|
dashver "github.com/grafana/grafana/pkg/services/dashboardversion"
|
|
)
|
|
|
|
type FakeDashboardVersionService struct {
|
|
ExpectedDashboardVersion *dashver.DashboardVersionDTO
|
|
ExpectedDashboardVersions []*dashver.DashboardVersionDTO
|
|
ExpectedListDashboarVersions []*dashver.DashboardVersionDTO
|
|
ExpectedContinueToken string
|
|
counter int
|
|
ExpectedError error
|
|
}
|
|
|
|
func NewDashboardVersionServiceFake() *FakeDashboardVersionService {
|
|
return &FakeDashboardVersionService{}
|
|
}
|
|
|
|
func (f *FakeDashboardVersionService) Get(ctx context.Context, query *dashver.GetDashboardVersionQuery) (*dashver.DashboardVersionDTO, error) {
|
|
if len(f.ExpectedDashboardVersions) == 0 {
|
|
return f.ExpectedDashboardVersion, f.ExpectedError
|
|
}
|
|
f.counter++
|
|
return f.ExpectedDashboardVersions[f.counter-1], f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeDashboardVersionService) DeleteExpired(ctx context.Context, cmd *dashver.DeleteExpiredVersionsCommand) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeDashboardVersionService) List(ctx context.Context, query *dashver.ListDashboardVersionsQuery) (*dashver.DashboardVersionResponse, error) {
|
|
return &dashver.DashboardVersionResponse{
|
|
ContinueToken: f.ExpectedContinueToken,
|
|
Versions: f.ExpectedListDashboarVersions,
|
|
}, f.ExpectedError
|
|
}
|