refactor: split database model tests into organized modules
Split the monolithic test_database_models.py into focused test modules: Database model tests (tests/unit/models/database/): - test_customer.py: Customer model and authentication tests - test_inventory.py: Inventory model tests - test_marketplace_import_job.py: Import job model tests - test_marketplace_product.py: Marketplace product model tests - test_order.py: Order and OrderItem model tests - test_product.py: Product model tests - test_team.py: Team invitation and membership tests - test_user.py: User model tests - test_vendor.py: Vendor model tests Schema validation tests (tests/unit/models/schema/): - test_auth.py: Auth schema validation tests - test_customer.py: Customer schema validation tests - test_inventory.py: Inventory schema validation tests - test_marketplace_import_job.py: Import job schema tests - test_order.py: Order schema validation tests - test_product.py: Product schema validation tests This improves test organization and makes it easier to find and maintain tests for specific models. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
179
tests/unit/models/database/test_team.py
Normal file
179
tests/unit/models/database/test_team.py
Normal file
@@ -0,0 +1,179 @@
|
||||
# tests/unit/models/database/test_team.py
|
||||
"""Unit tests for VendorUser and Role database models."""
|
||||
import pytest
|
||||
|
||||
from models.database.vendor import Role, Vendor, VendorUser
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.database
|
||||
class TestRoleModel:
|
||||
"""Test Role model."""
|
||||
|
||||
def test_role_creation(self, db, test_vendor):
|
||||
"""Test Role model creation."""
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Manager",
|
||||
permissions=["products.create", "orders.view"],
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
db.refresh(role)
|
||||
|
||||
assert role.id is not None
|
||||
assert role.vendor_id == test_vendor.id
|
||||
assert role.name == "Manager"
|
||||
assert "products.create" in role.permissions
|
||||
assert "orders.view" in role.permissions
|
||||
|
||||
def test_role_default_permissions(self, db, test_vendor):
|
||||
"""Test Role model with default empty permissions."""
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Viewer",
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
db.refresh(role)
|
||||
|
||||
assert role.permissions == [] or role.permissions is None
|
||||
|
||||
def test_role_vendor_relationship(self, db, test_vendor):
|
||||
"""Test Role-Vendor relationship."""
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Admin",
|
||||
permissions=["*"],
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
db.refresh(role)
|
||||
|
||||
assert role.vendor is not None
|
||||
assert role.vendor.id == test_vendor.id
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.database
|
||||
class TestVendorUserModel:
|
||||
"""Test VendorUser model."""
|
||||
|
||||
def test_vendor_user_creation(self, db, test_vendor, test_user):
|
||||
"""Test VendorUser model for team management."""
|
||||
# Create a role
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Manager",
|
||||
permissions=["products.create", "orders.view"],
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
|
||||
# Create vendor user
|
||||
vendor_user = VendorUser(
|
||||
vendor_id=test_vendor.id,
|
||||
user_id=test_user.id,
|
||||
role_id=role.id,
|
||||
is_active=True,
|
||||
)
|
||||
db.add(vendor_user)
|
||||
db.commit()
|
||||
db.refresh(vendor_user)
|
||||
|
||||
assert vendor_user.id is not None
|
||||
assert vendor_user.vendor_id == test_vendor.id
|
||||
assert vendor_user.user_id == test_user.id
|
||||
assert vendor_user.role.name == "Manager"
|
||||
assert "products.create" in vendor_user.role.permissions
|
||||
|
||||
def test_vendor_user_multiple_vendors(self, db, test_vendor, test_user, other_company):
|
||||
"""Test same user can be added to multiple vendors."""
|
||||
# Create another vendor
|
||||
other_vendor = Vendor(
|
||||
company_id=other_company.id,
|
||||
vendor_code="OTHER_VENDOR",
|
||||
subdomain="othervendor",
|
||||
name="Other Vendor",
|
||||
)
|
||||
db.add(other_vendor)
|
||||
db.commit()
|
||||
|
||||
role1 = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Editor1",
|
||||
permissions=["products.view"],
|
||||
)
|
||||
role2 = Role(
|
||||
vendor_id=other_vendor.id,
|
||||
name="Editor2",
|
||||
permissions=["products.view"],
|
||||
)
|
||||
db.add_all([role1, role2])
|
||||
db.commit()
|
||||
|
||||
# Same user can be added to different vendors
|
||||
vendor_user1 = VendorUser(
|
||||
vendor_id=test_vendor.id,
|
||||
user_id=test_user.id,
|
||||
role_id=role1.id,
|
||||
)
|
||||
vendor_user2 = VendorUser(
|
||||
vendor_id=other_vendor.id,
|
||||
user_id=test_user.id,
|
||||
role_id=role2.id,
|
||||
)
|
||||
db.add_all([vendor_user1, vendor_user2])
|
||||
db.commit()
|
||||
|
||||
assert vendor_user1.vendor_id != vendor_user2.vendor_id
|
||||
assert vendor_user1.user_id == vendor_user2.user_id
|
||||
|
||||
def test_vendor_user_relationships(self, db, test_vendor, test_user):
|
||||
"""Test VendorUser relationships."""
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Staff",
|
||||
permissions=["orders.view"],
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
|
||||
vendor_user = VendorUser(
|
||||
vendor_id=test_vendor.id,
|
||||
user_id=test_user.id,
|
||||
role_id=role.id,
|
||||
is_active=True,
|
||||
)
|
||||
db.add(vendor_user)
|
||||
db.commit()
|
||||
db.refresh(vendor_user)
|
||||
|
||||
assert vendor_user.vendor is not None
|
||||
assert vendor_user.user is not None
|
||||
assert vendor_user.role is not None
|
||||
assert vendor_user.vendor.vendor_code == test_vendor.vendor_code
|
||||
assert vendor_user.user.email == test_user.email
|
||||
|
||||
def test_vendor_user_with_active_flag(self, db, test_vendor, test_user):
|
||||
"""Test VendorUser is_active field."""
|
||||
role = Role(
|
||||
vendor_id=test_vendor.id,
|
||||
name="Default",
|
||||
permissions=[],
|
||||
)
|
||||
db.add(role)
|
||||
db.commit()
|
||||
|
||||
# Create with explicit is_active=True
|
||||
vendor_user = VendorUser(
|
||||
vendor_id=test_vendor.id,
|
||||
user_id=test_user.id,
|
||||
role_id=role.id,
|
||||
is_active=True,
|
||||
)
|
||||
db.add(vendor_user)
|
||||
db.commit()
|
||||
db.refresh(vendor_user)
|
||||
|
||||
assert vendor_user.is_active is True
|
||||
Reference in New Issue
Block a user