mirror of
https://github.com/grafana/grafana.git
synced 2025-02-20 11:48:34 -06:00
6afad51761
* Move SignedInUser to user service and RoleType and Roles to org * Use go naming convention for roles * Fix some imports and leftovers * Fix ldap debug test * Fix lint * Fix lint 2 * Fix lint 3 * Fix type and not needed conversion * Clean up messages in api tests * Clean up api tests 2
51 lines
2.2 KiB
Go
51 lines
2.2 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/grafana/grafana/pkg/services/accesscontrol"
|
|
"github.com/grafana/grafana/pkg/services/user"
|
|
)
|
|
|
|
var _ accesscontrol.PermissionsService = new(MockPermissionsService)
|
|
|
|
func NewMockedPermissionsService() *MockPermissionsService {
|
|
return &MockPermissionsService{}
|
|
}
|
|
|
|
type MockPermissionsService struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockPermissionsService) GetPermissions(ctx context.Context, user *user.SignedInUser, resourceID string) ([]accesscontrol.ResourcePermission, error) {
|
|
mockedArgs := m.Called(ctx, user, resourceID)
|
|
return mockedArgs.Get(0).([]accesscontrol.ResourcePermission), mockedArgs.Error(1)
|
|
}
|
|
|
|
func (m *MockPermissionsService) SetUserPermission(ctx context.Context, orgID int64, user accesscontrol.User, resourceID, permission string) (*accesscontrol.ResourcePermission, error) {
|
|
mockedArgs := m.Called(ctx, orgID, user, resourceID, permission)
|
|
return mockedArgs.Get(0).(*accesscontrol.ResourcePermission), mockedArgs.Error(1)
|
|
}
|
|
|
|
func (m *MockPermissionsService) SetTeamPermission(ctx context.Context, orgID, teamID int64, resourceID, permission string) (*accesscontrol.ResourcePermission, error) {
|
|
mockedArgs := m.Called(ctx, orgID, teamID, resourceID, permission)
|
|
return mockedArgs.Get(0).(*accesscontrol.ResourcePermission), mockedArgs.Error(1)
|
|
}
|
|
|
|
func (m *MockPermissionsService) SetBuiltInRolePermission(ctx context.Context, orgID int64, builtInRole, resourceID, permission string) (*accesscontrol.ResourcePermission, error) {
|
|
mockedArgs := m.Called(ctx, orgID, builtInRole, resourceID, permission)
|
|
return mockedArgs.Get(0).(*accesscontrol.ResourcePermission), mockedArgs.Error(1)
|
|
}
|
|
|
|
func (m *MockPermissionsService) SetPermissions(ctx context.Context, orgID int64, resourceID string, commands ...accesscontrol.SetResourcePermissionCommand) ([]accesscontrol.ResourcePermission, error) {
|
|
mockedArgs := m.Called(ctx, orgID, resourceID, commands)
|
|
return mockedArgs.Get(0).([]accesscontrol.ResourcePermission), mockedArgs.Error(1)
|
|
}
|
|
|
|
func (m *MockPermissionsService) MapActions(permission accesscontrol.ResourcePermission) string {
|
|
mockedArgs := m.Called(permission)
|
|
return mockedArgs.Get(0).(string)
|
|
}
|