33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import ldap
|
|
|
|
|
|
def test_uwldap_get(uwldap_srv, uwldap_user):
|
|
retrieved_user = uwldap_srv.get_user(uwldap_user.uid)
|
|
assert retrieved_user is not None
|
|
assert retrieved_user.uid == uwldap_user.uid
|
|
|
|
assert uwldap_srv.get_user('no_such_user') is None
|
|
|
|
|
|
def test_ldap_updateprograms(cfg, ldap_srv, uwldap_srv, ldap_user, uwldap_user):
|
|
# sanity check
|
|
assert ldap_user.uid == uwldap_user.uid
|
|
# modify the user's program in UWLDAP
|
|
conn = ldap.initialize(cfg.get('uwldap_server_url'))
|
|
conn.sasl_gssapi_bind_s()
|
|
base_dn = cfg.get('uwldap_base')
|
|
dn = f'uid={uwldap_user.uid},{base_dn}'
|
|
conn.modify_s(dn, ldap.modlist.modifyModlist(
|
|
{'ou': [uwldap_user.program.encode()]},
|
|
{'ou': [b'New Program']},
|
|
))
|
|
|
|
assert ldap_srv.update_programs(dry_run=True) == [
|
|
(uwldap_user.uid, uwldap_user.program, 'New Program'),
|
|
]
|
|
assert ldap_srv.get_user(uwldap_user.uid).program == uwldap_user.program
|
|
assert ldap_srv.update_programs(members=['no_such_user']) == []
|
|
|
|
ldap_srv.update_programs()
|
|
assert uwldap_srv.get_user(uwldap_user.uid).program == 'New Program'
|