public void waitForTaskStatus(long timeout, TimeUnit unit) {

        await().atMost(timeout, unit)
                .pollInterval(25, MILLISECONDS)
                .until(() ->

                        getRunningTasks().getTasks()
                                .stream()
                                .filter(task -> task.getStatus().isEmpty())
                                .collect(Collectors.toList())
                                .isEmpty());
    }

    public void waitForTotalNumberUpdate(int expected) {
        await().atMost(3, SECONDS)
            .pollInterval(50, MILLISECONDS)
            .untilAsserted(() -> assertEquals(getTotalNumberOfResults(), expected,
                "Number of filtered hosts is incorrect"));
    }