feat(scan): add template to scan create/retest

closes #464
NeuraLegion · Oct 4, 2023 · 845d79b · 845d79b
1 parent 19f53a6
commit 845d79b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/Commands/RunScan.ts b/src/Commands/RunScan.ts
@@ -94,6 +94,12 @@ export class RunScan implements CommandModule {
         string: true,
         describe: 'ID of the project'
       })
+      .option('template', {
+        alias: 'tp',
+        requiresArg: false,
+        string: true,
+        describe: 'ID of the template'
+      })
       .option('module', {
         default: Module.DAST,
         requiresArg: true,
@@ -178,6 +184,7 @@ export class RunScan implements CommandModule {
         module: args.module,
         authObjectId: args.auth,
         projectId: args.project,
+        templateId: args.template,
         buckets: args.bucket,
         hostsFilter: args.hostFilter,
         headers: Helpers.parseHeaders(args.header as string[]),

diff --git a/src/Scan/RestScans.spec.ts b/src/Scan/RestScans.spec.ts
@@ -65,6 +65,7 @@ describe('RestScans', () => {
         const scanConfig: ScanConfig = {
           name: 'scan',
           module: Module.DAST,
+          buckets: [],
           tests: [],
           fileId: 'id'
         };
@@ -100,6 +101,7 @@ describe('RestScans', () => {
       const scanConfig: ScanConfig = {
         name: 'scan',
         module: Module.DAST,
+        buckets: [],
         tests: [],
         fileId: 'id'
       };

diff --git a/src/Scan/Scans.ts b/src/Scan/Scans.ts
@@ -135,6 +135,7 @@ export interface ScanConfig {
   module: Module;
   authObjectId?: string;
   projectId?: string;
+  templateId?: string;
   discoveryTypes?: Discovery[];
   tests: TestType[];
   buckets: string[];