Mocking AsyncLocalStorage Contexts for Isolated Server Component Unit Tests

import { AsyncLocalStorage } from 'node:async_hooks';
 
const requestContextStorage = new AsyncLocalStorage();
 
export function runInRequestContext(store, callback) {
  return requestContextStorage.run(store, callback);
}
 
export function testRequestContextIsolation() {
  const storeA = { tenantId: 'tenant-alpha' };
  const storeB = { tenantId: 'tenant-beta' };
  let resultA, resultB;
 
  runInRequestContext(storeA, () => {
    resultA = requestContextStorage.getStore()?.tenantId;
  });
  runInRequestContext(storeB, () => {
    resultB = requestContextStorage.getStore()?.tenantId;
  });
 
  console.assert(resultA === 'tenant-alpha', 'Tenant A isolation check failed');
  console.assert(resultB === 'tenant-beta', 'Tenant B isolation check failed');
}