def test_get_message(self):
model = Mock()
model.a = 1
model.b = 2
model.c = 3
ml = ModelListener(model, fields=['a','b'])
mess = ml.get_message(model)
assert mess['a'] == 1
assert mess['b'] == 2
assert not 'c' in mess
def test_setup(self):
model = Mock()
signal = Mock()
ml = ModelListener(model, signal)
ml.get_message = Mock()
publish = Mock()
ml.setup(publish)
assert not publish.called
assert not ml.get_message.called
assert signal.connect.called
handler = signal.connect.call_args[0][0]
handler()
assert publish.called
assert ml.get_message.called