From f24b9f98e2fa997a70a63dcbae573fc5640172ff Mon Sep 17 00:00:00 2001
From: devkeydet <devkeydet@users.noreply.github.com>
Date: Sat, 4 May 2024 09:02:51 -0400
Subject: [PATCH] Add connection-count parameter to dataImport action

---
 src/actions/dataImport.ts       | 2 ++
 test/actions/dataImport.test.ts | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/actions/dataImport.ts b/src/actions/dataImport.ts
index 37a3aed7..3eadd0f8 100644
--- a/src/actions/dataImport.ts
+++ b/src/actions/dataImport.ts
@@ -9,6 +9,7 @@ import { AuthCredentials } from "../pac/auth/authParameters";
 export interface DataImportParameters extends CommonActionParameters {
   credentials: AuthCredentials;
   dataFile: HostParameterEntry;
+  connectionCount: HostParameterEntry
   verbose: HostParameterEntry;
   environmentUrl: string;
 }
@@ -29,6 +30,7 @@ export async function dataImport(parameters: DataImportParameters, runnerParamet
     logger.log("The Authentication Result: " + authenticateResult);
 
     validator.pushInput(pacArgs, "--data", parameters.dataFile);
+    validator.pushInput(pacArgs, "--connection-count", parameters.connectionCount);
     validator.pushInput(pacArgs, "--verbose", parameters.verbose);
     validator.pushCommon(pacArgs, parameters);
 
diff --git a/test/actions/dataImport.test.ts b/test/actions/dataImport.test.ts
index 969f51e7..4e428abb 100644
--- a/test/actions/dataImport.test.ts
+++ b/test/actions/dataImport.test.ts
@@ -53,6 +53,7 @@ describe("action: dataImport", () => {
   const createMinMockParameters = (): DataImportParameters => ({
     credentials: mockClientCredentials,
     dataFile: { name: "DataFile", required: true },
+    connectionCount: { name: "ConnectionCount", required: false },
     verbose: { name: "Verbose", required: false },
     environmentUrl: envUrl,
     logToConsole: false