jsdom-testing-mocks/src/mocks/web-animations-api/elementAnimations.ts

A set of tools for emulating browser behavior in jsdom environment

const elementAnimations = new Map<Element, Animation[]>();

export function removeAnimation(element: Element, animation: Animation) {
  const animations = elementAnimations.get(element);

  if (animations) {
    const index = animations.indexOf(animation);

    if (index !== -1) {
      animations.splice(index, 1);
    }
  }
}

export function addAnimation(element: Element, animation: Animation) {
  const animations = elementAnimations.get(element) ?? [];
  animations.push(animation);

  elementAnimations.set(element, animations);
}

export function getAnimations(this: Element) {
  return elementAnimations.get(this) ?? [];
}

export function getAllAnimations() {
  return Array.from(elementAnimations.values()).flat();
}

export function clearAnimations() {
  elementAnimations.clear();
}