linux-odyssey/packages/utils/tests/dag.test.ts

import { describe, it, expect } from '@jest/globals'
import DAG from '../src/dag.js'

describe('dag layers', () => {
  it('two layer', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
    ]
    const dag = new DAG(data)
    expect(dag.getNode('helloworld')?.layer).toBe(1)
    expect(dag.getNode('spell')?.layer).toBe(2)
  })
  it('branch', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
      {
        id: 'discover',
        requirements: ['helloworld'],
      },
    ]
    const dag = new DAG(data)
    expect(dag.getNode('helloworld')?.layer).toBe(1)
    expect(dag.getNode('spell')?.layer).toBe(2)
    expect(dag.getNode('discover')?.layer).toBe(2)
  })
  it('merge', () => {
    const data = [
      {
        id: 'A',
        requirements: [],
      },
      {
        id: 'B',
        requirements: [],
      },
      {
        id: 'C',
        requirements: ['A', 'B'],
      },
    ]
    const dag = new DAG(data)
    expect(dag.getNode('A')?.layer).toBe(1)
    expect(dag.getNode('B')?.layer).toBe(1)
    expect(dag.getNode('C')?.layer).toBe(2)
  })
  it('skip', () => {
    const data = [
      {
        id: 'A',
        requirements: [],
      },
      {
        id: 'B',
        requirements: [],
      },
      {
        id: 'C',
        requirements: ['A'],
      },
      {
        id: 'D',
        requirements: ['B', 'C'],
      },
    ]
    const dag = new DAG(data)
    expect(dag.getNode('A')?.layer).toBe(1)
    expect(dag.getNode('B')?.layer).toBe(1)
    expect(dag.getNode('C')?.layer).toBe(2)
    expect(dag.getNode('D')?.layer).toBe(3)
  })
})

describe('dag edges', () => {
  it('one edge', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
    ]

    const dag = new DAG(data)
    expect(dag.getEdges()).toEqual(new Set([['helloworld', 'spell']]))
  })

  it('branch edges', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
      {
        id: 'discover',
        requirements: ['helloworld'],
      },
    ]

    const dag = new DAG(data)
    expect(dag.getEdges()).toEqual(
      new Set([
        ['helloworld', 'discover'],
        ['helloworld', 'spell'],
      ])
    )
  })
})

describe('dag layers', () => {
  it('two layer', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
    ]

    const dag = new DAG(data)
    expect(dag.getLayers()).toEqual([1, 1])
  })

  it('branch layers', () => {
    const data = [
      {
        id: 'helloworld',
        requirements: [],
      },
      {
        id: 'spell',
        requirements: ['helloworld'],
      },
      {
        id: 'discover',
        requirements: ['helloworld'],
      },
    ]

    const dag = new DAG(data)
    expect(dag.getLayers()).toEqual([1, 2])
  })
})