bytedance-flowgram/packages/canvas-engine/renderer/__mocks__/renderer.mock.ts

/**
 * Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 * SPDX-License-Identifier: MIT
 */

import { EntityManager, PlaygroundConfigEntity, PlaygroundContainerModule } from '@flowgram.ai/core'
import { Container } from 'inversify'

export function createPlaygroundContainer(): Container {
  const container = new Container()
  container.load(PlaygroundContainerModule)
  return container
}

export function createPlaygroundConfigEntity(): PlaygroundConfigEntity {
  return createPlaygroundContainer()
    .get<EntityManager>(EntityManager)
    .getEntity<PlaygroundConfigEntity>(PlaygroundConfigEntity, true)!
}