Add scheduler mock for bench