@Test
  public void testGetInvoices() {

    // Given
    InvoiceNotification invoiceNotification = new InvoiceNotification();
    when(raindanceInvoiceService.getInvoices(anyString(), anyBoolean()))
        .thenReturn(Arrays.asList(invoiceNotification));

    // When
    List<InvoiceNotification> invoiceNotifications = notificationService.getInvoices("asldkfj");

    // Then
    assertEquals(1, invoiceNotifications.size());
  }
  @Test
  public void testGetInvoicesCount() throws Exception {

    // Given
    InvoiceNotification invoiceNotification = new InvoiceNotification();

    when(raindanceInvoiceService.getInvoices(anyString(), anyBoolean()))
        .thenReturn(Arrays.asList(invoiceNotification));

    // When
    Future<CountResult> count = notificationService.getInvoicesCount("anyScreenName");

    // Then
    assertEquals(1, (int) count.get().getCount());
  }