@Before
  public void setup() throws Exception {
    super.setup();
    driverProvider = wfDriverProvider;
    dataSourceProvider =
        new WildflyDataSourceProvider(wfDriverProvider) {
          @Override
          protected Object jndiLookupDataSource(String jndi) {
            return sqlDataSource;
          }

          @Override
          protected String generateRandomUUID() {
            return RANDOM_UUID;
          }
        };

    ((WildflyDataSourceProvider) dataSourceProvider).setDataSourceMgmtClient(managementClient);

    wfDataSources = createWFDataSources();
  }