def test_add_bind_multiple_with_result(self):
rest = 'BIND dn="uid=bind-generateusers,ou=logins,dc=example" method=128'
rest2 = 'BIND dn="uid=bind-generateusers,ou=logins,dc=example" mech=SIMPLE ssf=0'
rest3 = 'RESULT tag=97 err=0 text='
operation = Operation(0)
operation.add_event(rest)
operation.add_event(rest2)
operation.add_event(rest3)
assert isinstance(operation, Operation)
assert operation.op_id == 0
assert operation.requests == [
{'verb': 'BIND',
'details': ['dn="uid=bind-generateusers,ou=logins,dc=example"', 'mech=SIMPLE', 'method=128', 'ssf=0'],
}]
assert operation.response_verb == "RESULT"
assert operation.response_verb_details == set(
['tag=97', 'err=0', 'text='])
assert operation.loggable() == True
def test_add_search_multiple_with_result(self):
rest = 'SRCH base="ou=groups,dc=example" scope=2 deref=0 filter="(cn=group_name)"'
rest2 = 'SRCH attr=memberUid'
rest3 = 'SEARCH RESULT tag=101 err=0 nentries=1 text='
operation = Operation(0)
operation.add_event(rest)
operation.add_event(rest2)
operation.add_event(rest3)
assert isinstance(operation, Operation)
assert operation.op_id == 0
assert operation.requests == [
{
'verb': 'SRCH',
'details': ['attrs=memberUid', 'base="ou=groups,dc=example"', 'deref=0', 'filter="(cn=group_name)"', 'scope=2'],
}
]
assert operation.response_verb == "SEARCH RESULT"
assert operation.response_verb_details == set(
['tag=101', 'err=0', 'nentries=1', 'text='])
assert operation.loggable() == True
assert operation.error == "LDAP_SUCCESS"
def add_operation(self, rest):
# Expecting something like:
# op=1 BIND dn="uid=bind-generateusers,ou=logins,dc=example" mech=SIMPLE ssf=0
#
pattern = r'^op=(\d+) (.*)$'
match = re.search(pattern, rest)
if match:
op_id = match.group(1)
operation = self.operations.get(int(op_id))
# if an existing operation, update it's context
if operation:
operation.add_event(match.group(2))
# if a new operation, add it to our operations list
else:
operation = Operation(int(op_id))
operation.add_event(match.group(2))
self.operations[int(op_id)] = operation
if operation.loggable():
self.logger.log(self.reconstitute(operation.dict()))
else:
raise Exception('Malformed operation: {}'.format(rest))