feat(ecs): set container when adding component to entity

RuggeroVisintin · Dec 7, 2023 · 4e80c04 · 4e80c04
1 parent e460c4d
commit 4e80c04
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/ecs/entities/BaseEntity.ts b/src/ecs/entities/BaseEntity.ts
@@ -6,6 +6,7 @@ export class BaseEntity implements IEntity {
 
     public addComponent(key: string, component: IComponent): void {
         this.components.set(key, component);
+        component.setContainer(this);
     }
 
     public getComponent<Component extends IComponent>(key: string): Component | undefined {

diff --git a/test/unit/ecs/entities/BaseEntity.test.ts b/test/unit/ecs/entities/BaseEntity.test.ts
@@ -1,15 +1,26 @@
 import { BaseComponent, BaseEntity } from "../../../../src";
 
 describe('ecs/entities/BaseEntity', () => {
-    const baseEntity = new BaseEntity();
+    let baseEntity = new BaseEntity();
 
+    beforeEach(() => {
+        baseEntity = new BaseEntity();
+    });
+
     describe('.addComponent()', () => {
         it('Should add component to entity', () => {
             const testComponent = new BaseComponent();
 
             baseEntity.addComponent('testComponent', testComponent);
             expect(baseEntity.getComponent<BaseComponent>('testComponent')).toEqual(testComponent);
         });
+
+        it('Should set the entity as container of the given component', () => {
+            const testComponent = new BaseComponent();
+
+            baseEntity.addComponent('testComponent', testComponent);
+            expect(testComponent.getContainer()).toEqual(baseEntity);
+        })
     })
 
     describe('.getComponent()', () => {