mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
23 lines
550 B
Go
23 lines
550 B
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/services/team"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type TeamGuardianStoreMock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (t *TeamGuardianStoreMock) GetTeamMembers(ctx context.Context, query team.GetTeamMembersQuery) ([]*team.TeamMemberDTO, error) {
|
|
args := t.Called(ctx, query)
|
|
return args.Get(0).([]*team.TeamMemberDTO), args.Error(1)
|
|
}
|
|
|
|
func (t *TeamGuardianStoreMock) DeleteByUser(ctx context.Context, userID int64) error {
|
|
args := t.Called(ctx, userID)
|
|
return args.Get(0).(error)
|
|
}
|