temporeest/integration-tests/node/src/__tests__/take.test.ts

import { context, Context, viewer, Cache, asId, commit, P } from '@aphro/runtime-ts';
import { destroyDb, initDb } from './testBase.js';
import domain from '@aphro/integration-tests-shared';
const { UserMutations } = domain.sql;
const { User } = domain.sql;

const { User: MemoryUser } = domain.mem;
const { UserMutations: MemoryUserMutations } = domain.mem;

let ctx: Context;
const cache = new Cache();
beforeAll(async () => {
  const resolver = await initDb();
  ctx = context(viewer(asId('me')), resolver, cache);
});

test('Take', async () => {
  await testTake(User, UserMutations);
  await testTake(MemoryUser, MemoryUserMutations);
});

test('Take is optimized', async () => {
  const plan = User.queryAll(ctx).take(2).plan().optimize();
  expect(plan.derivations.length).toBe(1);
  expect(plan.derivations[0].type).toEqual('modelLoad');
});

afterAll(async () => {
  await destroyDb();
});

async function testTake(
  Model: typeof User | typeof MemoryUser,
  Mutations: typeof UserMutations | typeof MemoryUserMutations,
): Promise<void> {
  await commit(
    ctx,
    [1, 2, 3, 4].flatMap(i => Mutations.create(ctx, { name: 'U' + i }).toChangesets()),
  );

  const two = await Model.queryAll(ctx).take(2).gen();
  expect(two.length).toEqual(2);
}