mocks

package
v0.4.0 Latest Latest
Warning

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

 Go to latest Published: Jul 19, 2026 License: AGPL-3.0 Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type MockAuthMiddlewareService 

type MockAuthMiddlewareService struct {
	mock.Mock
}

MockAuthMiddlewareService is a mock implementation of the AuthService interface for middleware tests

func (*MockAuthMiddlewareService) AuthMiddleware 

func (m *MockAuthMiddlewareService) AuthMiddleware() gin.HandlerFunc

func (*MockAuthMiddlewareService) AuthenticateAPIKey 

func (m *MockAuthMiddlewareService) AuthenticateAPIKey(ctx context.Context, apiKey string) (string, error)

func (*MockAuthMiddlewareService) CheckResourceAccess 

func (m *MockAuthMiddlewareService) CheckResourceAccess(userID, resourceType, resourceID, action string) bool

func (*MockAuthMiddlewareService) ClearUserSuspended 

func (m *MockAuthMiddlewareService) ClearUserSuspended(ctx context.Context, userID string) error

ClearUserSuspended records the F4 marker-clear on unsuspend.

func (*MockAuthMiddlewareService) CreateAPIKey 

func (m *MockAuthMiddlewareService) CreateAPIKey(ctx context.Context, userID string, req types.CreateAPIKeyRequest, sessionID string, matchedSigningKey []byte) (*types.APIKey, error)

func (*MockAuthMiddlewareService) DeleteAPIKey 

func (m *MockAuthMiddlewareService) DeleteAPIKey(ctx context.Context, userID, keyID string) error

func (*MockAuthMiddlewareService) GenerateToken 

func (m *MockAuthMiddlewareService) GenerateToken(userID string) (string, error)

func (*MockAuthMiddlewareService) GetUserID 

func (m *MockAuthMiddlewareService) GetUserID(c *gin.Context) string

func (*MockAuthMiddlewareService) ListAPIKeys 

func (m *MockAuthMiddlewareService) ListAPIKeys(ctx context.Context, userID string) ([]*types.APIKey, error)

func (*MockAuthMiddlewareService) Login 

func (m *MockAuthMiddlewareService) Login(ctx context.Context, req types.LoginRequest) (*types.AuthResponse, error)

func (*MockAuthMiddlewareService) MarkUserSuspended 

func (m *MockAuthMiddlewareService) MarkUserSuspended(ctx context.Context, userID string) error

MarkUserSuspended records the F4 revocation-marker write for assertion. No-op by default (returns nil) so tests that do not exercise the suspend path are unaffected.

func (*MockAuthMiddlewareService) OptionalAuthMiddleware 

func (m *MockAuthMiddlewareService) OptionalAuthMiddleware() gin.HandlerFunc

func (*MockAuthMiddlewareService) Register 

func (m *MockAuthMiddlewareService) Register(ctx context.Context, req types.RegisterRequest) (*types.AuthResponse, error)

func (*MockAuthMiddlewareService) RevokeToken 

func (m *MockAuthMiddlewareService) RevokeToken(_ context.Context, token string) error

RevokeToken records the call and returns the configured error. Used by /auth/logout (G18, Epic 17). Tests that don't care about revocation can omit `On("RevokeToken", ...)` — the mock returns nil by default when called via the testify framework.

func (*MockAuthMiddlewareService) Start 

func (m *MockAuthMiddlewareService) Start() error

func (*MockAuthMiddlewareService) Stop 

func (m *MockAuthMiddlewareService) Stop() error

func (*MockAuthMiddlewareService) ValidateToken 

func (m *MockAuthMiddlewareService) ValidateToken(ctx context.Context, token string) (string, error)

type MockCacheService 

type MockCacheService struct {
	mock.Mock
}

MockCacheService implements the CacheService interface for testing.

func (*MockCacheService) Delete 

func (m *MockCacheService) Delete(ctx context.Context, key string) error

func (*MockCacheService) DeleteByPrefix 

func (m *MockCacheService) DeleteByPrefix(ctx context.Context, prefix string) error

func (*MockCacheService) DeleteSession 

func (m *MockCacheService) DeleteSession(ctx context.Context, sessionID string) error

func (*MockCacheService) Get 

func (m *MockCacheService) Get(ctx context.Context, key string) (string, error)

func (*MockCacheService) GetObject 

func (m *MockCacheService) GetObject(ctx context.Context, key string, value interface{}) error

func (*MockCacheService) GetSession 

func (m *MockCacheService) GetSession(ctx context.Context, sessionID string) (*types.CachedSession, error)

func (*MockCacheService) Ping 

func (m *MockCacheService) Ping(ctx context.Context) error

func (*MockCacheService) Set 

func (m *MockCacheService) Set(ctx context.Context, key string, value string, expiration time.Duration) error

func (*MockCacheService) SetNX 

func (m *MockCacheService) SetNX(ctx context.Context, key string, value string, expiration time.Duration) (bool, error)

func (*MockCacheService) SetObject 

func (m *MockCacheService) SetObject(ctx context.Context, key string, value interface{}, expiration time.Duration) error

func (*MockCacheService) SetSession 

func (m *MockCacheService) SetSession(ctx context.Context, sessionID string, session types.CachedSession, expiration time.Duration) error

func (*MockCacheService) Start 

func (m *MockCacheService) Start() error

func (*MockCacheService) Stop 

func (m *MockCacheService) Stop() error

type MockDatabaseService 

type MockDatabaseService struct {
	mock.Mock
}

MockDatabaseService implements the DatabaseService interface for testing.

func (*MockDatabaseService) CheckPermission 

func (m *MockDatabaseService) CheckPermission(ctx context.Context, userID, resourceType, resourceID, action string) (bool, error)

func (*MockDatabaseService) CheckResourceOwnership 

func (m *MockDatabaseService) CheckResourceOwnership(ctx context.Context, userID, resourceType, resourceID string) (bool, error)

func (*MockDatabaseService) CountActiveWorkspacesByUserAndOrg 

func (m *MockDatabaseService) CountActiveWorkspacesByUserAndOrg(ctx context.Context, userID, orgID string) (int, error)

func (*MockDatabaseService) CountUsers 

func (m *MockDatabaseService) CountUsers(ctx context.Context) (int, error)

func (*MockDatabaseService) CountWorkspacesByUserAndOrg 

func (m *MockDatabaseService) CountWorkspacesByUserAndOrg(ctx context.Context, userID, orgID string) (int, error)

func (*MockDatabaseService) CreateAPIKey 

func (m *MockDatabaseService) CreateAPIKey(ctx context.Context, apiKey *types.APIKey) error

func (*MockDatabaseService) CreateUser 

func (m *MockDatabaseService) CreateUser(ctx context.Context, user *types.User) error

func (*MockDatabaseService) CreateWorkspace 

func (m *MockDatabaseService) CreateWorkspace(ctx context.Context, workspace *types.WorkspaceMetadata) error

func (*MockDatabaseService) DeleteAPIKey 

func (m *MockDatabaseService) DeleteAPIKey(ctx context.Context, userID, keyID string) error

func (*MockDatabaseService) DeleteSessionIndex 

func (m *MockDatabaseService) DeleteSessionIndex(ctx context.Context, workspaceID string) error

func (*MockDatabaseService) DeleteSessionTree 

func (m *MockDatabaseService) DeleteSessionTree(ctx context.Context, workspaceID, sessionID string) error

func (*MockDatabaseService) DeleteUser 

func (m *MockDatabaseService) DeleteUser(ctx context.Context, userID string) error

func (*MockDatabaseService) DeleteWorkspace 

func (m *MockDatabaseService) DeleteWorkspace(ctx context.Context, workspaceID string) error

func (*MockDatabaseService) GetAPIKey 

func (m *MockDatabaseService) GetAPIKey(ctx context.Context, userID, keyID string) (*types.APIKey, error)

func (*MockDatabaseService) GetAPIKeyRecordByHash 

func (m *MockDatabaseService) GetAPIKeyRecordByHash(ctx context.Context, keyHash string) (*types.APIKey, error)

func (*MockDatabaseService) GetUser 

func (m *MockDatabaseService) GetUser(ctx context.Context, userID string) (*types.User, error)

func (*MockDatabaseService) GetUserByAPIKey 

func (m *MockDatabaseService) GetUserByAPIKey(ctx context.Context, apiKey string) (*types.User, error)

func (*MockDatabaseService) GetUserByEmail 

func (m *MockDatabaseService) GetUserByEmail(ctx context.Context, email string) (*types.User, error)

func (*MockDatabaseService) GetWorkspace 

func (m *MockDatabaseService) GetWorkspace(ctx context.Context, workspaceID string) (*types.WorkspaceMetadata, error)

func (*MockDatabaseService) ListAPIKeys 

func (m *MockDatabaseService) ListAPIKeys(ctx context.Context, userID string) ([]*types.APIKey, error)

func (*MockDatabaseService) ListAPIKeysWithDecrypt 

func (m *MockDatabaseService) ListAPIKeysWithDecrypt(ctx context.Context, userID string) ([]*types.APIKey, error)

func (*MockDatabaseService) ListAllWorkspaceOwners 

func (m *MockDatabaseService) ListAllWorkspaceOwners(ctx context.Context) (map[string]string, error)

func (*MockDatabaseService) ListSessionIndex 

func (m *MockDatabaseService) ListSessionIndex(ctx context.Context, workspaceID string) ([]types.SessionListItem, error)

func (*MockDatabaseService) ListWorkspaces 

func (m *MockDatabaseService) ListWorkspaces(ctx context.Context, userID string, limit, offset int) ([]*types.WorkspaceMetadata, *types.PaginationMetadata, error)

func (*MockDatabaseService) MarkWorkspaceDeleted 

func (m *MockDatabaseService) MarkWorkspaceDeleted(ctx context.Context, workspaceID string)

func (*MockDatabaseService) Ping 

func (m *MockDatabaseService) Ping(ctx context.Context) error

func (*MockDatabaseService) SetUserStatus 

func (m *MockDatabaseService) SetUserStatus(ctx context.Context, userID string, status types.UserStatus) error

func (*MockDatabaseService) Start 

func (m *MockDatabaseService) Start() error

func (*MockDatabaseService) Stop 

func (m *MockDatabaseService) Stop() error

func (*MockDatabaseService) SyncWorkspaceVersionInfo 

func (m *MockDatabaseService) SyncWorkspaceVersionInfo(ctx context.Context, workspaceID, imageTag, agentVersion string)

func (*MockDatabaseService) UpdateAPIKeyDEK 

func (m *MockDatabaseService) UpdateAPIKeyDEK(ctx context.Context, keyID string, wrappedDEK, kekSalt []byte, synced bool) error

func (*MockDatabaseService) UpdateSessionLastSeen 

func (m *MockDatabaseService) UpdateSessionLastSeen(ctx context.Context, workspaceID, sessionID string) error

func (*MockDatabaseService) UpdateUser 

func (m *MockDatabaseService) UpdateUser(ctx context.Context, userID string, updates types.UserUpdates) error

func (*MockDatabaseService) UpdateWorkspace 

func (m *MockDatabaseService) UpdateWorkspace(ctx context.Context, workspaceID string, updates types.WorkspaceUpdates) error

func (*MockDatabaseService) UpsertSessionContextUsed 

func (m *MockDatabaseService) UpsertSessionContextUsed(ctx context.Context, workspaceID, sessionID string, contextUsed int64) error

func (*MockDatabaseService) UpsertSessionMessage 

func (m *MockDatabaseService) UpsertSessionMessage(ctx context.Context, workspaceID, sessionID string, at time.Time) error

func (*MockDatabaseService) UpsertSessionParent 

func (m *MockDatabaseService) UpsertSessionParent(ctx context.Context, workspaceID, sessionID, parentID string) error

func (*MockDatabaseService) UpsertSessionTitle 

func (m *MockDatabaseService) UpsertSessionTitle(ctx context.Context, workspaceID, sessionID, title string) error

type MockMeteringService 

type MockMeteringService struct {
	mock.Mock
}

func (*MockMeteringService) CheckQuota 

func (m *MockMeteringService) CheckQuota(ctx context.Context, owner types.BillingOwner, eventType string) (bool, int64, error)

func (*MockMeteringService) ExportUsage 

func (m *MockMeteringService) ExportUsage(ctx context.Context) (int, error)

func (*MockMeteringService) GetQuotaStatus 

func (m *MockMeteringService) GetQuotaStatus(ctx context.Context, owner types.BillingOwner) ([]types.QuotaStatus, error)

func (*MockMeteringService) GetUsage 

func (m *MockMeteringService) GetUsage(ctx context.Context, owner types.BillingOwner, from, to time.Time) (*types.UsageReport, error)

func (*MockMeteringService) GetUsageByWorkspace 

func (m *MockMeteringService) GetUsageByWorkspace(ctx context.Context, owner types.BillingOwner, workspaceID string, from, to time.Time) (*types.UsageReport, error)

func (*MockMeteringService) Record 

func (m *MockMeteringService) Record(event types.UsageEvent)

func (*MockMeteringService) RecordLifecycleEvent 

func (m *MockMeteringService) RecordLifecycleEvent(ctx context.Context, workspaceID, ownerID string, ownerType types.OwnerType, fromPhase, toPhase, resourceTier string, eventTime time.Time) error

func (*MockMeteringService) Start 

func (m *MockMeteringService) Start() error

func (*MockMeteringService) Stop 

func (m *MockMeteringService) Stop() error

type MockMetricsService 

type MockMetricsService struct {
	mock.Mock
}

func (*MockMetricsService) DecrementActiveConnections 

func (m *MockMetricsService) DecrementActiveConnections(connType, userID string)

func (*MockMetricsService) IncrementActiveConnections 

func (m *MockMetricsService) IncrementActiveConnections(connType, userID string)

func (*MockMetricsService) RecordError 

func (m *MockMetricsService) RecordError(errorType, endpoint, code string)

func (*MockMetricsService) RecordRequest 

func (m *MockMetricsService) RecordRequest(method, path string, status int, duration time.Duration, size int)

func (*MockMetricsService) RecordResourceUsage 

func (m *MockMetricsService) RecordResourceUsage(workspaceID string, cpu float64, memoryBytes int64)

func (*MockMetricsService) RecordWorkspaceCreation 

func (m *MockMetricsService) RecordWorkspaceCreation(runtime, userID string)

func (*MockMetricsService) RecordWorkspaceTermination 

func (m *MockMetricsService) RecordWorkspaceTermination(runtime, reason string)

func (*MockMetricsService) Start 

func (m *MockMetricsService) Start() error

func (*MockMetricsService) Stop 

func (m *MockMetricsService) Stop() error

type MockRateLimiterService 

type MockRateLimiterService struct {
	mock.Mock
}

MockRateLimiterService implements the RateLimiterService interface for testing

func (*MockRateLimiterService) AddToWindow 

func (m *MockRateLimiterService) AddToWindow(ctx context.Context, key string, timestamp int64, member string, expiration time.Duration) error

func (*MockRateLimiterService) Allow 

func (m *MockRateLimiterService) Allow(key string, rate float64, burst int) bool

func (*MockRateLimiterService) CountInWindow 

func (m *MockRateLimiterService) CountInWindow(ctx context.Context, key string, min, max int64) (int, error)

func (*MockRateLimiterService) GetTTL 

func (m *MockRateLimiterService) GetTTL(ctx context.Context, key string) (time.Duration, error)

func (*MockRateLimiterService) GetWindowEntries 

func (m *MockRateLimiterService) GetWindowEntries(ctx context.Context, key string, start, stop int) ([]string, error)

func (*MockRateLimiterService) Increment 

func (m *MockRateLimiterService) Increment(ctx context.Context, key string, value int64, expiration time.Duration) (int64, error)

func (*MockRateLimiterService) RemoveFromWindow 

func (m *MockRateLimiterService) RemoveFromWindow(ctx context.Context, key string, cutoff int64) error

func (*MockRateLimiterService) Start 

func (m *MockRateLimiterService) Start() error

func (*MockRateLimiterService) Stop 

func (m *MockRateLimiterService) Stop() error

type MockSessionManager 

type MockSessionManager struct {
	mock.Mock
}

MockSessionManager implements the SessionManager interface for testing

func (*MockSessionManager) CancelExecution 

func (m *MockSessionManager) CancelExecution(sessionID, executionID string) bool

func (*MockSessionManager) CloseSession 

func (m *MockSessionManager) CloseSession(sessionID string)

func (*MockSessionManager) CreateSession 

func (m *MockSessionManager) CreateSession(userID, workspaceID string, conn types.WSConnection) (*types.Session, error)

func (*MockSessionManager) GetSession 

func (m *MockSessionManager) GetSession(sessionID string) (*types.Session, error)

func (*MockSessionManager) SetCancellationFunc 

func (m *MockSessionManager) SetCancellationFunc(sessionID, executionID string, cancel context.CancelFunc)

func (*MockSessionManager) Start 

func (m *MockSessionManager) Start() error

func (*MockSessionManager) Stop 

func (m *MockSessionManager) Stop() error

type MockWorkspaceService 

type MockWorkspaceService struct {
	mock.Mock
}

MockWorkspaceService implements interfaces.WorkspaceService for testing.

func (*MockWorkspaceService) ActivateWorkspace 

func (m *MockWorkspaceService) ActivateWorkspace(ctx context.Context, userID, workspaceID string) (*types.ActivateWorkspaceResponse, error)

func (*MockWorkspaceService) CheckOwnership 

func (m *MockWorkspaceService) CheckOwnership(ctx context.Context, userID string, meta *types.WorkspaceMetadata) error

func (*MockWorkspaceService) CreateWorkspace 

func (m *MockWorkspaceService) CreateWorkspace(ctx context.Context, userID string, req types.CreateWorkspaceRequest) (*types.Workspace, error)

func (*MockWorkspaceService) DeleteWorkspace 

func (m *MockWorkspaceService) DeleteWorkspace(ctx context.Context, userID, workspaceID string) error

func (*MockWorkspaceService) EnsureSession 

func (m *MockWorkspaceService) EnsureSession(ctx context.Context, userID, workspaceID string) (*types.EnsureSessionResponse, error)

func (*MockWorkspaceService) GetWorkspace 

func (m *MockWorkspaceService) GetWorkspace(ctx context.Context, userID, workspaceID string) (*types.Workspace, error)

func (*MockWorkspaceService) GetWorkspaceStatus 

func (m *MockWorkspaceService) GetWorkspaceStatus(ctx context.Context, userID, workspaceID string) (*types.WorkspaceStatusResult, error)

func (*MockWorkspaceService) ListWorkspaceSessions 

func (m *MockWorkspaceService) ListWorkspaceSessions(ctx context.Context, userID, workspaceID string) ([]types.SessionListItem, error)

func (*MockWorkspaceService) ListWorkspaces 

func (m *MockWorkspaceService) ListWorkspaces(ctx context.Context, userID string, opts types.ListOptions) (*types.WorkspaceListResult, error)

func (*MockWorkspaceService) MarkSessionSeen 

func (m *MockWorkspaceService) MarkSessionSeen(ctx context.Context, userID, workspaceID, sessionID string) error

func (*MockWorkspaceService) RefreshWorkspaceCompute 

func (m *MockWorkspaceService) RefreshWorkspaceCompute(ctx context.Context, userID, workspaceID string) (*types.RefreshWorkspaceResult, error)

func (*MockWorkspaceService) RenameSession 

func (m *MockWorkspaceService) RenameSession(ctx context.Context, userID, workspaceID, sessionID, title string) error

func (*MockWorkspaceService) RenameWorkspace 

func (m *MockWorkspaceService) RenameWorkspace(ctx context.Context, userID, workspaceID, name string) error

func (*MockWorkspaceService) ResolveWorkspace 

func (m *MockWorkspaceService) ResolveWorkspace(ctx context.Context, workspaceID string) (*types.WorkspaceMetadata, error)

func (*MockWorkspaceService) RestartWorkspace 

func (m *MockWorkspaceService) RestartWorkspace(ctx context.Context, userID, workspaceID string) error

func (*MockWorkspaceService) Start 

func (m *MockWorkspaceService) Start() error

func (*MockWorkspaceService) Stop 

func (m *MockWorkspaceService) Stop() error

func (*MockWorkspaceService) SuspendWorkspace 

func (m *MockWorkspaceService) SuspendWorkspace(ctx context.Context, userID, workspaceID string) error

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.