From 537f81d27680988ccff45b11f25bc0bef963a422 Mon Sep 17 00:00:00 2001
From: bbimber <bbimber@gmail.com>
Date: Mon, 8 Apr 2024 14:07:54 -0700
Subject: [PATCH] Switch from ETLHelper to DataIntegrationHelper

---
 .../src/org/labkey/test/tests/pmr/PMRTest.java     | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/PMR/test/src/org/labkey/test/tests/pmr/PMRTest.java b/PMR/test/src/org/labkey/test/tests/pmr/PMRTest.java
index 6dd2d5746..650b7ffd8 100644
--- a/PMR/test/src/org/labkey/test/tests/pmr/PMRTest.java
+++ b/PMR/test/src/org/labkey/test/tests/pmr/PMRTest.java
@@ -17,38 +17,30 @@
 package org.labkey.test.tests.pmr;
 
 import au.com.bytecode.opencsv.CSVReader;
-import org.apache.hc.client5.http.classic.methods.HttpUriRequest;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
-import org.labkey.remoteapi.Command;
-import org.labkey.remoteapi.CommandResponse;
-import org.labkey.remoteapi.PostCommand;
 import org.labkey.remoteapi.query.SelectRowsCommand;
 import org.labkey.remoteapi.query.TruncateTableCommand;
 import org.labkey.serverapi.reader.Readers;
 import org.labkey.test.BaseWebDriverTest;
 import org.labkey.test.Locator;
-import org.labkey.test.Locators;
 import org.labkey.test.ModulePropertyValue;
 import org.labkey.test.TestFileUtils;
 import org.labkey.test.TestTimeoutException;
 import org.labkey.test.WebTestHelper;
 import org.labkey.test.categories.External;
 import org.labkey.test.categories.LabModule;
-import org.labkey.test.tests.di.ETLHelper;
 import org.labkey.test.util.Ext4Helper;
-import org.labkey.test.util.PipelineStatusTable;
 import org.labkey.test.util.RReportHelper;
 import org.labkey.test.util.RemoteConnectionHelper;
 import org.labkey.test.util.SqlserverOnlyTest;
+import org.labkey.test.util.di.DataIntegrationHelper;
 import org.labkey.test.util.ehr.EHRClientAPIHelper;
-import org.labkey.test.util.ehr.EHRTestHelper;
 
 import java.io.File;
-import java.net.URI;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.Date;
@@ -60,7 +52,7 @@
 @Category({External.class, LabModule.class})
 public class PMRTest extends BaseWebDriverTest implements SqlserverOnlyTest
 {
-    private final ETLHelper _etlHelper = new ETLHelper(this, getProjectName());
+    private final DataIntegrationHelper _etlHelper = new DataIntegrationHelper(getProjectName());
 
     @Override
     protected void doCleanup(boolean afterTest) throws TestTimeoutException
@@ -158,7 +150,7 @@ private void testKinshipEtl() throws Exception
         Assert.assertEquals("Incorrect number of kinship rows", 0, select1.execute(getApiHelper().getConnection(), getProjectName()).getRowCount().intValue());
 
         // Kick off ETL to stage data. This should also kick off a separate pipeline job to import, using geneticscore-importGeneticsData.view
-        _etlHelper.runETL("{PMR}/KinshipDataStaging");
+        _etlHelper.runTransform("{PMR}/KinshipDataStaging");
         goToDataPipeline();
         waitForPipelineJobsToComplete(4, "ETL Job: Import PRIMe-seq Kinship Data", false);