SubMiner/src/main/early-single-instance.test.ts

import assert from 'node:assert/strict';
import test from 'node:test';
import {
  registerSecondInstanceHandlerEarly,
  requestSingleInstanceLockEarly,
  resetEarlySingleInstanceStateForTests,
} from './early-single-instance';
import * as earlySingleInstance from './early-single-instance';

function createFakeApp(lockValue = true) {
  let requestCalls = 0;
  let requestData: unknown = null;
  let secondInstanceListener:
    | ((
        _event: unknown,
        argv: string[],
        workingDirectory?: string,
        additionalData?: unknown,
      ) => void)
    | null = null;

  return {
    app: {
      requestSingleInstanceLock: (additionalData?: unknown) => {
        requestCalls += 1;
        requestData = additionalData ?? null;
        return lockValue;
      },
      on: (
        _event: 'second-instance',
        listener: (
          _event: unknown,
          argv: string[],
          workingDirectory?: string,
          additionalData?: unknown,
        ) => void,
      ) => {
        secondInstanceListener = listener;
      },
    },
    emitSecondInstance: (argv: string[], additionalData?: unknown) => {
      secondInstanceListener?.({}, argv, '/tmp', additionalData);
    },
    getRequestCalls: () => requestCalls,
    getRequestData: () => requestData,
  };
}

test('requestSingleInstanceLockEarly caches the lock result per process', () => {
  resetEarlySingleInstanceStateForTests();
  const fake = createFakeApp(true);

  assert.equal(requestSingleInstanceLockEarly(fake.app), true);
  assert.equal(requestSingleInstanceLockEarly(fake.app), true);
  assert.equal(fake.getRequestCalls(), 1);
});

test('registerSecondInstanceHandlerEarly replays queued argv and forwards new events', () => {
  resetEarlySingleInstanceStateForTests();
  const fake = createFakeApp(true);
  const calls: string[][] = [];

  assert.equal(requestSingleInstanceLockEarly(fake.app), true);
  fake.emitSecondInstance(['SubMiner.exe', '--start', '--socket', '\\\\.\\pipe\\subminer']);

  registerSecondInstanceHandlerEarly(fake.app, (_event, argv) => {
    calls.push(argv);
  });
  fake.emitSecondInstance(['SubMiner.exe', '--start', '--show-visible-overlay']);

  assert.deepEqual(calls, [
    ['SubMiner.exe', '--start', '--socket', '\\\\.\\pipe\\subminer'],
    ['SubMiner.exe', '--start', '--show-visible-overlay'],
  ]);
});

test('requestSingleInstanceLockEarly sends normalized argv through second-instance data', () => {
  resetEarlySingleInstanceStateForTests();
  const fake = createFakeApp(true);
  const primaryArgv = ['SubMiner.AppImage', '--start'];
  const transportedArgv = ['SubMiner.AppImage', '--stop'];
  const calls: string[][] = [];

  assert.equal(requestSingleInstanceLockEarly(fake.app, primaryArgv), true);
  registerSecondInstanceHandlerEarly(fake.app, (_event, argv) => {
    calls.push(argv);
  });
  fake.emitSecondInstance(['SubMiner.AppImage'], { subminerArgv: transportedArgv });

  assert.deepEqual(fake.getRequestData(), { subminerArgv: primaryArgv });
  assert.deepEqual(calls, [transportedArgv]);
});

test('stats daemon args bypass the normal single-instance lock path', () => {
  const shouldBypass = (
    earlySingleInstance as typeof earlySingleInstance & {
      shouldBypassSingleInstanceLockForArgv?: (argv: string[]) => boolean;
    }
  ).shouldBypassSingleInstanceLockForArgv;

  assert.equal(typeof shouldBypass, 'function');
  assert.equal(shouldBypass?.(['SubMiner', '--stats', '--stats-background']), true);
  assert.equal(shouldBypass?.(['SubMiner', '--stats', '--stats-stop']), true);
  assert.equal(shouldBypass?.(['SubMiner', '--stats']), false);
});