3
0
Fork 0

fix: add make_proxy() convenience method for data model Base

This commit is contained in:
Lance Edgar 2025-01-25 16:26:14 -06:00
parent 2289928337
commit 4a6897c6de
2 changed files with 109 additions and 2 deletions

View file

@ -3,12 +3,34 @@
from unittest import TestCase
try:
import sqlalchemy as sa
from sqlalchemy import orm
from wuttjamaican.db.model import base as mod
from wuttjamaican.db.model.auth import User
except ImportError:
pass
else:
class MockUser(mod.Base):
__tablename__ = 'mock_user'
uuid = mod.uuid_column(sa.ForeignKey('user.uuid'), default=False)
user = orm.relationship(
User,
backref=orm.backref('_mock', uselist=False, cascade='all, delete-orphan'))
favorite_color = sa.Column(sa.String(length=100), nullable=False)
class TestWuttaModelBase(TestCase):
def test_make_proxy(self):
self.assertFalse(hasattr(User, 'favorite_color'))
MockUser.make_proxy(User, '_mock', 'favorite_color')
self.assertTrue(hasattr(User, 'favorite_color'))
user = User(favorite_color='green')
self.assertEqual(user.favorite_color, 'green')
class TestSetting(TestCase):
def test_basic(self):
@ -17,6 +39,7 @@ else:
setting.name = 'foo'
self.assertEqual(str(setting), "foo")
class TestPerson(TestCase):
def test_basic(self):