add testing of org and user processing
This commit is contained in:
52
internal/metal/providers/mock.go
Normal file
52
internal/metal/providers/mock.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package providers
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/stretchr/testify/mock"
|
||||
"go.infratographer.com/x/gidx"
|
||||
|
||||
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/models"
|
||||
)
|
||||
|
||||
var _ Provider = &MockProvider{}
|
||||
|
||||
// MockProvider implements Provider used for testing.
|
||||
type MockProvider struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
// GetOrganizationDetails implements Provider used for testing.
|
||||
func (p *MockProvider) GetOrganizationDetails(ctx context.Context, id gidx.PrefixedID) (*models.OrganizationDetails, error) {
|
||||
args := p.Called(id)
|
||||
|
||||
return args.Get(0).(*models.OrganizationDetails), args.Error(1)
|
||||
}
|
||||
|
||||
// GetProjectDetails implements Provider used for testing.
|
||||
func (p *MockProvider) GetProjectDetails(ctx context.Context, id gidx.PrefixedID) (*models.ProjectDetails, error) {
|
||||
args := p.Called(id)
|
||||
|
||||
return args.Get(0).(*models.ProjectDetails), args.Error(1)
|
||||
}
|
||||
|
||||
// GetUserDetails implements Provider used for testing.
|
||||
func (p *MockProvider) GetUserDetails(ctx context.Context, id gidx.PrefixedID) (*models.UserDetails, error) {
|
||||
args := p.Called(id)
|
||||
|
||||
return args.Get(0).(*models.UserDetails), args.Error(1)
|
||||
}
|
||||
|
||||
// GetUserOrganizationRole implements Provider used for testing.
|
||||
func (p *MockProvider) GetUserOrganizationRole(ctx context.Context, userID, orgID gidx.PrefixedID) (string, error) {
|
||||
args := p.Called(userID, orgID)
|
||||
|
||||
return args.String(0), args.Error(1)
|
||||
}
|
||||
|
||||
// GetUserProjectRole implements Provider used for testing.
|
||||
func (p *MockProvider) GetUserProjectRole(ctx context.Context, userID, projID gidx.PrefixedID) (string, error) {
|
||||
args := p.Called(userID, projID)
|
||||
|
||||
return args.String(0), args.Error(1)
|
||||
}
|
||||
Reference in New Issue
Block a user