Function testJoin

packages/db-ivm/tests/operators/join.test.ts:18–344

()

})

function testJoin() {
  test(`basic join operation`, () => {
    const graph = new D2()
    const inputA = graph.newInput<[number, string]>()
    const inputB = graph.newInput<[number, string]>()
    const tracker = new KeyedMessageTracker<number, [string, string]>()

    inputA.pipe(
      join(inputB),
      output((message) => {
        tracker.addMessage(message as MultiSet<[number, [string, string]]>)
      }),
    )

    graph.finalize()

    inputA.sendData(
      new MultiSet([
        [[1, `a`], 1],
        [[2, `b`], 1],
      ]),
    )

    inputB.sendData(
      new MultiSet([
        [[1, `x`], 1],
        [[2, `y`], 1],
        [[3, `z`], 1], // key 3 only exists in B, so no join output expected
      ]),
    )

    graph.run()

    const result = tracker.getResult()

    // Assert only keys that can actually join (1, 2) are affected, not key 3
    assertOnlyKeysAffected(`basic join operation`, result.messages, [1, 2])

    // Assert the final materialized results are correct
    assertKeyedResults(
      `basic join operation`,
      result,
      [
        [1, [`a`, `x`]],
        [2, [`b`, `y`]],
      ],
      4, // Expected message count
    )
  })

  test(`join with late arriving data`, () => {
    const graph = new D2()
    const inputA = graph.newInput<[number, string]>()
    const inputB = graph.newInput<[number, string]>()
    const tracker = new KeyedMessageTracker<number, [string, string]>()

    inputA.pipe(
      join(inputB),