@Test
public void testGetTestData() {

    // Given
    given(dataSource.getTestData()).willReturn(dataSet);
    // When
    FederatedDataSet testData = cut.getTestData();
    // Then
    assertNotNull(testData);
}