def test_labeleditems_addin(self):
"""
More tests
"""
circuit = Box2Circuit(name='TestCircuit')
log = Box2Log(pickup_line='TestPickupLine', circuit=circuit)
# column_label_transforms is weird. I should fix this.
mv = DocModelView(Box2Log,
include_columns=['circuit'],
column_label_transforms={'circuit': lambda k: 'Circuit'},
column_value_transforms={'circuit': lambda v: v.name if v else None})
assert mv.labeleditems(log)['Pickup Line'] == 'TestPickupLine'
print mv.labeleditems(log)
assert mv.labeleditems(log)['Circuit'] == 'TestCircuit'
class TestDocModelView(DatabaseTest):
def __init__(self, *args, **kwargs):
self.logview = DocModelView(Box2Log)
super(TestDocModelView, self).__init__(*args, **kwargs)
@property
def logobj(self):
# uncommitted item
return Box2Log(pickup_line='Sample Pickup Line')
@property
def logobj_joined(self):
return Session.query(Box2Log)\
.options(joinedload_all(Box2Log.circuit))\
.first()
def test_labeleditems_basic(self):
"""
Test stuff.
"""
logobj = self.logobj
labeleditems = self.logview.labeleditems(logobj)
assert labeleditems['Pickup Line'] == 'Sample Pickup Line'
assert labeleditems['Waste Bottle Full Value'] is None
def test_labeleditems_addin(self):
"""
More tests
"""
circuit = Box2Circuit(name='TestCircuit')
log = Box2Log(pickup_line='TestPickupLine', circuit=circuit)
# column_label_transforms is weird. I should fix this.
mv = DocModelView(Box2Log,
include_columns=['circuit'],
column_label_transforms={'circuit': lambda k: 'Circuit'},
column_value_transforms={'circuit': lambda v: v.name if v else None})
assert mv.labeleditems(log)['Pickup Line'] == 'TestPickupLine'
print mv.labeleditems(log)
assert mv.labeleditems(log)['Circuit'] == 'TestCircuit'
def __init__(self, *args, **kwargs): self.logview = DocModelView(Box2Log) super(TestDocModelView, self).__init__(*args, **kwargs)
