diff --git a/packages/react-components/lib/map/three-fiber/robot-three-maker.spec.tsx b/packages/react-components/lib/map/three-fiber/robot-three-maker.spec.tsx index d8835827e..4c12a6854 100644 --- a/packages/react-components/lib/map/three-fiber/robot-three-maker.spec.tsx +++ b/packages/react-components/lib/map/three-fiber/robot-three-maker.spec.tsx @@ -1,16 +1,17 @@ import React from 'react'; -import { RobotThreeMaker } from './robot-three-maker'; +import { RobotData, RobotThreeMaker } from './robot-three-maker'; import { Euler, Vector3 } from 'three'; import ReactThreeTestRenderer from '@react-three/test-renderer'; describe('RobotThreeMaker', () => { - it('renders color properly', async () => { - const robot = { + it('renders robot properly', async () => { + const robot: RobotData = { fleet: 'Fleet 1', name: 'Robot 1', model: 'Model 1', footprint: 1.0, color: 'blue', + scale: 1.0, }; const renderer = await ReactThreeTestRenderer.create( @@ -19,11 +20,37 @@ describe('RobotThreeMaker', () => { position={new Vector3(0, 0, 0)} rotation={new Euler(0, 0, 0)} circleSegment={64} + robotLabel={true} />, ); const mesh = renderer.scene.children[0].allChildren; - expect(mesh.length).toBe(1); + expect(mesh.length).toBe(3); + }); + + it('renders robot without label', async () => { + const robot: RobotData = { + fleet: 'Fleet 1', + name: 'Robot 1', + model: 'Model 1', + footprint: 1.0, + color: 'blue', + scale: 1.0, + }; + + const renderer = await ReactThreeTestRenderer.create( + , + ); + + const mesh = renderer.scene.children[0].allChildren; + + expect(mesh.length).toBe(2); }); });