feat: add fileExtension option (#29)

satya164 · Jul 8, 2024 · 33d32bd · 33d32bd
1 parent ad1cb3b
commit 33d32bd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -10,6 +10,7 @@ export const toMatchFile: (
 
 declare interface FileMatcherOptions {
   diff?: DiffOptions;
+  fileExtension?: string;
 }
 
 declare global {

diff --git a/index.js b/index.js
@@ -22,12 +22,17 @@ const isEqual = (a, b) => {
  *
  * @param {string | Buffer} content Output content to match
  * @param {string} [filepath] Path to the file to match against
- * @param {{ diff?: import('jest-diff').DiffOptions }} options Additional options for matching
+ * @param {{
+ *   diff?: import('jest-diff').DiffOptions,
+ *   fileExtension?: string,
+ * }} options
  * @this {{ testPath: string, currentTestName: string, assertionCalls: number, isNot: boolean, snapshotState: { added: number, updated: number, unmatched: number, _updateSnapshot: 'none' | 'new' | 'all' } }}
  */
 exports.toMatchFile = function toMatchFile(content, filepath, options = {}) {
   const { isNot, snapshotState } = this;
 
+  const fileExtension = options.fileExtension || '';
+
   const filename =
     filepath === undefined
       ? // If file name is not specified, generate one from the test title
@@ -37,7 +42,7 @@ exports.toMatchFile = function toMatchFile(content, filepath, options = {}) {
           `${filenamify(this.currentTestName, {
             replacement: '-',
           }).replace(/\s/g, '-')}-${this.assertionCalls}`
-        )
+        ) + fileExtension
       : filepath;
 
   options = {