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': []})