@Test
public void testCloseDataSource() throws Throwable {

    // GIVEN
    final DataSourceDecorator decorator = new DataSourceDecorator();
    // WHEN
    decorator.afterAll(invocation);
    // THEN
    verify(ds).close();
    verify(ctx).storeData(eq(Constants.KEY_DATA_SOURCE), eq(null));
}