@Override
  public CompletableFuture<Void> applyExperimentTreatment(final Account account, final Device device) {
    return idleDeviceNotificationScheduler.scheduleNotification(account, device, PREFERRED_NOTIFICATION_TIME);
  }
@Test
  void applyExperimentTreatment() {
    final Account account = mock(Account.class);
    final Device device = mock(Device.class);

    experiment.applyExperimentTreatment(account, device);

    verify(idleDeviceNotificationScheduler)
        .scheduleNotification(account, device, NotifyIdleDevicesWithMessagesExperiment.PREFERRED_NOTIFICATION_TIME);
  }