def test_operation(self):
instance = UMLObject()
operation = Mock()
instance.add_operation(operation)
result = instance.operations()[0]
self.assertEquals(result, operation)
def test_not_eq_operation(self):
instance = UMLObject()
other = UMLObject()
(o1, o2) = (Mock(), Mock())
instance.add_operation(o1)
instance.add_operation(o2)
self.assertNotEqual(instance, other)
def test_eq_operation(self):
instance = UMLObject()
other = UMLObject()
o1 = MagicMock()
o1.__eq__.return_value = True
o2 = MagicMock()
o2.__eq__.return_value = True
instance.add_operation(o1)
other.add_operation(o2)
self.assertEqual(instance, other)
