mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
9855e74b92
Chore: Refactor quota service (#57586) * Chore: refactore quota service * Apply suggestions from code review
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package quotatest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
)
|
|
|
|
type FakeQuotaService struct {
|
|
reached bool
|
|
err error
|
|
}
|
|
|
|
func New(reached bool, err error) *FakeQuotaService {
|
|
return &FakeQuotaService{reached, err}
|
|
}
|
|
|
|
func (f *FakeQuotaService) GetQuotasByScope(ctx context.Context, scope quota.Scope, id int64) ([]quota.QuotaDTO, error) {
|
|
return []quota.QuotaDTO{}, nil
|
|
}
|
|
|
|
func (f *FakeQuotaService) Update(ctx context.Context, cmd *quota.UpdateQuotaCmd) error {
|
|
return nil
|
|
}
|
|
|
|
func (f *FakeQuotaService) QuotaReached(c *models.ReqContext, target quota.TargetSrv) (bool, error) {
|
|
return f.reached, f.err
|
|
}
|
|
|
|
func (f *FakeQuotaService) CheckQuotaReached(c context.Context, target quota.TargetSrv, params *quota.ScopeParameters) (bool, error) {
|
|
return f.reached, f.err
|
|
}
|
|
|
|
func (f *FakeQuotaService) DeleteQuotaForUser(c context.Context, userID int64) error {
|
|
return f.err
|
|
}
|
|
|
|
func (f *FakeQuotaService) RegisterQuotaReporter(e *quota.NewUsageReporter) error {
|
|
return f.err
|
|
}
|
|
|
|
type FakeQuotaStore struct {
|
|
ExpectedError error
|
|
}
|
|
|
|
func (f *FakeQuotaStore) DeleteByUser(ctx quota.Context, userID int64) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeQuotaStore) Get(ctx quota.Context, scopeParams *quota.ScopeParameters) (*quota.Map, error) {
|
|
return nil, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeQuotaStore) Update(ctx quota.Context, cmd *quota.UpdateQuotaCmd) error {
|
|
return f.ExpectedError
|
|
}
|