class Neo4JDriverDecorator implements TestClassDecorator {

    @Override
    public void beforeAll(final TestInvocation invocation) throws Exception {
        final ExecutionContext context = invocation.getContext();
        final Configuration configuration = configurationRegistry.getConfiguration(context.getDescriptor());
        context.storeData(Constants.KEY_DATA_SOURCE, configuration.createDataSource());
    }

}