Source code for sqlobject.inheritance.tests.test_asdict

fromsqlobjectimport StringCol
fromsqlobject.inheritanceimport InheritableSQLObject
fromsqlobject.tests.dbtestimport setupClass
########################################
# sqlmeta.asDict
########################################

[docs]
classInheritablePersonAD(InheritableSQLObject):
 firstName = StringCol()
 lastName = StringCol(alternateID=True, length=100)

[docs]
classManagerAD(InheritablePersonAD):
 department = StringCol()

[docs]
classEmployeeAD(InheritablePersonAD):
 _inheritable = False
 so_position = StringCol()

[docs]
deftest_getColumns():
 setupClass([InheritablePersonAD, ManagerAD, EmployeeAD])
 for klass, columns in (
 (InheritablePersonAD, ['firstName', 'lastName']),
 (ManagerAD, ['department', 'firstName', 'lastName']),
 (EmployeeAD, ['firstName', 'lastName', 'so_position'])):
 _columns = sorted(klass.sqlmeta.getColumns().keys())
 assert _columns == columns

[docs]
deftest_asDict():
 setupClass([InheritablePersonAD, ManagerAD, EmployeeAD], force=True)
 InheritablePersonAD(firstName='Oneof', lastName='Authors')
 ManagerAD(firstName='ManagerAD', lastName='The', department='Dep')
 EmployeeAD(firstName='Project', lastName='Leader',
 so_position='Project leader')
 assert InheritablePersonAD.get(1).sqlmeta.asDict() == \
 dict(firstName='Oneof', lastName='Authors', id=1)
 assert InheritablePersonAD.get(2).sqlmeta.asDict() == \
 dict(firstName='ManagerAD', lastName='The', department='Dep', id=2)
 assert InheritablePersonAD.get(3).sqlmeta.asDict() == \
 dict(firstName='Project', lastName='Leader',
 so_position='Project leader', id=3)