def test__can_watch_fields(self): connect_to_field_change = self.patch_autospec( signals_module, "connect_to_field_change") connect_to_field_change.return_value = ( sentinel.connect, sentinel.disconnect, ) manager = SignalsManager() manager.watch_fields(sentinel.callback, sentinel.model, sentinel.fields, sentinel.delete) self.assertThat( manager._signals, Equals({Signal(sentinel.connect, sentinel.disconnect)}), ) self.assertThat( connect_to_field_change, MockCalledOnceWith( sentinel.callback, sentinel.model, sentinel.fields, sentinel.delete, ), )
def make_Signal(self): return Signal(Mock(name="connect"), Mock(name="disconnect"))