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