def test_add_method_registers(self): b = Block("blockname") m = MagicMock() m.name = "mymethod" b.add_method(m) self.assertEqual(b._methods.keys(), ["mymethod"]) self.assertFalse(m.called) m.return_value = 42 self.assertEqual(b.mymethod(), 42) m.assert_called_once_with()
def test_add_method_registers(self): b = Block() b.on_changed = MagicMock(side_effect=b.on_changed) m = MagicMock() b.add_method("mymethod", m) self.assertEqual(list(b.methods), ["mymethod"]) self.assertFalse(m.called) b.on_changed.assert_called_with([[m.name], m.to_dict.return_value], True) m.return_value = 42 self.assertEqual(b.mymethod(), 42) m.assert_called_once_with()