Step 1: Add PricingConfig model and PricingRepository
- Create PricingConfig model with all required fields (premium settings, trade limits) - Implement PricingRepository with singleton pattern (get_current, create_or_update) - Add comprehensive tests for repository functionality - Export model and repository in __init__.py files
This commit is contained in:
parent
82c4d0168e
commit
32ce27180d
5 changed files with 291 additions and 0 deletions
160
backend/tests/test_pricing_repository.py
Normal file
160
backend/tests/test_pricing_repository.py
Normal file
|
|
@ -0,0 +1,160 @@
|
|||
"""
|
||||
Pricing Repository Tests
|
||||
|
||||
Tests for the pricing configuration repository.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
|
||||
from models import PricingConfig
|
||||
from repositories.pricing import PricingRepository
|
||||
|
||||
|
||||
class TestPricingRepository:
|
||||
"""Test the PricingRepository class."""
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_current_returns_none_when_no_config(self, client_factory):
|
||||
"""get_current() returns None when no config exists."""
|
||||
async with client_factory.get_db_session() as db:
|
||||
repo = PricingRepository(db)
|
||||
config = await repo.get_current()
|
||||
assert config is None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_or_update_creates_first_config(self, client_factory):
|
||||
"""create_or_update() creates a new config when none exists."""
|
||||
async with client_factory.get_db_session() as db:
|
||||
repo = PricingRepository(db)
|
||||
config = await repo.create_or_update(
|
||||
premium_buy=5,
|
||||
premium_sell=5,
|
||||
small_trade_threshold_eur=20000, # €200
|
||||
small_trade_extra_premium=2,
|
||||
eur_min_buy=10000, # €100
|
||||
eur_max_buy=300000, # €3000
|
||||
eur_min_sell=10000,
|
||||
eur_max_sell=300000,
|
||||
)
|
||||
|
||||
assert config is not None
|
||||
assert config.id is not None
|
||||
assert config.premium_buy == 5
|
||||
assert config.premium_sell == 5
|
||||
assert config.small_trade_threshold_eur == 20000
|
||||
assert config.small_trade_extra_premium == 2
|
||||
assert config.eur_min_buy == 10000
|
||||
assert config.eur_max_buy == 300000
|
||||
assert config.eur_min_sell == 10000
|
||||
assert config.eur_max_sell == 300000
|
||||
assert config.created_at is not None
|
||||
assert config.updated_at is not None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_or_update_updates_existing_config(self, client_factory):
|
||||
"""create_or_update() updates existing config (singleton behavior)."""
|
||||
async with client_factory.get_db_session() as db:
|
||||
repo = PricingRepository(db)
|
||||
|
||||
# Create initial config
|
||||
config1 = await repo.create_or_update(
|
||||
premium_buy=5,
|
||||
premium_sell=5,
|
||||
small_trade_threshold_eur=20000,
|
||||
small_trade_extra_premium=2,
|
||||
eur_min_buy=10000,
|
||||
eur_max_buy=300000,
|
||||
eur_min_sell=10000,
|
||||
eur_max_sell=300000,
|
||||
)
|
||||
original_id = config1.id
|
||||
original_created_at = config1.created_at
|
||||
|
||||
# Update config
|
||||
config2 = await repo.create_or_update(
|
||||
premium_buy=7,
|
||||
premium_sell=6,
|
||||
small_trade_threshold_eur=25000,
|
||||
small_trade_extra_premium=3,
|
||||
eur_min_buy=15000,
|
||||
eur_max_buy=350000,
|
||||
eur_min_sell=12000,
|
||||
eur_max_sell=320000,
|
||||
)
|
||||
|
||||
# Should be the same record (singleton)
|
||||
assert config2.id == original_id
|
||||
assert config2.created_at == original_created_at
|
||||
assert config2.premium_buy == 7
|
||||
assert config2.premium_sell == 6
|
||||
assert config2.small_trade_threshold_eur == 25000
|
||||
assert config2.small_trade_extra_premium == 3
|
||||
assert config2.eur_min_buy == 15000
|
||||
assert config2.eur_max_buy == 350000
|
||||
assert config2.eur_min_sell == 12000
|
||||
assert config2.eur_max_sell == 320000
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_current_returns_existing_config(self, client_factory):
|
||||
"""get_current() returns existing config."""
|
||||
async with client_factory.get_db_session() as db:
|
||||
repo = PricingRepository(db)
|
||||
|
||||
# Create config
|
||||
created_config = await repo.create_or_update(
|
||||
premium_buy=5,
|
||||
premium_sell=5,
|
||||
small_trade_threshold_eur=20000,
|
||||
small_trade_extra_premium=2,
|
||||
eur_min_buy=10000,
|
||||
eur_max_buy=300000,
|
||||
eur_min_sell=10000,
|
||||
eur_max_sell=300000,
|
||||
)
|
||||
|
||||
# Retrieve config
|
||||
retrieved_config = await repo.get_current()
|
||||
|
||||
assert retrieved_config is not None
|
||||
assert retrieved_config.id == created_config.id
|
||||
assert retrieved_config.premium_buy == 5
|
||||
assert retrieved_config.premium_sell == 5
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_singleton_pattern_only_one_config_exists(self, client_factory):
|
||||
"""Only one config record can exist (singleton pattern)."""
|
||||
async with client_factory.get_db_session() as db:
|
||||
repo = PricingRepository(db)
|
||||
|
||||
# Create config multiple times
|
||||
config1 = await repo.create_or_update(
|
||||
premium_buy=5,
|
||||
premium_sell=5,
|
||||
small_trade_threshold_eur=20000,
|
||||
small_trade_extra_premium=2,
|
||||
eur_min_buy=10000,
|
||||
eur_max_buy=300000,
|
||||
eur_min_sell=10000,
|
||||
eur_max_sell=300000,
|
||||
)
|
||||
|
||||
config2 = await repo.create_or_update(
|
||||
premium_buy=10,
|
||||
premium_sell=10,
|
||||
small_trade_threshold_eur=30000,
|
||||
small_trade_extra_premium=5,
|
||||
eur_min_buy=20000,
|
||||
eur_max_buy=400000,
|
||||
eur_min_sell=20000,
|
||||
eur_max_sell=400000,
|
||||
)
|
||||
|
||||
# Should be same record
|
||||
assert config1.id == config2.id
|
||||
|
||||
# Verify only one record exists
|
||||
from sqlalchemy import select
|
||||
|
||||
result = await db.execute(select(PricingConfig))
|
||||
all_configs = list(result.scalars().all())
|
||||
assert len(all_configs) == 1
|
||||
Loading…
Add table
Add a link
Reference in a new issue