class DbUnitDatabaseConnectionDecorator implements TestClassDecorator {

    @Override
    public void afterAll(final TestInvocation invocation) throws Exception {
        final ExecutionContext context = invocation.getContext();
        final IDatabaseConnection connection = (IDatabaseConnection) context.getData(Constants.KEY_CONNECTION);
        context.storeData(Constants.KEY_CONNECTION, null);

        connection.close();
    }

}