def test_missing_required_attributes_attribute(self):
"""Test decorator with missing attribute _required_attributes."""
function = mock.Mock(__name__='fake')
slf = mock.Mock(spec=[])
wrapper = common.is_complete(function)
self.assertRaises(Exception, wrapper, slf, 1, entry=2)
self.assertFalse(function.called)
def test_required_attributes_is_empty(self):
"""Test decorator with empty attribute _required_attributes."""
function = mock.Mock(__name__='fake',
return_value=mock.sentinel.return_value)
slf = mock.Mock(spec=[], _required_attributes=[])
wrapper = common.is_complete(function)
self.assertEqual(mock.sentinel.return_value, wrapper(slf, 1, entry=2))
function.assert_called_once_with(slf, 1, entry=2)
def test_missing_attribute(self):
"""Test decorator when missing required attribute.."""
function = mock.Mock(__name__='fake',
return_value=mock.sentinel.return_value)
slf = mock.Mock(spec=['attr1'],
_required_attributes=['attr1', 'attr2'])
wrapper = common.is_complete(function)
self.assertRaises(Exception, wrapper, slf, 1, entry=2)
self.assertFalse(function.called)
