testutils

package
v0.2.5 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 27, 2026 License: MIT Imports: 9 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type MockDatabase

type MockDatabase struct {
	Objects                map[string]*drs.DrsObject
	ObjectAuth             map[string]models.AuthPathMap
	ObjectAuthz            map[string]map[string][]string
	Credentials            map[string]models.S3Credential
	BucketScopes           map[string]models.BucketScope
	PendingMeta            map[string]models.PendingLFSMeta
	Usage                  map[string]models.FileUsage
	TransferEvents         []models.TransferAttributionEvent
	ProviderTransferEvents []models.ProviderTransferEvent
	ProviderSyncRuns       []models.ProviderTransferSyncRun
	NoDefaultCreds         bool
	GetObjectErr           error
}

MockDatabase implements db.DatabaseInterface for testing

func (*MockDatabase) BulkDeleteObjects

func (m *MockDatabase) BulkDeleteObjects(ctx context.Context, ids []string) error

func (*MockDatabase) BulkUpdateAccessMethods

func (m *MockDatabase) BulkUpdateAccessMethods(ctx context.Context, updates map[string][]drs.AccessMethod) error

func (*MockDatabase) CreateBucketScope

func (m *MockDatabase) CreateBucketScope(ctx context.Context, scope *models.BucketScope) error

func (*MockDatabase) CreateObject

func (m *MockDatabase) CreateObject(ctx context.Context, obj *models.InternalObject) error

func (*MockDatabase) CreateObjectAlias

func (m *MockDatabase) CreateObjectAlias(ctx context.Context, aliasID, canonicalObjectID string) error

func (*MockDatabase) DeleteObject

func (m *MockDatabase) DeleteObject(ctx context.Context, id string) error

func (*MockDatabase) DeleteObjectAlias

func (m *MockDatabase) DeleteObjectAlias(ctx context.Context, aliasID string) error

func (*MockDatabase) DeleteS3Credential

func (m *MockDatabase) DeleteS3Credential(ctx context.Context, bucket string) error

func (*MockDatabase) GetBucketScope

func (m *MockDatabase) GetBucketScope(ctx context.Context, organization, projectID string) (*models.BucketScope, error)

func (*MockDatabase) GetBulkObjects

func (m *MockDatabase) GetBulkObjects(ctx context.Context, ids []string) ([]models.InternalObject, error)

func (*MockDatabase) GetFileUsage

func (m *MockDatabase) GetFileUsage(ctx context.Context, objectID string) (*models.FileUsage, error)

func (*MockDatabase) GetFileUsageSummary

func (m *MockDatabase) GetFileUsageSummary(ctx context.Context, inactiveSince *time.Time) (models.FileUsageSummary, error)

func (*MockDatabase) GetObject

func (m *MockDatabase) GetObject(ctx context.Context, id string) (*models.InternalObject, error)

func (*MockDatabase) GetObjectsByChecksum

func (m *MockDatabase) GetObjectsByChecksum(ctx context.Context, checksum string) ([]models.InternalObject, error)

func (*MockDatabase) GetObjectsByChecksums

func (m *MockDatabase) GetObjectsByChecksums(ctx context.Context, checksums []string) (map[string][]models.InternalObject, error)

func (*MockDatabase) GetPendingLFSMeta

func (m *MockDatabase) GetPendingLFSMeta(ctx context.Context, oid string) (*models.PendingLFSMeta, error)

func (*MockDatabase) GetS3Credential

func (m *MockDatabase) GetS3Credential(ctx context.Context, bucket string) (*models.S3Credential, error)

func (*MockDatabase) GetServiceInfo

func (m *MockDatabase) GetServiceInfo(ctx context.Context) (*drs.Service, error)

func (*MockDatabase) GetTransferAttributionBreakdown added in v0.2.5

func (m *MockDatabase) GetTransferAttributionBreakdown(ctx context.Context, filter models.TransferAttributionFilter, groupBy string) ([]models.TransferAttributionBreakdown, error)

func (*MockDatabase) GetTransferAttributionSummary added in v0.2.5

func (m *MockDatabase) GetTransferAttributionSummary(ctx context.Context, filter models.TransferAttributionFilter) (models.TransferAttributionSummary, error)

func (*MockDatabase) ListBucketScopes

func (m *MockDatabase) ListBucketScopes(ctx context.Context) ([]models.BucketScope, error)

func (*MockDatabase) ListFileUsage

func (m *MockDatabase) ListFileUsage(ctx context.Context, limit, offset int, inactiveSince *time.Time) ([]models.FileUsage, error)

func (*MockDatabase) ListObjectIDsByScope added in v0.2.5

func (m *MockDatabase) ListObjectIDsByScope(ctx context.Context, organization, project string) ([]string, error)

func (*MockDatabase) ListProviderTransferSyncRuns added in v0.2.5

func (m *MockDatabase) ListProviderTransferSyncRuns(ctx context.Context, filter models.TransferAttributionFilter, limit int) ([]models.ProviderTransferSyncRun, error)

func (*MockDatabase) ListS3Credentials

func (m *MockDatabase) ListS3Credentials(ctx context.Context) ([]models.S3Credential, error)

func (*MockDatabase) PopPendingLFSMeta

func (m *MockDatabase) PopPendingLFSMeta(ctx context.Context, oid string) (*models.PendingLFSMeta, error)

func (*MockDatabase) RecordFileDownload

func (m *MockDatabase) RecordFileDownload(ctx context.Context, objectID string) error

func (*MockDatabase) RecordFileUpload

func (m *MockDatabase) RecordFileUpload(ctx context.Context, objectID string) error

func (*MockDatabase) RecordProviderTransferEvents added in v0.2.5

func (m *MockDatabase) RecordProviderTransferEvents(ctx context.Context, events []models.ProviderTransferEvent) error

func (*MockDatabase) RecordProviderTransferSyncRuns added in v0.2.5

func (m *MockDatabase) RecordProviderTransferSyncRuns(ctx context.Context, runs []models.ProviderTransferSyncRun) error

func (*MockDatabase) RecordTransferAttributionEvents added in v0.2.5

func (m *MockDatabase) RecordTransferAttributionEvents(ctx context.Context, events []models.TransferAttributionEvent) error

func (*MockDatabase) RegisterObjects

func (m *MockDatabase) RegisterObjects(ctx context.Context, objects []models.InternalObject) error

func (*MockDatabase) ResolveObjectAlias

func (m *MockDatabase) ResolveObjectAlias(ctx context.Context, aliasID string) (string, error)

func (*MockDatabase) SavePendingLFSMeta

func (m *MockDatabase) SavePendingLFSMeta(ctx context.Context, entries []models.PendingLFSMeta) error

func (*MockDatabase) SaveS3Credential

func (m *MockDatabase) SaveS3Credential(ctx context.Context, cred *models.S3Credential) error

func (*MockDatabase) UpdateObjectAccessMethods

func (m *MockDatabase) UpdateObjectAccessMethods(ctx context.Context, objectID string, accessMethods []drs.AccessMethod) error

type MockUrlManager

type MockUrlManager struct{}

MockUrlManager implements urlmanager.UrlManager for testing

func (*MockUrlManager) CompleteMultipartUpload

func (m *MockUrlManager) CompleteMultipartUpload(ctx context.Context, bucket string, key string, uploadId string, parts []urlmanager.MultipartPart) error

func (*MockUrlManager) InitMultipartUpload

func (m *MockUrlManager) InitMultipartUpload(ctx context.Context, bucket string, key string) (string, error)

func (*MockUrlManager) SignDownloadPart

func (m *MockUrlManager) SignDownloadPart(ctx context.Context, accessId string, url string, start int64, end int64, opts urlmanager.SignOptions) (string, error)

func (*MockUrlManager) SignMultipartPart

func (m *MockUrlManager) SignMultipartPart(ctx context.Context, bucket string, key string, uploadId string, partNumber int32) (string, error)

func (*MockUrlManager) SignURL

func (m *MockUrlManager) SignURL(ctx context.Context, accessId string, url string, opts urlmanager.SignOptions) (string, error)

func (*MockUrlManager) SignUploadURL

func (m *MockUrlManager) SignUploadURL(ctx context.Context, accessId string, url string, opts urlmanager.SignOptions) (string, error)

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL