Fix problem with missing LDAP batching cookie in search

conestack · Apr 16, 2019 · a97e1b8 · a97e1b8
1 parent fcf2716
commit a97e1b8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,6 +4,9 @@ History
 1.0b9 (unreleased)
 ------------------
 
+- Fix problem with missing LDAP batching cookie in search.
+  [jensens]
+
 - Remove ``smbpasswd`` dependency. Use ``passlib`` instead.
   [rnix]
 

diff --git a/src/node/ext/ldap/ugm/_api.py b/src/node/ext/ldap/ugm/_api.py
@@ -606,7 +606,7 @@ def raw_search(self, criteria=None, attrlist=None,
             )
         except ldap.NO_SUCH_OBJECT:  # pragma: no cover
             return []
-        if type(results) is tuple:
+        if isinstance(results, tuple):
             results, cookie = results
         if attrlist is not None:
             _results = list()
@@ -630,14 +630,18 @@ def search(self, criteria=None, attrlist=None,
         result = []
         cookie = None
         while True:
-            chunk, cookie = self.raw_search(
+            chunk_cookie = self.raw_search(
                 criteria=criteria,
                 attrlist=attrlist,
                 exact_match=exact_match,
                 or_search=or_search,
                 page_size=self.context.ldap_session._props.page_size,
                 cookie=cookie
             )
+            if isinstance(chunk_cookie, tuple):
+                chunk, cookie = chunk_cookie
+            else:
+                chunk, cookie = chunk_cookie, None
             result += chunk
             if not cookie:
                 break