class DataSourceDecorator implements TestClassDecorator {

    @Override
    public void afterAll(final TestInvocation invocation) throws Exception {
        final ExecutionContext context = invocation.getContext();
        final BasicDataSource ds = (BasicDataSource) context.getData(Constants.KEY_DATA_SOURCE);
        ds.close();
        context.storeData(Constants.KEY_DATA_SOURCE, null);
    }

}