refactor: standardize modular architecture patterns

- Rename module definition variables to follow naming convention:
  - catalog/definition.py: module → catalog_module
  - checkout/definition.py: module → checkout_module
  - cart/definition.py: module → cart_module

- Add router attachment functions for lazy loading:
  - get_catalog_module_with_routers()
  - get_checkout_module_with_routers()
  - get_cart_module_with_routers()

- Move billing exceptions to exceptions.py:
  - Add backwards-compatible aliases (BillingServiceError, etc.)
  - Update billing_service.py to import from exceptions.py

- Standardize VendorEmailSettingsService DI pattern:
  - Change from db in __init__ to db as method parameter
  - Create singleton vendor_email_settings_service instance
  - Update routes and tests to use new pattern

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
2026-02-03 18:40:03 +01:00
parent b769f5a047
commit a76128e016
11 changed files with 209 additions and 142 deletions

View File

@@ -19,10 +19,13 @@ from app.exceptions.base import (
__all__ = [ __all__ = [
# Base billing exception # Base billing exception
"BillingException", "BillingException",
"BillingServiceError", # Alias for backwards compatibility
# Subscription exceptions # Subscription exceptions
"SubscriptionNotFoundException", "SubscriptionNotFoundException",
"NoActiveSubscriptionException", "NoActiveSubscriptionException",
"NoActiveSubscriptionError", # Alias for backwards compatibility
"SubscriptionNotCancelledException", "SubscriptionNotCancelledException",
"SubscriptionNotCancelledError", # Alias for backwards compatibility
"SubscriptionAlreadyCancelledException", "SubscriptionAlreadyCancelledException",
# Tier exceptions # Tier exceptions
"TierNotFoundException", "TierNotFoundException",
@@ -30,8 +33,10 @@ __all__ = [
"TierLimitExceededException", "TierLimitExceededException",
# Payment exceptions # Payment exceptions
"PaymentSystemNotConfiguredException", "PaymentSystemNotConfiguredException",
"PaymentSystemNotConfiguredError", # Alias for backwards compatibility
"StripeNotConfiguredException", "StripeNotConfiguredException",
"StripePriceNotConfiguredException", "StripePriceNotConfiguredException",
"StripePriceNotConfiguredError", # Alias for backwards compatibility
"PaymentFailedException", "PaymentFailedException",
# Webhook exceptions # Webhook exceptions
"InvalidWebhookSignatureException", "InvalidWebhookSignatureException",
@@ -57,6 +62,10 @@ class BillingException(BusinessLogicException):
super().__init__(message=message, error_code=error_code, details=details) super().__init__(message=message, error_code=error_code, details=details)
# Alias for backwards compatibility with billing_service.py
BillingServiceError = BillingException
# ============================================================================= # =============================================================================
# Subscription Exceptions # Subscription Exceptions
# ============================================================================= # =============================================================================
@@ -83,6 +92,10 @@ class NoActiveSubscriptionException(BusinessLogicException):
) )
# Alias for backwards compatibility with billing_service.py
NoActiveSubscriptionError = NoActiveSubscriptionException
class SubscriptionNotCancelledException(BusinessLogicException): class SubscriptionNotCancelledException(BusinessLogicException):
"""Raised when trying to reactivate a subscription that is not cancelled.""" """Raised when trying to reactivate a subscription that is not cancelled."""
@@ -93,6 +106,10 @@ class SubscriptionNotCancelledException(BusinessLogicException):
) )
# Alias for backwards compatibility with billing_service.py
SubscriptionNotCancelledError = SubscriptionNotCancelledException
class SubscriptionAlreadyCancelledException(BusinessLogicException): class SubscriptionAlreadyCancelledException(BusinessLogicException):
"""Raised when trying to cancel an already cancelled subscription.""" """Raised when trying to cancel an already cancelled subscription."""
@@ -163,6 +180,10 @@ class PaymentSystemNotConfiguredException(ServiceUnavailableException):
super().__init__(message="Payment system not configured") super().__init__(message="Payment system not configured")
# Alias for backwards compatibility with billing_service.py
PaymentSystemNotConfiguredError = PaymentSystemNotConfiguredException
class StripeNotConfiguredException(BillingException): class StripeNotConfiguredException(BillingException):
"""Raised when Stripe is not configured.""" """Raised when Stripe is not configured."""
@@ -185,6 +206,10 @@ class StripePriceNotConfiguredException(BusinessLogicException):
self.tier_code = tier_code self.tier_code = tier_code
# Alias for backwards compatibility with billing_service.py
StripePriceNotConfiguredError = StripePriceNotConfiguredException
class PaymentFailedException(BillingException): class PaymentFailedException(BillingException):
"""Raised when a payment fails.""" """Raised when a payment fails."""

View File

@@ -20,6 +20,8 @@ from app.modules.billing.services.admin_subscription_service import (
from app.modules.billing.services.billing_service import ( from app.modules.billing.services.billing_service import (
BillingService, BillingService,
billing_service, billing_service,
)
from app.modules.billing.exceptions import (
BillingServiceError, BillingServiceError,
PaymentSystemNotConfiguredError, PaymentSystemNotConfiguredError,
TierNotFoundError, TierNotFoundError,

View File

@@ -23,54 +23,19 @@ from app.modules.billing.models import (
VendorAddOn, VendorAddOn,
VendorSubscription, VendorSubscription,
) )
from app.modules.billing.exceptions import (
BillingServiceError,
NoActiveSubscriptionError,
PaymentSystemNotConfiguredError,
StripePriceNotConfiguredError,
SubscriptionNotCancelledError,
TierNotFoundError,
)
from app.modules.tenancy.models import Vendor from app.modules.tenancy.models import Vendor
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class BillingServiceError(Exception):
"""Base exception for billing service errors."""
pass
class PaymentSystemNotConfiguredError(BillingServiceError):
"""Raised when Stripe is not configured."""
def __init__(self):
super().__init__("Payment system not configured")
class TierNotFoundError(BillingServiceError):
"""Raised when a tier is not found."""
def __init__(self, tier_code: str):
self.tier_code = tier_code
super().__init__(f"Tier '{tier_code}' not found")
class StripePriceNotConfiguredError(BillingServiceError):
"""Raised when Stripe price is not configured for a tier."""
def __init__(self, tier_code: str):
self.tier_code = tier_code
super().__init__(f"Stripe price not configured for tier '{tier_code}'")
class NoActiveSubscriptionError(BillingServiceError):
"""Raised when no active subscription exists."""
def __init__(self):
super().__init__("No active subscription found")
class SubscriptionNotCancelledError(BillingServiceError):
"""Raised when trying to reactivate a non-cancelled subscription."""
def __init__(self):
super().__init__("Subscription is not cancelled")
class BillingService: class BillingService:
"""Service for billing operations.""" """Service for billing operations."""

View File

@@ -9,7 +9,20 @@ It is session-based and does not require customer authentication.
from app.modules.base import ModuleDefinition, PermissionDefinition from app.modules.base import ModuleDefinition, PermissionDefinition
module = ModuleDefinition( # =============================================================================
# Router Lazy Imports
# =============================================================================
def _get_storefront_router():
"""Lazy import of storefront router to avoid circular imports."""
from app.modules.cart.routes.api.storefront import router
return router
# Cart module definition
cart_module = ModuleDefinition(
code="cart", code="cart",
name="Shopping Cart", name="Shopping Cart",
description="Session-based shopping cart for storefronts", description="Session-based shopping cart for storefronts",
@@ -42,3 +55,17 @@ module = ModuleDefinition(
# Cart is storefront-only - no admin/vendor menus needed # Cart is storefront-only - no admin/vendor menus needed
menu_items={}, menu_items={},
) )
def get_cart_module_with_routers() -> ModuleDefinition:
"""
Get cart module with routers attached.
This function attaches the routers lazily to avoid circular imports
during module initialization.
"""
cart_module.storefront_router = _get_storefront_router()
return cart_module
__all__ = ["cart_module", "get_cart_module_with_routers"]

View File

@@ -9,7 +9,28 @@ from app.modules.base import (
) )
from app.modules.enums import FrontendType from app.modules.enums import FrontendType
module = ModuleDefinition(
# =============================================================================
# Router Lazy Imports
# =============================================================================
def _get_admin_router():
"""Lazy import of admin router to avoid circular imports."""
from app.modules.catalog.routes.api.admin import admin_router
return admin_router
def _get_vendor_router():
"""Lazy import of vendor router to avoid circular imports."""
from app.modules.catalog.routes.api.vendor import vendor_router
return vendor_router
# Catalog module definition
catalog_module = ModuleDefinition(
code="catalog", code="catalog",
name="Product Catalog", name="Product Catalog",
description="Product catalog browsing and search for storefronts", description="Product catalog browsing and search for storefronts",
@@ -85,3 +106,18 @@ module = ModuleDefinition(
], ],
}, },
) )
def get_catalog_module_with_routers() -> ModuleDefinition:
"""
Get catalog module with routers attached.
This function attaches the routers lazily to avoid circular imports
during module initialization.
"""
catalog_module.admin_router = _get_admin_router()
catalog_module.vendor_router = _get_vendor_router()
return catalog_module
__all__ = ["catalog_module", "get_catalog_module_with_routers"]

View File

@@ -9,7 +9,20 @@ Orchestrates payment processing and order creation.
from app.modules.base import ModuleDefinition, PermissionDefinition from app.modules.base import ModuleDefinition, PermissionDefinition
module = ModuleDefinition( # =============================================================================
# Router Lazy Imports
# =============================================================================
def _get_storefront_router():
"""Lazy import of storefront router to avoid circular imports."""
from app.modules.checkout.routes.api.storefront import router
return router
# Checkout module definition
checkout_module = ModuleDefinition(
code="checkout", code="checkout",
name="Checkout", name="Checkout",
description="Checkout and order creation for storefronts", description="Checkout and order creation for storefronts",
@@ -42,3 +55,17 @@ module = ModuleDefinition(
# Checkout is storefront-only - no admin/vendor menus needed # Checkout is storefront-only - no admin/vendor menus needed
menu_items={}, menu_items={},
) )
def get_checkout_module_with_routers() -> ModuleDefinition:
"""
Get checkout module with routers attached.
This function attaches the routers lazily to avoid circular imports
during module initialization.
"""
checkout_module.storefront_router = _get_storefront_router()
return checkout_module
__all__ = ["checkout_module", "get_checkout_module_with_routers"]

View File

@@ -19,7 +19,8 @@ from app.modules.cms.services.vendor_theme_service import (
) )
from app.modules.cms.services.vendor_email_settings_service import ( from app.modules.cms.services.vendor_email_settings_service import (
VendorEmailSettingsService, VendorEmailSettingsService,
get_vendor_email_settings_service, vendor_email_settings_service,
get_vendor_email_settings_service, # Deprecated: use vendor_email_settings_service
) )
__all__ = [ __all__ = [
@@ -30,5 +31,6 @@ __all__ = [
"VendorThemeService", "VendorThemeService",
"vendor_theme_service", "vendor_theme_service",
"VendorEmailSettingsService", "VendorEmailSettingsService",
"get_vendor_email_settings_service", "vendor_email_settings_service",
"get_vendor_email_settings_service", # Deprecated
] ]

View File

@@ -42,24 +42,21 @@ PREMIUM_TIERS = {TierCode.BUSINESS, TierCode.ENTERPRISE}
class VendorEmailSettingsService: class VendorEmailSettingsService:
"""Service for managing vendor email settings.""" """Service for managing vendor email settings."""
def __init__(self, db: Session):
self.db = db
# ========================================================================= # =========================================================================
# READ OPERATIONS # READ OPERATIONS
# ========================================================================= # =========================================================================
def get_settings(self, vendor_id: int) -> VendorEmailSettings | None: def get_settings(self, db: Session, vendor_id: int) -> VendorEmailSettings | None:
"""Get email settings for a vendor.""" """Get email settings for a vendor."""
return ( return (
self.db.query(VendorEmailSettings) db.query(VendorEmailSettings)
.filter(VendorEmailSettings.vendor_id == vendor_id) .filter(VendorEmailSettings.vendor_id == vendor_id)
.first() .first()
) )
def get_settings_or_404(self, vendor_id: int) -> VendorEmailSettings: def get_settings_or_404(self, db: Session, vendor_id: int) -> VendorEmailSettings:
"""Get email settings or raise 404.""" """Get email settings or raise 404."""
settings = self.get_settings(vendor_id) settings = self.get_settings(db, vendor_id)
if not settings: if not settings:
raise ResourceNotFoundException( raise ResourceNotFoundException(
resource_type="vendor_email_settings", resource_type="vendor_email_settings",
@@ -67,19 +64,19 @@ class VendorEmailSettingsService:
) )
return settings return settings
def is_configured(self, vendor_id: int) -> bool: def is_configured(self, db: Session, vendor_id: int) -> bool:
"""Check if vendor has configured email settings.""" """Check if vendor has configured email settings."""
settings = self.get_settings(vendor_id) settings = self.get_settings(db, vendor_id)
return settings is not None and settings.is_configured return settings is not None and settings.is_configured
def get_status(self, vendor_id: int) -> dict: def get_status(self, db: Session, vendor_id: int) -> dict:
""" """
Get email configuration status for a vendor. Get email configuration status for a vendor.
Returns: Returns:
dict with is_configured, is_verified, provider, etc. dict with is_configured, is_verified, provider, etc.
""" """
settings = self.get_settings(vendor_id) settings = self.get_settings(db, vendor_id)
if not settings: if not settings:
return { return {
"is_configured": False, "is_configured": False,
@@ -115,6 +112,7 @@ class VendorEmailSettingsService:
def create_or_update( def create_or_update(
self, self,
db: Session,
vendor_id: int, vendor_id: int,
data: dict, data: dict,
current_tier: TierCode | None = None, current_tier: TierCode | None = None,
@@ -123,6 +121,7 @@ class VendorEmailSettingsService:
Create or update vendor email settings. Create or update vendor email settings.
Args: Args:
db: Database session
vendor_id: Vendor ID vendor_id: Vendor ID
data: Settings data (from_email, from_name, smtp_*, etc.) data: Settings data (from_email, from_name, smtp_*, etc.)
current_tier: Vendor's current subscription tier (for premium provider validation) current_tier: Vendor's current subscription tier (for premium provider validation)
@@ -143,10 +142,10 @@ class VendorEmailSettingsService:
details={"required_permission": "business_tier"}, details={"required_permission": "business_tier"},
) )
settings = self.get_settings(vendor_id) settings = self.get_settings(db, vendor_id)
if not settings: if not settings:
settings = VendorEmailSettings(vendor_id=vendor_id) settings = VendorEmailSettings(vendor_id=vendor_id)
self.db.add(settings) db.add(settings)
# Update fields # Update fields
for field in [ for field in [
@@ -190,36 +189,41 @@ class VendorEmailSettingsService:
settings.is_verified = False settings.is_verified = False
settings.verification_error = None settings.verification_error = None
self.db.flush() db.flush()
logger.info(f"Updated email settings for vendor {vendor_id}: provider={settings.provider}") logger.info(f"Updated email settings for vendor {vendor_id}: provider={settings.provider}")
return settings return settings
def delete(self, vendor_id: int) -> None: def delete(self, db: Session, vendor_id: int) -> None:
""" """
Delete email settings for a vendor. Delete email settings for a vendor.
Args:
db: Database session
vendor_id: Vendor ID
Raises: Raises:
ResourceNotFoundException: If settings not found ResourceNotFoundException: If settings not found
""" """
settings = self.get_settings(vendor_id) settings = self.get_settings(db, vendor_id)
if not settings: if not settings:
raise ResourceNotFoundException( raise ResourceNotFoundException(
resource_type="vendor_email_settings", resource_type="vendor_email_settings",
identifier=str(vendor_id), identifier=str(vendor_id),
) )
self.db.delete(settings) db.delete(settings)
self.db.flush() db.flush()
logger.info(f"Deleted email settings for vendor {vendor_id}") logger.info(f"Deleted email settings for vendor {vendor_id}")
# ========================================================================= # =========================================================================
# VERIFICATION # VERIFICATION
# ========================================================================= # =========================================================================
def verify_settings(self, vendor_id: int, test_email: str) -> dict: def verify_settings(self, db: Session, vendor_id: int, test_email: str) -> dict:
""" """
Verify email settings by sending a test email. Verify email settings by sending a test email.
Args: Args:
db: Database session
vendor_id: Vendor ID vendor_id: Vendor ID
test_email: Email address to send test email to test_email: Email address to send test email to
@@ -230,7 +234,7 @@ class VendorEmailSettingsService:
ResourceNotFoundException: If settings not found ResourceNotFoundException: If settings not found
ValidationException: If settings incomplete ValidationException: If settings incomplete
""" """
settings = self.get_settings_or_404(vendor_id) settings = self.get_settings_or_404(db, vendor_id)
if not settings.is_fully_configured(): if not settings.is_fully_configured():
raise ValidationException( raise ValidationException(
@@ -256,7 +260,7 @@ class VendorEmailSettingsService:
# Mark as verified # Mark as verified
settings.mark_verified() settings.mark_verified()
self.db.flush() db.flush()
logger.info(f"Email settings verified for vendor {vendor_id}") logger.info(f"Email settings verified for vendor {vendor_id}")
return { return {
@@ -269,7 +273,7 @@ class VendorEmailSettingsService:
except Exception as e: except Exception as e:
error_msg = str(e) error_msg = str(e)
settings.mark_verification_failed(error_msg) settings.mark_verification_failed(error_msg)
self.db.flush() db.flush()
logger.warning(f"Email verification failed for vendor {vendor_id}: {error_msg}") logger.warning(f"Email verification failed for vendor {vendor_id}: {error_msg}")
# Return error dict instead of raising - verification failure is not a server error # Return error dict instead of raising - verification failure is not a server error
@@ -476,7 +480,16 @@ class VendorEmailSettingsService:
return providers return providers
# Module-level service factory # Module-level service instance (singleton pattern)
vendor_email_settings_service = VendorEmailSettingsService()
# Deprecated: Factory function for backwards compatibility
def get_vendor_email_settings_service(db: Session) -> VendorEmailSettingsService: def get_vendor_email_settings_service(db: Session) -> VendorEmailSettingsService:
"""Factory function to get a VendorEmailSettingsService instance.""" """
return VendorEmailSettingsService(db) Factory function to get a VendorEmailSettingsService instance.
Deprecated: Use the singleton `vendor_email_settings_service` instead and pass
`db` to individual methods.
"""
return vendor_email_settings_service

View File

@@ -20,7 +20,7 @@ from sqlalchemy.orm import Session
from app.api.deps import get_current_vendor_api from app.api.deps import get_current_vendor_api
from app.core.database import get_db from app.core.database import get_db
from app.modules.cms.services.vendor_email_settings_service import VendorEmailSettingsService from app.modules.cms.services.vendor_email_settings_service import vendor_email_settings_service
from app.modules.billing.services.subscription_service import subscription_service from app.modules.billing.services.subscription_service import subscription_service
from models.schema.auth import UserContext from models.schema.auth import UserContext
@@ -137,9 +137,8 @@ def get_email_settings(
Returns settings with sensitive fields masked. Returns settings with sensitive fields masked.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db)
settings = service.get_settings(vendor_id) settings = vendor_email_settings_service.get_settings(db, vendor_id)
if not settings: if not settings:
return EmailSettingsResponse( return EmailSettingsResponse(
configured=False, configured=False,
@@ -165,8 +164,7 @@ def get_email_status(
Used by frontend to show warning banner if not configured. Used by frontend to show warning banner if not configured.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db) status = vendor_email_settings_service.get_status(db, vendor_id)
status = service.get_status(vendor_id)
return EmailStatusResponse(**status) return EmailStatusResponse(**status)
@@ -181,13 +179,12 @@ def get_available_providers(
Returns list of providers with availability status. Returns list of providers with availability status.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db)
# Get vendor's current tier # Get vendor's current tier
tier = subscription_service.get_current_tier(db, vendor_id) tier = subscription_service.get_current_tier(db, vendor_id)
return ProvidersResponse( return ProvidersResponse(
providers=service.get_available_providers(tier), providers=vendor_email_settings_service.get_available_providers(tier),
current_tier=tier.value if tier else None, current_tier=tier.value if tier else None,
) )
@@ -206,13 +203,13 @@ def update_email_settings(
Raises ValidationException if data is invalid. Raises ValidationException if data is invalid.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db)
# Get vendor's current tier for validation # Get vendor's current tier for validation
tier = subscription_service.get_current_tier(db, vendor_id) tier = subscription_service.get_current_tier(db, vendor_id)
# Service raises appropriate exceptions (API-003 compliance) # Service raises appropriate exceptions (API-003 compliance)
settings = service.create_or_update( settings = vendor_email_settings_service.create_or_update(
db=db,
vendor_id=vendor_id, vendor_id=vendor_id,
data=data.model_dump(exclude_unset=True), data=data.model_dump(exclude_unset=True),
current_tier=tier, current_tier=tier,
@@ -240,10 +237,9 @@ def verify_email_settings(
Raises ValidationException if verification fails. Raises ValidationException if verification fails.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db)
# Service raises appropriate exceptions (API-003 compliance) # Service raises appropriate exceptions (API-003 compliance)
result = service.verify_settings(vendor_id, data.test_email) result = vendor_email_settings_service.verify_settings(db, vendor_id, data.test_email)
db.commit() db.commit()
return EmailVerifyResponse( return EmailVerifyResponse(
@@ -264,10 +260,9 @@ def delete_email_settings(
Raises ResourceNotFoundException if settings not found. Raises ResourceNotFoundException if settings not found.
""" """
vendor_id = current_user.token_vendor_id vendor_id = current_user.token_vendor_id
service = VendorEmailSettingsService(db)
# Service raises ResourceNotFoundException if not found (API-003 compliance) # Service raises ResourceNotFoundException if not found (API-003 compliance)
service.delete(vendor_id) vendor_email_settings_service.delete(db, vendor_id)
db.commit() db.commit()
return EmailDeleteResponse( return EmailDeleteResponse(

View File

@@ -650,4 +650,3 @@ ADD COLUMN is_custom BOOLEAN DEFAULT FALSE;
4. **Soft validation** - Invalid permissions in existing roles are not automatically removed (audit trail) 4. **Soft validation** - Invalid permissions in existing roles are not automatically removed (audit trail)
5. **Backward compatible** - Existing roles without `source_template_id` continue to work 5. **Backward compatible** - Existing roles without `source_template_id` continue to work
cl

View File

@@ -10,7 +10,7 @@ from app.exceptions import (
ResourceNotFoundException, ResourceNotFoundException,
ValidationException, ValidationException,
) )
from app.modules.cms.services.vendor_email_settings_service import VendorEmailSettingsService from app.modules.cms.services.vendor_email_settings_service import vendor_email_settings_service
from app.modules.messaging.models import VendorEmailSettings from app.modules.messaging.models import VendorEmailSettings
from app.modules.billing.models import TierCode from app.modules.billing.models import TierCode
@@ -78,8 +78,7 @@ class TestVendorEmailSettingsRead:
def test_get_settings_exists(self, db, test_email_settings): def test_get_settings_exists(self, db, test_email_settings):
"""Test getting settings when they exist.""" """Test getting settings when they exist."""
service = VendorEmailSettingsService(db) settings = vendor_email_settings_service.get_settings(db, test_email_settings.vendor_id)
settings = service.get_settings(test_email_settings.vendor_id)
assert settings is not None assert settings is not None
assert settings.from_email == "test@example.com" assert settings.from_email == "test@example.com"
@@ -87,46 +86,39 @@ class TestVendorEmailSettingsRead:
def test_get_settings_not_exists(self, db, test_vendor): def test_get_settings_not_exists(self, db, test_vendor):
"""Test getting settings when they don't exist.""" """Test getting settings when they don't exist."""
service = VendorEmailSettingsService(db) settings = vendor_email_settings_service.get_settings(db, test_vendor.id)
settings = service.get_settings(test_vendor.id)
assert settings is None assert settings is None
def test_get_settings_or_404_exists(self, db, test_email_settings): def test_get_settings_or_404_exists(self, db, test_email_settings):
"""Test get_settings_or_404 when settings exist.""" """Test get_settings_or_404 when settings exist."""
service = VendorEmailSettingsService(db) settings = vendor_email_settings_service.get_settings_or_404(db, test_email_settings.vendor_id)
settings = service.get_settings_or_404(test_email_settings.vendor_id)
assert settings is not None assert settings is not None
assert settings.id == test_email_settings.id assert settings.id == test_email_settings.id
def test_get_settings_or_404_not_exists(self, db, test_vendor): def test_get_settings_or_404_not_exists(self, db, test_vendor):
"""Test get_settings_or_404 raises exception when not found.""" """Test get_settings_or_404 raises exception when not found."""
service = VendorEmailSettingsService(db)
with pytest.raises(ResourceNotFoundException) as exc: with pytest.raises(ResourceNotFoundException) as exc:
service.get_settings_or_404(test_vendor.id) vendor_email_settings_service.get_settings_or_404(db, test_vendor.id)
assert "vendor_email_settings" in str(exc.value) assert "vendor_email_settings" in str(exc.value)
def test_is_configured_true(self, db, test_email_settings): def test_is_configured_true(self, db, test_email_settings):
"""Test is_configured returns True for configured settings.""" """Test is_configured returns True for configured settings."""
service = VendorEmailSettingsService(db) result = vendor_email_settings_service.is_configured(db, test_email_settings.vendor_id)
result = service.is_configured(test_email_settings.vendor_id)
assert result is True assert result is True
def test_is_configured_false_not_exists(self, db, test_vendor): def test_is_configured_false_not_exists(self, db, test_vendor):
"""Test is_configured returns False when settings don't exist.""" """Test is_configured returns False when settings don't exist."""
service = VendorEmailSettingsService(db) result = vendor_email_settings_service.is_configured(db, test_vendor.id)
result = service.is_configured(test_vendor.id)
assert result is False assert result is False
def test_get_status_configured(self, db, test_email_settings): def test_get_status_configured(self, db, test_email_settings):
"""Test get_status for configured settings.""" """Test get_status for configured settings."""
service = VendorEmailSettingsService(db) status = vendor_email_settings_service.get_status(db, test_email_settings.vendor_id)
status = service.get_status(test_email_settings.vendor_id)
assert status["is_configured"] is True assert status["is_configured"] is True
assert status["is_verified"] is False assert status["is_verified"] is False
@@ -135,8 +127,7 @@ class TestVendorEmailSettingsRead:
def test_get_status_not_configured(self, db, test_vendor): def test_get_status_not_configured(self, db, test_vendor):
"""Test get_status when settings don't exist.""" """Test get_status when settings don't exist."""
service = VendorEmailSettingsService(db) status = vendor_email_settings_service.get_status(db, test_vendor.id)
status = service.get_status(test_vendor.id)
assert status["is_configured"] is False assert status["is_configured"] is False
assert status["is_verified"] is False assert status["is_verified"] is False
@@ -155,8 +146,6 @@ class TestVendorEmailSettingsWrite:
def test_create_settings(self, db, test_vendor): def test_create_settings(self, db, test_vendor):
"""Test creating new email settings.""" """Test creating new email settings."""
service = VendorEmailSettingsService(db)
data = { data = {
"from_email": "new@example.com", "from_email": "new@example.com",
"from_name": "New Sender", "from_name": "New Sender",
@@ -167,7 +156,8 @@ class TestVendorEmailSettingsWrite:
"smtp_password": "pass", "smtp_password": "pass",
} }
settings = service.create_or_update( settings = vendor_email_settings_service.create_or_update(
db=db,
vendor_id=test_vendor.id, vendor_id=test_vendor.id,
data=data, data=data,
current_tier=TierCode.ESSENTIAL, current_tier=TierCode.ESSENTIAL,
@@ -179,14 +169,13 @@ class TestVendorEmailSettingsWrite:
def test_update_existing_settings(self, db, test_email_settings): def test_update_existing_settings(self, db, test_email_settings):
"""Test updating existing settings.""" """Test updating existing settings."""
service = VendorEmailSettingsService(db)
data = { data = {
"from_email": "updated@example.com", "from_email": "updated@example.com",
"from_name": "Updated Sender", "from_name": "Updated Sender",
} }
settings = service.create_or_update( settings = vendor_email_settings_service.create_or_update(
db=db,
vendor_id=test_email_settings.vendor_id, vendor_id=test_email_settings.vendor_id,
data=data, data=data,
current_tier=TierCode.ESSENTIAL, current_tier=TierCode.ESSENTIAL,
@@ -199,8 +188,6 @@ class TestVendorEmailSettingsWrite:
def test_premium_provider_requires_business_tier(self, db, test_vendor): def test_premium_provider_requires_business_tier(self, db, test_vendor):
"""Test that premium providers require Business tier.""" """Test that premium providers require Business tier."""
service = VendorEmailSettingsService(db)
data = { data = {
"from_email": "test@example.com", "from_email": "test@example.com",
"from_name": "Test", "from_name": "Test",
@@ -209,7 +196,8 @@ class TestVendorEmailSettingsWrite:
} }
with pytest.raises(AuthorizationException) as exc: with pytest.raises(AuthorizationException) as exc:
service.create_or_update( vendor_email_settings_service.create_or_update(
db=db,
vendor_id=test_vendor.id, vendor_id=test_vendor.id,
data=data, data=data,
current_tier=TierCode.ESSENTIAL, current_tier=TierCode.ESSENTIAL,
@@ -219,8 +207,6 @@ class TestVendorEmailSettingsWrite:
def test_premium_provider_allowed_for_business(self, db, test_vendor): def test_premium_provider_allowed_for_business(self, db, test_vendor):
"""Test that premium providers work with Business tier.""" """Test that premium providers work with Business tier."""
service = VendorEmailSettingsService(db)
data = { data = {
"from_email": "test@example.com", "from_email": "test@example.com",
"from_name": "Test", "from_name": "Test",
@@ -228,7 +214,8 @@ class TestVendorEmailSettingsWrite:
"sendgrid_api_key": "test-key", "sendgrid_api_key": "test-key",
} }
settings = service.create_or_update( settings = vendor_email_settings_service.create_or_update(
db=db,
vendor_id=test_vendor.id, vendor_id=test_vendor.id,
data=data, data=data,
current_tier=TierCode.BUSINESS, current_tier=TierCode.BUSINESS,
@@ -238,13 +225,12 @@ class TestVendorEmailSettingsWrite:
def test_provider_change_resets_verification(self, db, test_verified_email_settings): def test_provider_change_resets_verification(self, db, test_verified_email_settings):
"""Test that changing provider resets verification status.""" """Test that changing provider resets verification status."""
service = VendorEmailSettingsService(db)
assert test_verified_email_settings.is_verified is True assert test_verified_email_settings.is_verified is True
data = {"smtp_host": "new-smtp.example.com"} data = {"smtp_host": "new-smtp.example.com"}
settings = service.create_or_update( settings = vendor_email_settings_service.create_or_update(
db=db,
vendor_id=test_verified_email_settings.vendor_id, vendor_id=test_verified_email_settings.vendor_id,
data=data, data=data,
current_tier=TierCode.ESSENTIAL, current_tier=TierCode.ESSENTIAL,
@@ -254,22 +240,19 @@ class TestVendorEmailSettingsWrite:
def test_delete_settings(self, db, test_email_settings): def test_delete_settings(self, db, test_email_settings):
"""Test deleting email settings.""" """Test deleting email settings."""
service = VendorEmailSettingsService(db)
vendor_id = test_email_settings.vendor_id vendor_id = test_email_settings.vendor_id
service.delete(vendor_id) vendor_email_settings_service.delete(db, vendor_id)
db.commit() db.commit()
# Verify deletion # Verify deletion
settings = service.get_settings(vendor_id) settings = vendor_email_settings_service.get_settings(db, vendor_id)
assert settings is None assert settings is None
def test_delete_settings_not_found(self, db, test_vendor): def test_delete_settings_not_found(self, db, test_vendor):
"""Test deleting non-existent settings raises exception.""" """Test deleting non-existent settings raises exception."""
service = VendorEmailSettingsService(db)
with pytest.raises(ResourceNotFoundException): with pytest.raises(ResourceNotFoundException):
service.delete(test_vendor.id) vendor_email_settings_service.delete(db, test_vendor.id)
# ============================================================================= # =============================================================================
@@ -284,10 +267,8 @@ class TestVendorEmailSettingsVerification:
def test_verify_settings_not_configured(self, db, test_vendor): def test_verify_settings_not_configured(self, db, test_vendor):
"""Test verification fails for non-existent settings.""" """Test verification fails for non-existent settings."""
service = VendorEmailSettingsService(db)
with pytest.raises(ResourceNotFoundException): with pytest.raises(ResourceNotFoundException):
service.verify_settings(test_vendor.id, "test@example.com") vendor_email_settings_service.verify_settings(db, test_vendor.id, "test@example.com")
def test_verify_settings_incomplete(self, db, test_vendor): def test_verify_settings_incomplete(self, db, test_vendor):
"""Test verification fails for incomplete settings.""" """Test verification fails for incomplete settings."""
@@ -303,10 +284,8 @@ class TestVendorEmailSettingsVerification:
db.add(settings) db.add(settings)
db.commit() db.commit()
service = VendorEmailSettingsService(db)
with pytest.raises(ValidationException) as exc: with pytest.raises(ValidationException) as exc:
service.verify_settings(test_vendor.id, "test@example.com") vendor_email_settings_service.verify_settings(db, test_vendor.id, "test@example.com")
assert "incomplete" in str(exc.value).lower() assert "incomplete" in str(exc.value).lower()
@@ -317,8 +296,8 @@ class TestVendorEmailSettingsVerification:
mock_server = MagicMock() mock_server = MagicMock()
mock_smtp.return_value = mock_server mock_smtp.return_value = mock_server
service = VendorEmailSettingsService(db) result = vendor_email_settings_service.verify_settings(
result = service.verify_settings( db,
test_email_settings.vendor_id, test_email_settings.vendor_id,
"recipient@example.com", "recipient@example.com",
) )
@@ -332,8 +311,8 @@ class TestVendorEmailSettingsVerification:
# Mock SMTP error # Mock SMTP error
mock_smtp.side_effect = Exception("Connection refused") mock_smtp.side_effect = Exception("Connection refused")
service = VendorEmailSettingsService(db) result = vendor_email_settings_service.verify_settings(
result = service.verify_settings( db,
test_email_settings.vendor_id, test_email_settings.vendor_id,
"recipient@example.com", "recipient@example.com",
) )
@@ -354,8 +333,7 @@ class TestVendorEmailProvidersAvailability:
def test_get_providers_essential_tier(self, db): def test_get_providers_essential_tier(self, db):
"""Test available providers for Essential tier.""" """Test available providers for Essential tier."""
service = VendorEmailSettingsService(db) providers = vendor_email_settings_service.get_available_providers(TierCode.ESSENTIAL)
providers = service.get_available_providers(TierCode.ESSENTIAL)
# Find SMTP provider # Find SMTP provider
smtp = next((p for p in providers if p["code"] == "smtp"), None) smtp = next((p for p in providers if p["code"] == "smtp"), None)
@@ -369,8 +347,7 @@ class TestVendorEmailProvidersAvailability:
def test_get_providers_business_tier(self, db): def test_get_providers_business_tier(self, db):
"""Test available providers for Business tier.""" """Test available providers for Business tier."""
service = VendorEmailSettingsService(db) providers = vendor_email_settings_service.get_available_providers(TierCode.BUSINESS)
providers = service.get_available_providers(TierCode.BUSINESS)
# All providers should be available # All providers should be available
for provider in providers: for provider in providers:
@@ -378,8 +355,7 @@ class TestVendorEmailProvidersAvailability:
def test_get_providers_no_tier(self, db): def test_get_providers_no_tier(self, db):
"""Test available providers with no subscription.""" """Test available providers with no subscription."""
service = VendorEmailSettingsService(db) providers = vendor_email_settings_service.get_available_providers(None)
providers = service.get_available_providers(None)
# Only SMTP should be available # Only SMTP should be available
smtp = next((p for p in providers if p["code"] == "smtp"), None) smtp = next((p for p in providers if p["code"] == "smtp"), None)