tldraw/__tests__/test-utils.ts

import { Data, Shape, ShapeType } from 'types'
import { getSelectedIds, getSelectedShapes, getShape } from 'utils'

export const rectangleId = '1f6c251c-e12e-40b4-8dd2-c1847d80b72f'
export const arrowId = '5ca167d7-54de-47c9-aa8f-86affa25e44d'

interface PointerOptions {
  id?: string
  x?: number
  y?: number
  shiftKey?: boolean
  altKey?: boolean
  ctrlKey?: boolean
}

export function point(
  options: PointerOptions = {} as PointerOptions
): PointerEvent {
  const {
    id = '1',
    x = 0,
    y = 0,
    shiftKey = false,
    altKey = false,
    ctrlKey = false,
  } = options

  return {
    shiftKey,
    altKey,
    ctrlKey,
    pointerId: id,
    clientX: x,
    clientY: y,
  } as any
}

export function idsAreSelected(
  data: Data,
  ids: string[],
  strict = true
): boolean {
  const selectedIds = getSelectedIds(data)
  return (
    (strict ? selectedIds.size === ids.length : true) &&
    ids.every((id) => selectedIds.has(id))
  )
}

export function hasParent(
  data: Data,
  childId: string,
  parentId: string
): boolean {
  return getShape(data, childId).parentId === parentId
}

export function getOnlySelectedShape(data: Data): Shape {
  const selectedShapes = getSelectedShapes(data)
  return selectedShapes.length === 1 ? selectedShapes[0] : undefined
}

export function assertShapeType(
  data: Data,
  shapeId: string,
  type: ShapeType
): boolean {
  const shape = getShape(data, shapeId)
  if (shape.type !== type) {
    throw new TypeError(
      `expected shape ${shapeId} to be of type ${type}, found ${shape?.type} instead`
    )
  }
  return true
}

export function assertShapeProps<T extends Shape>(
  shape: T,
  props: { [K in keyof Partial<T>]: T[K] }
): boolean {
  for (const key in props) {
    if (shape[key] !== props[key]) {
      throw new TypeError(
        `expected shape ${shape.id} to have property ${key}: ${props[key]}, found ${key}: ${shape[key]} instead`
      )
    }
  }
  return true
}