def test_has_permission(self, connection):
pg_base = Base(connection.engine.url.database)
pg_base.verbose = False
value = pg_base.has_permission(
connection.engine.url.username,
"usesuper",
)
assert value is True
with pytest.raises(RuntimeError) as excinfo:
value = pg_base.has_permission(
connection.engine.url.username,
"sudo",
)
assert "Invalid user permission sudo" in str(excinfo.value)
def test_has_permissions(self, connection):
pg_base = Base(connection.engine.url.database)
pg_base.verbose = False
value = pg_base.has_permissions(
connection.engine.url.username,
["usesuper"],
)
assert (pg_base.has_permissions(
connection.engine.url.username,
["usesuper"],
) is True)
assert (pg_base.has_permissions(
"spiderman",
["usesuper"],
) is False)
with pytest.raises(InvalidPermissionError) as excinfo:
pg_base.has_permissions(
connection.engine.url.username,
["sudo"],
)
def test_pg_settings(self, connection):
pg_base = Base(connection.engine.url.database)
pg_base.verbose = False
value = pg_base.pg_settings("max_replication_slots")
assert int(value) > 0
assert pg_base.pg_settings("xyz") is None