7cb0b30dae
- Replace `import logging` with `from aitbc.logging import get_logger` across blockchain-node scripts and coordinator-api modules - Update logger initialization from `logging.getLogger(__name__)` to `get_logger(__name__)` in 30+ files - Add production configuration validators for API keys, HMAC secret, and JWT secret in coordinator config - Enhance coordinator startup with comprehensive initialization logging
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import pytest
|
|
from sqlmodel import Session, delete, text
|
|
|
|
from app.domain import Job, Miner
|
|
from app.models import JobCreate
|
|
from app.services.jobs import JobService
|
|
from app.storage.db import init_db, session_scope
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def _init_db(tmp_path_factory):
|
|
db_file = tmp_path_factory.mktemp("data") / "test.db"
|
|
# override settings dynamically
|
|
from app.config import settings
|
|
|
|
settings.database_url = f"sqlite:///{db_file}"
|
|
|
|
# Initialize database and create tables
|
|
init_db()
|
|
|
|
# Ensure payment_id column exists (handle schema migration)
|
|
with session_scope() as sess:
|
|
try:
|
|
# Check if columns exist and add them if needed
|
|
result = sess.exec(text("PRAGMA table_info(job)"))
|
|
columns = [row[1] for row in result.fetchall()]
|
|
|
|
if 'payment_id' not in columns:
|
|
sess.exec(text("ALTER TABLE job ADD COLUMN payment_id TEXT"))
|
|
if 'payment_status' not in columns:
|
|
sess.exec(text("ALTER TABLE job ADD COLUMN payment_status TEXT"))
|
|
sess.commit()
|
|
except Exception as e:
|
|
print(f"Schema migration error: {e}")
|
|
sess.rollback()
|
|
|
|
yield
|
|
|
|
|
|
@pytest.fixture()
|
|
def session():
|
|
with session_scope() as sess:
|
|
from sqlmodel import select
|
|
# Clear all data
|
|
sess.query(Job).delete()
|
|
sess.query(Miner).delete()
|
|
sess.commit()
|
|
yield sess
|
|
|
|
|
|
def test_create_and_fetch_job(session: Session):
|
|
svc = JobService(session)
|
|
job = svc.create_job("client1", JobCreate(payload={"task": "noop"}))
|
|
fetched = svc.get_job(job.id, client_id="client1")
|
|
assert fetched.id == job.id
|
|
assert fetched.payload["task"] == "noop"
|
|
|
|
|
|
def test_acquire_next_job(session: Session):
|
|
svc = JobService(session)
|
|
job1 = svc.create_job("client1", JobCreate(payload={"n": 1}))
|
|
job2 = svc.create_job("client1", JobCreate(payload={"n": 2}))
|
|
|
|
miner = Miner(id="miner1", capabilities={}, concurrency=1)
|
|
session.add(miner)
|
|
session.commit()
|
|
|
|
next_job = svc.acquire_next_job(miner)
|
|
assert next_job is not None
|
|
assert next_job.id == job1.id
|
|
assert next_job.state == "RUNNING"
|
|
|
|
next_job2 = svc.acquire_next_job(miner)
|
|
assert next_job2 is not None
|
|
assert next_job2.id == job2.id
|
|
|
|
# No more jobs
|
|
assert svc.acquire_next_job(miner) is None
|