@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());
}