diff --git a/backend/internal/handler/auth_handler_async_test.go b/backend/internal/handler/auth_handler_async_test.go index c7ff6139..00212e60 100644 --- a/backend/internal/handler/auth_handler_async_test.go +++ b/backend/internal/handler/auth_handler_async_test.go @@ -171,6 +171,9 @@ func (m *AsyncMockTenantService) ListTenants(ctx context.Context, limit, offset func (m *AsyncMockTenantService) ListManageableTenants(ctx context.Context, userID string) ([]domain.Tenant, error) { return nil, nil } +func (m *AsyncMockTenantService) IsDomainAllowed(ctx context.Context, domainName string) (bool, error) { + return false, nil +} func (m *AsyncMockTenantService) ApproveTenant(ctx context.Context, id string) error { return nil } func (m *AsyncMockTenantService) SetKetoService(keto service.KetoService) {} func (m *AsyncMockTenantService) AddTenantAdmin(ctx context.Context, tenantID, userID string) error { diff --git a/backend/internal/handler/tenant_handler_test.go b/backend/internal/handler/tenant_handler_test.go index b608b550..f6e785ca 100644 --- a/backend/internal/handler/tenant_handler_test.go +++ b/backend/internal/handler/tenant_handler_test.go @@ -71,6 +71,19 @@ func (m *MockTenantService) ListTenants(ctx context.Context, limit, offset int, return args.Get(0).([]domain.Tenant), args.Get(1).(int64), args.Error(2) } +func (m *MockTenantService) ListManageableTenants(ctx context.Context, userID string) ([]domain.Tenant, error) { + args := m.Called(ctx, userID) + if args.Get(0) == nil { + return nil, args.Error(1) + } + return args.Get(0).([]domain.Tenant), args.Error(1) +} + +func (m *MockTenantService) IsDomainAllowed(ctx context.Context, domainName string) (bool, error) { + args := m.Called(ctx, domainName) + return args.Bool(0), args.Error(1) +} + func (m *MockTenantService) SetKetoService(keto service.KetoService) { m.Called(keto) } diff --git a/backend/internal/middleware/tenant_middleware_test.go b/backend/internal/middleware/tenant_middleware_test.go index 5d8f21a2..b3d1cd2e 100644 --- a/backend/internal/middleware/tenant_middleware_test.go +++ b/backend/internal/middleware/tenant_middleware_test.go @@ -4,7 +4,6 @@ import ( "baron-sso-backend/internal/domain" "baron-sso-backend/internal/service" "context" - "log/slog" "net/http/httptest" "os" "testing" @@ -21,9 +20,11 @@ type MockTenantServiceForMiddleware struct { func (m *MockTenantServiceForMiddleware) RegisterTenant(ctx context.Context, name, slug, tenantType, description string, domains []string, parentID *string, creatorID string) (*domain.Tenant, error) { return nil, nil } + func (m *MockTenantServiceForMiddleware) RequestRegistration(ctx context.Context, name, slug, description string, domainName string, adminEmail string) (*domain.Tenant, error) { return nil, nil } + func (m *MockTenantServiceForMiddleware) GetTenantByDomain(ctx context.Context, emailDomain string) (*domain.Tenant, error) { args := m.Called(mock.Anything, emailDomain) if args.Get(0) == nil { @@ -31,6 +32,7 @@ func (m *MockTenantServiceForMiddleware) GetTenantByDomain(ctx context.Context, } return args.Get(0).(*domain.Tenant), args.Error(1) } + func (m *MockTenantServiceForMiddleware) GetTenantBySlug(ctx context.Context, slug string) (*domain.Tenant, error) { args := m.Called(mock.Anything, slug) if args.Get(0) == nil { @@ -38,18 +40,23 @@ func (m *MockTenantServiceForMiddleware) GetTenantBySlug(ctx context.Context, sl } return args.Get(0).(*domain.Tenant), args.Error(1) } + func (m *MockTenantServiceForMiddleware) GetTenant(ctx context.Context, id string) (*domain.Tenant, error) { return nil, nil } + func (m *MockTenantServiceForMiddleware) ListTenants(ctx context.Context, limit, offset int, parentID string) ([]domain.Tenant, int64, error) { return nil, 0, nil } + func (m *MockTenantServiceForMiddleware) ListManageableTenants(ctx context.Context, userID string) ([]domain.Tenant, error) { return nil, nil } + func (m *MockTenantServiceForMiddleware) IsDomainAllowed(ctx context.Context, domainName string) (bool, error) { return false, nil } + func (m *MockTenantServiceForMiddleware) ApproveTenant(ctx context.Context, id string) error { return nil }