Replace node-module 'tempfile' seamlessly with 'tmp'

Author: d4vidi

detox/package.json

@@ -99,6 +99,7 @@
     "strip-ansi": "^6.0.1",
     "telnet-client": "1.2.8",
     "tempfile": "^2.0.0",
+    "tmp": "^0.2.1",
     "trace-event-lib": "^1.3.1",
     "which": "^1.3.1",
     "ws": "^7.0.0",

detox/src/android/AndroidExpect.test.js

@@ -9,7 +9,7 @@ describe('AndroidExpect', () => {
 
   beforeEach(() => {
     jest.mock('../utils/logger');
-    jest.mock('tempfile');
+    jest.mock('../utils/tempfile');
     jest.mock('fs-extra');
 
     mockExecutor = new MockExecutor();
@@ -341,7 +341,7 @@ describe('AndroidExpect', () => {
         mockExecutor.executeResult = Promise.resolve(invokeResultInBase64);
 
         fs = require('fs-extra');
-        tempfile = require('tempfile');
+        tempfile = require('../utils/tempfile');
         tempfile.mockReturnValue(tempFilePath);
 
         _element = e.element(e.by.id('FancyElement'));

detox/src/android/core/NativeElement.js

@@ -1,7 +1,7 @@
 const path = require('path');
 
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('../../utils/tempfile');
 
 const DetoxRuntimeError = require('../../errors/DetoxRuntimeError');
 const invoke = require('../../invoke');

detox/src/artifacts/log/ios/SimulatorLogPlugin.test.js

@@ -5,7 +5,7 @@ const path = require('path');
 
 const fs = require('fs-extra');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('../../../utils/tempfile');
 
 const childProcess = require('../../../utils/childProcess');
 

detox/src/artifacts/templates/artifact/FileArtifact.js

@@ -2,7 +2,7 @@
 const path = require('path');
 
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('../../../utils/tempfile');
 
 const appendFile = require('../../../utils/appendFile');
 

detox/src/artifacts/templates/artifact/FileArtifact.test.js

@@ -1,6 +1,6 @@
 const fs = require('fs-extra');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('../../../utils/tempfile');
 
 describe('FileArtifact', () => {
   let FileArtifact, fileArtifact, logger, temporaryPath, temporaryData, destinationPath;

detox/src/artifacts/utils/temporaryPath.js

@@ -3,7 +3,7 @@ const { promisify } = require('util');
 
 const glob = require('glob');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('../../utils/tempfile');
 
 const { useForwardSlashes } = require('../../utils/shellUtils');
 

detox/src/artifacts/utils/temporaryPath.test.js

@@ -1,7 +1,7 @@
 const path = require('path');
 
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('../../utils/tempfile');
 
 const temporaryPath = require('./temporaryPath');
 

detox/src/client/Client.test.js

@@ -2,7 +2,7 @@
 jest.useFakeTimers('modern');
 
 const { serializeError } = require('serialize-error');
-const tempfile = require('tempfile');
+const tempfile = require('../utils/tempfile');
 
 const { validSession } = require('../configuration/configurations.mock');
 const Deferred = require('../utils/Deferred');

detox/src/devices/allocation/DeviceRegistry.test.js

@@ -1,5 +1,5 @@
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('../../utils/tempfile');
 
 const DeviceRegistry = require('./DeviceRegistry');  // Adjust the path to your actual file
 

detox/src/ios/expectTwo.js

@@ -3,7 +3,7 @@ const path = require('path');
 
 const fs = require('fs-extra');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('../utils/tempfile');
 
 const { assertTraceDescription, assertEnum, assertNormalized } = require('../utils/assertArgument');
 const { removeMilliseconds } = require('../utils/dateUtils');

detox/src/ios/expectTwo.test.js

@@ -11,8 +11,8 @@ describe('expectTwo', () => {
 
   beforeEach(() => {
     jest.mock('../utils/logger');
+    jest.mock('../utils/tempfile');
     jest.mock('fs-extra');
-    jest.mock('tempfile');
 
     fs = require('fs-extra');
     const IosExpect = require('./expectTwo');
@@ -690,7 +690,7 @@ describe('expectTwo', () => {
         screenshotPath: deviceTmpFilePath,
       });
 
-      require('tempfile').mockReturnValue(tmpFilePath);
+      require('../utils/tempfile').mockReturnValue(tmpFilePath);
       result = await e.element(e.by.id('uniqueId')).takeScreenshot(imageName);
     });
 

detox/src/logger/DetoxLogger.test.js

@@ -8,7 +8,7 @@ const os = require('os');
 
 const fs = require('fs-extra');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('../utils/tempfile');
 
 const sleep = require('../utils/sleep');
 

detox/src/logger/utils/DetoxLogFinalizer.test.js

@@ -5,7 +5,7 @@ const { PassThrough } = require('stream');
 
 const fs = require('fs-extra');
 const glob = require('glob');
-const tempfile = require('tempfile');
+const tempfile = require('../../utils/tempfile');
 
 const temporary = require('../../artifacts/utils/temporaryPath');
 const DetoxLogger = require('../DetoxLogger');

detox/src/logger/utils/streams/BunyanTransformer.test.js

@@ -1,5 +1,5 @@
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('../../../utils/tempfile');
 
 describe('BunyanTransformer', () => {
   /** @type {import('./BunyanTransformer')} */

detox/src/utils/ExclusiveLockfile.test.js

@@ -2,7 +2,7 @@ jest.mock('proper-lockfile');
 
 const fs = require('fs-extra');
 const plock = require('proper-lockfile');
-const tempfile = require('tempfile');
+const tempfile = require('./tempfile');
 
 const ExclusiveLockFile = require('./ExclusiveLockfile');
 

detox/src/utils/appendFile.test.js

@@ -1,5 +1,5 @@
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('./tempfile');
 
 const appendFile = require('./appendFile');
 

detox/src/utils/environment.test.js

@@ -3,7 +3,7 @@ const path = require('path');
 
 const fs = require('fs-extra');
 const _ = require('lodash');
-const tempfile = require('tempfile');
+const tempfile = require('./tempfile');
 
 describe('Environment', () => {
   let Environment;

detox/src/utils/fsext.test.js

@@ -1,7 +1,7 @@
 const path = require('path');
 
 const fs = require('fs-extra');
-const tempfile = require('tempfile');
+const tempfile = require('./tempfile');
 
 const fsext = require('./fsext');
 

detox/src/utils/tempfile.js

@@ -0,0 +1,18 @@
+const tmp = require('tmp');
+
+tmp.setGracefulCleanup();
+
+/**
+ * Creates a temporary file path. If extension is provided, it will be appended to the path.
+ * @param {string} [extension] - Optional file extension to append to the temporary file path
+ * @returns {string} Path to a temporary file
+ */
+module.exports = function(extension) {
+  const _extension = (extension && extension.startsWith('.'))
+    ? extension
+    : (extension && `.${extension}`);
+
+  return tmp.tmpNameSync({
+    template: `detox-${process.pid}-XXXXXX${_extension || ''}`,
+  });
+};

detox/src/utils/tempfile.test.js

@@ -0,0 +1,52 @@
+describe('tempfile', () => {
+  let tmp;
+  let tempfile;
+
+  beforeEach(() => {
+    jest.mock('tmp');
+    tmp = require('tmp');
+    tempfile = require('./tempfile');
+  });
+
+  const expectTmpCalled = (extension = '') => {
+    const template = `detox-${process.pid}-XXXXXX${extension}`;
+    expect(tmp.tmpNameSync).toHaveBeenCalledWith({ template });
+  };
+
+  it(`should enable tmp's graceful cleanup`, () => {
+    expect(tmp.setGracefulCleanup).toHaveBeenCalled();
+  });
+
+  it('should return the value from tmp.tmpNameSync', () => {
+    const mockPath = '/tmp/detox-123-abc123';
+    tmp.tmpNameSync.mockReturnValueOnce(mockPath);
+
+    const result = tempfile();
+    expect(result).toEqual(mockPath);
+  });
+
+  it('should create a temporary file path without extension', () => {
+    tempfile();
+    expectTmpCalled();
+  });
+
+  it('should create a temporary file path with extension', () => {
+    tempfile('txt');
+    expectTmpCalled('.txt');
+  });
+
+  it('should handle extension with leading dot', () => {
+    tempfile('.txt');
+    expectTmpCalled('.txt');
+  });
+
+  it('should handle empty extension', () => {
+    tempfile('');
+    expectTmpCalled();
+  });
+
+  it('should handle undefined extension', () => {
+    tempfile();
+    expectTmpCalled();
+  });
+});