def test_replace_attrs(self):
"""Ensure replace_attrs behaves correctly"""
class MockValidator(Validator):
def validate_object(self, obj, write=True):
pass
def validate_modify(self, dn, modlist, current):
pass
def _validate_attribute(self, attr_name, values, write):
pass
mock_sock = MockLDAPSocket()
mock_sock.add_root_dse()
ldap = LDAP(mock_sock)
mock_sock.clear_sent()
# current is None and have validators and not strict_modify
# SHOULD perform a search before modify
ldap.validators = [MockValidator()]
ldap.strict_modify = False
mock_sock.add_search_res_entry('', {'foo': ['bar']})
mock_sock.add_search_res_done('')
mock_sock.add_ldap_result(rfc4511.ModifyResponse, 'modifyResponse')
ldap.strict_modify = False
ldap.replace_attrs('', {'foo': ['bar']})
self.assertEqual(mock_sock.num_sent(), 2)
protoutils.unpack('searchRequest', mock_sock.read_sent())
protoutils.unpack('modifyRequest', mock_sock.read_sent())
# else
# SHOULD NOT perform a search, only modify
ldap.validators = []
mock_sock.add_ldap_result(rfc4511.ModifyResponse, 'modifyResponse')
ldap.replace_attrs('', {'foo': ['bar']})
self.assertEqual(mock_sock.num_sent(), 1)
protoutils.unpack('modifyRequest', mock_sock.read_sent())
def test_error_empty_list(self):
"""Ensure the error_empty_list option is respected with all invocations"""
mock_sock = MockLDAPSocket()
mock_sock.add_root_dse()
ldap = LDAP(mock_sock, error_empty_list=True)
with self.assertRaises(exceptions.LDAPError):
ldap.modify('o=foo', [Mod(Mod.REPLACE, 'foo', [])])
with self.assertRaises(exceptions.LDAPError):
ldap.replace_attrs('o=foo', {'foo': []})
with self.assertRaises(exceptions.LDAPError):
ldap.delete_attrs('o=foo', {'foo': []}, current={'foo': ['bar']})
with self.assertRaises(exceptions.LDAPError):
ldap.obj('o=foo').modify([Mod(Mod.REPLACE, 'foo', [])])
with self.assertRaises(exceptions.LDAPError):
ldap.obj('o=foo').replace_attrs({'foo': []})
with self.assertRaises(exceptions.LDAPError):
ldap.obj('o=foo', attrs_dict={'foo': ['bar']}).delete_attrs({'foo': []})
