Source code for sqlobject.tests.test_style

from sqlobject import ForeignKey, SQLObject, StringCol, sqlmeta, styles
from sqlobject.tests.dbtest import setupClass


[docs]class AnotherStyle(styles.MixedCaseUnderscoreStyle):
[docs] def pythonAttrToDBColumn(self, attr): if attr.lower().endswith('id'): return 'id' + styles.MixedCaseUnderscoreStyle.\ pythonAttrToDBColumn(self, attr[:-2]) else: return styles.MixedCaseUnderscoreStyle.\ pythonAttrToDBColumn(self, attr)
[docs]class SOStyleTest1(SQLObject): a = StringCol() st2 = ForeignKey('SOStyleTest2')
[docs] class sqlmeta(sqlmeta): style = AnotherStyle()
[docs]class SOStyleTest2(SQLObject): b = StringCol()
[docs] class sqlmeta(sqlmeta): style = AnotherStyle()
[docs]def test_style(): setupClass([SOStyleTest2, SOStyleTest1]) st1 = SOStyleTest1(a='something', st2=None) st2 = SOStyleTest2(b='whatever') st1.st2 = st2 assert st1.sqlmeta.columns['st2ID'].dbName == 'idst2' assert st1.st2 == st2