feat(jest-docblock): support multiple of the same @pragma

Author: azz

packages/jest-docblock/README.md

@@ -99,17 +99,17 @@ string (`""`).
 Strips the top docblock from a file and return the result. If a file does not
 have a docblock at the top, then return the file unchanged.
 
-### `parse(docblock: string): {[key: string]: string}`
+### `parse(docblock: string): {[key: string]: string | string[] }`
 
 Parses the pragmas in a docblock string into an object whose keys are the pragma
 tags and whose values are the arguments to those pragmas.
 
-### `parseWithComments(docblock: string): { comments: string, pragmas: {[key: string]: string} }`
+### `parseWithComments(docblock: string): { comments: string, pragmas: {[key: string]: string | string[]} }`
 
 Similar to `parse` except this method also returns the comments from the
 docblock. Useful when used with `print()`.
 
-### `print({ comments?: string, pragmas?: {[key: string]: string} }): string`
+### `print({ comments?: string, pragmas?: {[key: string]: string | string[]} }): string`
 
 Prints an object of key-value pairs back into a docblock. If `comments` are
 provided, they will be positioned on the top of the docblock.

packages/jest-docblock/src/__tests__/index.test.js

@@ -119,6 +119,27 @@ describe('docblock', () => {
     });
   });
 
+  it('parses multiple of the same directives out of a docblock', () => {
+    const code =
+      '/**' +
+      os.EOL +
+      '' +
+      ' * @x foo' +
+      os.EOL +
+      '' +
+      ' * @x bar' +
+      os.EOL +
+      '' +
+      ' * @y' +
+      os.EOL +
+      '' +
+      ' */';
+    expect(docblock.parse(code)).toEqual({
+      x: ['foo', 'bar'],
+      y: '',
+    });
+  });
+
   it('parses directives out of a docblock with comments', () => {
     const code =
       '/**' +
@@ -395,6 +416,24 @@ describe('docblock', () => {
     );
   });
 
+  it('prints docblocks with multiple of the same pragma', () => {
+    const pragmas = {
+      x: ['a', 'b'],
+      y: 'c',
+    };
+    expect(docblock.print({pragmas})).toEqual(
+      '/**' +
+        os.EOL +
+        ' * @x a' +
+        os.EOL +
+        ' * @x b' +
+        os.EOL +
+        ' * @y c' +
+        os.EOL +
+        ' */',
+    );
+  });
+
   it('prints docblocks with pragmas', () => {
     const pragmas = {
       flow: 'foo',

packages/jest-docblock/src/index.js

@@ -64,7 +64,15 @@ export function parseWithComments(
   let match;
   while ((match = propertyRe.exec(docblock))) {
     // strip linecomments from pragmas
-    result[match[1]] = match[2].replace(lineCommentRe, '');
+    const nextPragma = match[2].replace(lineCommentRe, '');
+    if (
+      typeof result[match[1]] === 'string' ||
+      Array.isArray(result[match[1]])
+    ) {
+      result[match[1]] = [].concat(result[match[1]], nextPragma);
+    } else {
+      result[match[1]] = nextPragma;
+    }
   }
   return {comments, pragmas: result};
 }
@@ -85,15 +93,17 @@ export function print({
   const keys = Object.keys(pragmas);
 
   const printedObject = keys
-    .map(key => start + ' ' + printKeyValue(key, pragmas[key]) + line)
+    .map(key => printKeyValues(key, pragmas[key]))
+    .reduce((arr, next) => arr.concat(next), [])
+    .map(keyValue => start + ' ' + keyValue + line)
     .join('');
 
   if (!comments) {
     if (keys.length === 0) {
       return '';
     }
-    if (keys.length === 1) {
-      return `${head} ${printKeyValue(keys[0], pragmas[keys[0]])}${tail}`;
+    if (keys.length === 1 && !Array.isArray(pragmas[keys[0]])) {
+      return `${head} ${printKeyValues(keys[0], pragmas[keys[0]])}${tail}`;
     }
   }
 
@@ -113,6 +123,6 @@ export function print({
   );
 }
 
-function printKeyValue(key, value) {
-  return `@${key} ${value}`.trim();
+function printKeyValues(key, valueOrArray) {
+  return [].concat(valueOrArray).map(value => `@${key} ${value}`.trim());
 }