feat(models): extend artifcationGenerationCommand schema with single string, add more test cases

Author: AS

packages/models/src/lib/configuration.ts

@@ -12,12 +12,16 @@ export const artifactGenerationCommand = z.object({
     .optional(),
 });
 
-export type ArtifactGenerationCommand = z.infer<
-  typeof artifactGenerationCommand
->;
+export const artifactGenerationCommandSchema = z.union([
+  z.string({ description: 'Command as a single string' }).min(1),
+  z.object({
+    command: z.string({ description: 'Generate artifact files' }).min(1),
+    args: z.array(z.string()).optional(),
+  }),
+]);
 
 export const pluginArtifactOptionsSchema = z.object({
-  generateArtifacts: artifactGenerationCommand.optional(),
+  generateArtifacts: artifactGenerationCommandSchema.optional(),
   artifactsPaths: z.union([z.string(), z.array(z.string()).min(1)]),
 });
 

packages/models/src/lib/configuration.unit.test.ts

@@ -94,4 +94,34 @@ describe('pluginArtifactOptionsSchema', () => {
     };
     expect(pluginArtifactOptionsSchema.safeParse(data).success).toBe(false);
   });
+
+  it('should validate with generateArtifacts as a string', () => {
+    const data = {
+      generateArtifacts: 'yarn test --coverage',
+      artifactsPaths: 'coverage/lcov.info',
+    };
+    expect(pluginArtifactOptionsSchema.safeParse(data)).toStrictEqual({
+      success: true,
+      data: {
+        generateArtifacts: 'yarn test --coverage',
+        artifactsPaths: 'coverage/lcov.info',
+      },
+    });
+  });
+
+  it('should fail if generateArtifacts is an empty string', () => {
+    const data = {
+      generateArtifacts: '',
+      artifactsPaths: 'coverage/lcov.info',
+    };
+    expect(pluginArtifactOptionsSchema.safeParse(data).success).toBe(false);
+  });
+
+  it('should fail if generateArtifacts is a number', () => {
+    const data = {
+      generateArtifacts: 123,
+      artifactsPaths: 'coverage/lcov.info',
+    };
+    expect(pluginArtifactOptionsSchema.safeParse(data).success).toBe(false);
+  });
 });