shanxia2.0/backend/tests/test_db_inventory_id.py

"""出入库单 id 生成单元测试"""
from unittest.mock import MagicMock

from app.services.db.inventory_id import max_numeric_inbound_outbound_id


def test_max_numeric_inbound_outbound_id_parses_scalar():
    db = MagicMock()
    db.execute.return_value.scalar.return_value = 1094
    assert max_numeric_inbound_outbound_id(db, "shanxia2.0") == 1094
    db.execute.assert_called_once()


def test_max_numeric_inbound_outbound_id_empty():
    db = MagicMock()
    db.execute.return_value.scalar.return_value = None
    assert max_numeric_inbound_outbound_id(db, "shanxia2.0") == 0