def test_setAttribute_attr_value_is_not_None(self):
"""Tests setAttribute when attr_value is not None"""
_node = MagicMock()
_element = Element(_node)
_result = _element.setAttribute('some', 2)
self.assertIsNone(_result)
_node.setAttribute.assert_called_once_with('some', '2')
_node.removeAttribute.assert_not_called()
_node.hasAttribute.assert_not_called()
def test_setAttribute_attr_value_is_None_remove_attribute(self):
"""Tests setAttribute when attr_value is None, removeAttribute called"""
_node = MagicMock()
_node.hasAttribute.return_value = True
_element = Element(_node)
_result = _element.setAttribute('some')
self.assertIsNone(_result)
_node.setAttribute.assert_not_called()
_node.removeAttribute.assert_called_once_with('some')
_node.hasAttribute.assert_called_once_with('some')