TREK/server/tests/unit/plugins/rpc-host-inv-strip.test.ts

/**
 * The supervisor tags every host-bound request with `_inv`, resolves the acting user
 * from it BEFORE dispatch, and then passes the request on untouched. That made `_inv`
 * visible to every handler and, in effect, a reserved param name.
 *
 * dispatch() now strips it. These tests prove the strip is invisible: the same result,
 * the same audit row, and the acting user still resolved from the host side.
 */
import type { TripsRepository } from '../../../src/db/repositories/Trips.repository';
import { PluginGuards } from '../../../src/nest-rpc/plugin-guards.service';
import { createTestPluginRegistry } from '../../../src/nest-rpc/rpc-kit/testing';
import { PackingRpc } from '../../../src/nest/packing/packing.rpc';
import type { PackingService } from '../../../src/nest/packing/packing.service';
import { PluginRpcHost } from '../../../src/nest/plugins/host/rpc-host';
import { DbRpc } from '../../../src/nest/plugins/host/rpc/db.rpc';
import type { PluginUserSettingsService } from '../../../src/nest/plugins/plugin-user-settings.service';
import type { RpcRequest, RpcResponse } from '../../../src/nest/plugins/protocol/envelope';
import type { RealtimeService } from '../../../src/nest/realtime/realtime.service';
import { TripsRpc } from '../../../src/nest/trips/trips.rpc';
import { makeDeps } from '../../helpers/rpc-host-deps';
import type { EntityManager } from '@mikro-orm/core';

import { describe, it, expect, vi } from 'vitest';

const req = (method: string, params: Record<string, unknown>): RpcRequest => ({ k: 'req', id: 'x', method, params });

/**
 * db.* is the cheapest handler to watch. The only service it needs is the per-user
 * settings store behind `settings.get`, which none of these cases dispatch, so the stub
 * carries just that read, typed off the real signature.
 */
const dbRegistry = () => {
  const settings: Pick<PluginUserSettingsService, 'readOne'> = { readOne: () => undefined };
  return createTestPluginRegistry([new DbRpc(settings as PluginUserSettingsService)]);
};

/** trips.getById lives on the decorators now, so the audit cases bind it through them. */
const tripsRegistry = () => {
  const db = {
    findAccessible: async (tripId: number, userId: number) =>
      tripId === 1 && userId === 42 ? { id: 1, user_id: 42 } : undefined,
    prepare: () => ({ get: () => ({ id: 1, title: 'Japan' }), all: () => [] }),
  } as unknown as TripsRepository;
  const guards = new PluginGuards(db, {} as never, {} as never, {} as never);
  // Plan 3c Task 7: trips.getById (RP1) now reads through
  // `EntityManager.getRepository(Trips).findRaw(...)`, not a raw statement on the handle.
  const em = {
    getRepository: () => ({ findRaw: async () => ({ id: 1, title: 'Japan', feed_token: null }) }),
  } as unknown as EntityManager;
  return createTestPluginRegistry([
    new TripsRpc({} as never, {} as never, {} as never, {} as never, {} as never, guards, {} as never, {} as never, em),
  ]);
};

describe('dispatch strips the supervisor _inv marker', () => {
  it('RPCINV-001 the handler never sees _inv', async () => {
    const deps = makeDeps();
    const host = new PluginRpcHost('p', new Set(['db:own']), deps, dbRegistry());
    await host.dispatch(req('db.query', { sql: 'SELECT 1', args: [], _inv: 'req-7' }));
    expect(deps.data.query).toHaveBeenCalledWith('SELECT 1', []);
  });

  it('RPCINV-002 the result is identical with and without the marker', async () => {
    const host = new PluginRpcHost('p', new Set(['db:own']), makeDeps(), dbRegistry());
    const withMarker = (await host.dispatch(req('db.query', { sql: 'SELECT 1', _inv: 'req-7' }))) as RpcResponse;
    const without = (await host.dispatch(req('db.query', { sql: 'SELECT 1' }))) as RpcResponse;
    expect(withMarker.ok).toBe(true);
    expect(withMarker.result).toEqual(without.result);
  });

  it('RPCINV-003 the audit row is unchanged, because auditResource reads named fields only', async () => {
    const deps = makeDeps();
    const audit = vi.fn();
    const host = new PluginRpcHost('p', new Set(['db:read:trips']), { ...deps, audit }, tripsRegistry());
    await host.dispatch(req('trips.getById', { tripId: 1, _inv: 'req-7' }), 42);
    expect(audit).toHaveBeenCalledWith({
      pluginId: 'p',
      actingUserId: 42,
      method: 'trips.getById',
      resource: 'trip:1',
      code: 'ok',
    });
  });

  it('RPCINV-004 the acting user still comes from the host, not from the params', async () => {
    const deps = makeDeps();
    const host = new PluginRpcHost('p', new Set(['db:read:trips']), deps, tripsRegistry());
    // No acting user bound: the trip read is refused even though _inv is present.
    const res = await host.dispatch(req('trips.getById', { tripId: 1, _inv: 'req-7' }));
    expect(res.ok).toBe(false);
    expect((res as { error: { code: string } }).error.code).toBe('RESOURCE_FORBIDDEN');
  });

  it('RPCINV-005 params without the marker are passed through untouched', async () => {
    const deps = makeDeps();
    const host = new PluginRpcHost('p', new Set(['db:own']), deps, dbRegistry());
    await host.dispatch(req('db.exec', { sql: 'INSERT INTO t VALUES (1)', args: [1] }));
    expect(deps.data.exec).toHaveBeenCalledWith('INSERT INTO t VALUES (1)', [1]);
  });

  it('RPCINV-006 a handler reading the whole payload object no longer finds _inv in it', async () => {
    // packing.setBagMembers is the one handler that reads the params object as a
    // whole (asPayload(p).userIds) rather than a named field off it, which makes it
    // the case where a stray _inv could actually change behaviour.
    const setBagMembers = vi.fn(() => ({ bagId: 80, members: [3] }));
    const packing = { setBagMembers } as unknown as PackingService;
    const guards = {
      requireActor: () => 42,
      requireTripEdit: () => undefined,
    } as unknown as PluginGuards;
    const realtime = { broadcast: vi.fn() } as unknown as RealtimeService;
    const host = new PluginRpcHost(
      'p',
      new Set(['db:write:packing']),
      makeDeps(),
      createTestPluginRegistry([new PackingRpc(packing, realtime, guards)]),
    );
    await host.dispatch(req('packing.setBagMembers', { tripId: 1, bagId: 80, userIds: [3], _inv: 'req-7' }), 42);
    expect(setBagMembers).toHaveBeenCalledWith('1', 80, [3]);
  });

  /**
   * The `_inv` check runs before handle()'s try/catch, so a params value that is not
   * an object used to throw a TypeError out of dispatch instead of coming back as an
   * error envelope. The supervisor discards the promise dispatch lives on, which made
   * that an unhandledRejection — process exit, from one malformed envelope.
   */
  it('RPCINV-007 params that are not an object come back as an envelope, not a rejection', async () => {
    const host = new PluginRpcHost('p', new Set(['db:own']), makeDeps(), dbRegistry());

    for (const params of [5, 'text', true] as unknown as Record<string, unknown>[]) {
      const res = await host.dispatch({ k: 'req', id: 'x', method: 'db.query', params });
      expect(res).toMatchObject({ k: 'res', id: 'x', ok: false });
    }
  });

  it('RPCINV-008 null and undefined params still behave as an empty object', async () => {
    const host = new PluginRpcHost('p', new Set(['db:own']), makeDeps(), dbRegistry());

    const nulled = await host.dispatch({
      k: 'req',
      id: 'x',
      method: 'db.query',
      params: null as unknown as Record<string, unknown>,
    });
    // The type says params is required; the IPC channel does not, so a child can
    // leave it off entirely.
    const missing = await host.dispatch({
      k: 'req',
      id: 'x',
      method: 'db.query',
      params: undefined as unknown as Record<string, unknown>,
    });
    expect(nulled).toEqual(missing);
  });
});