def test_alter(): from ldap2pg.role import Role a = Role(name='toto', members=['titi'], options=dict(LOGIN=True)) b = Role(name='toto', members=['tata'], options=dict(LOGIN=False)) queries = [q.args[0] for q in a.alter(a)] assert not queries queries = [q.args[0] for q in a.alter(b)] assert fnfilter(queries, 'ALTER ROLE "toto" *;') assert fnfilter(queries, 'GRANT "toto" TO "tata";') assert fnfilter(queries, 'REVOKE "toto" FROM "titi";')
def test_comment(): from ldap2pg.role import Role a = Role(name='alan') b = Role(name='alan', comment='New comment') queries = [q.args[0] for q in a.alter(b)] assert 'COMMENT ON ROLE "alan"' in queries[0]