silphius/app/ecs/ecs.test.js

import {expect, test} from 'vitest';

import Component from './component.js';
import Ecs from './ecs.js';
import System from './system.js';

function wrapProperties(name, properties) {
  return class WrappedComponent extends Component {
    static componentName = name;
    static properties = properties;
  };
}

const Empty = wrapProperties('Empty', {});

const Name = wrapProperties('Name', {
  name: {type: 'string'},
});

const Position = wrapProperties('Position', {
  x: {type: 'int32', defaultValue: 32},
  y: {type: 'int32'},
  z: {type: 'int32'},
});

function asyncTimesTwo(x) {
  return new Promise((resolve) => {
    setTimeout(() => {
      resolve(x * 2)
    }, 5);
  });
}

class Async extends Component {
  static componentName = 'Async';
  static properties = {
    foo: {type: 'uint8'},
  };
  async load(instance) {
    instance.foo = await asyncTimesTwo(instance.foo);
  }
}

test('activates and deactivates systems at runtime', () => {
  let oneCount = 0;
  let twoCount = 0;
  const ecs = new Ecs({
    Systems: {
      OneSystem: class extends System {
        tick() {
          oneCount += 1;
        }
      },
      TwoSystem: class extends System {
        tick() {
          twoCount += 1;
        }
      },
    },
  });
  ecs.tick();
  expect(oneCount)
    .to.equal(0);
  expect(twoCount)
    .to.equal(0);
  ecs.system('OneSystem').active = true;
  ecs.tick();
  expect(oneCount)
    .to.equal(1);
  ecs.system('TwoSystem').active = true;
  ecs.tick();
  expect(oneCount)
    .to.equal(2);
  expect(twoCount)
    .to.equal(1);
  ecs.system('OneSystem').active = false;
  ecs.tick();
  expect(oneCount)
    .to.equal(2);
  expect(twoCount)
    .to.equal(2);
});

test('creates entities with components', async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  const entity = await ecs.create({Empty: {}, Position: {y: 128}});
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}, Position: {y: 128}}));
});

test("removes entities' components", async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  const entity = await ecs.create({Empty: {}, Position: {y: 128}});
  ecs.remove(entity, ['Position']);
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}}));
});

test('gets entities', async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  const entity = await ecs.create({Empty: {}, Position: {y: 128}});
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}, Position: {y: 128}}));
});

test('destroys entities', async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  const entity = await ecs.create({Empty: {}, Position: {y: 128}});
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}, Position: {y: 128}}));
  expect(ecs.get(entity))
    .to.not.be.undefined;
  ecs.destroyAll();
  expect(ecs.get(entity))
    .to.be.undefined;
  expect(() => {
    ecs.destroy(entity);
  })
    .to.throw();
});

test('inserts components into entities', async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  const entity = await ecs.create({Empty: {}});
  ecs.insert(entity, {Position: {y: 128}});
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}, Position: {y: 128}}));
  ecs.insert(entity, {Position: {y: 64}});
  expect(JSON.stringify(ecs.get(entity)))
    .to.deep.equal(JSON.stringify({Empty: {}, Position: {y: 64}}));
});

test('ticks systems', async () => {
  const Momentum = wrapProperties('Momentum', {
    x: {type: 'int32'},
    y: {type: 'int32'},
    z: {type: 'int32'},
  });
  const ecs = new Ecs({
    Components: {Momentum, Position},
    Systems: {
      Physics: class Physics extends System {

        static queries() {
          return {
            default: ['Position', 'Momentum'],
          };
        }

        tick(elapsed) {
          for (const {Position, Momentum} of this.select('default')) {
            Position.x += Momentum.x * elapsed;
            Position.y += Momentum.y * elapsed;
            Position.z += Momentum.z * elapsed;
          }
        }

      },
    },
  });
  ecs.system('Physics').active = true;
  const entity = await ecs.create({Momentum: {}, Position: {y: 128}});
  const position = JSON.stringify(ecs.get(entity).Position);
  ecs.tick(1);
  expect(JSON.stringify(ecs.get(entity).Position))
    .to.deep.equal(position);
  ecs.get(1).Momentum.y = 30;
  ecs.tick(1);
  expect(JSON.stringify(ecs.get(entity).Position))
    .to.deep.equal(JSON.stringify({y: 128 + 30}));
});

test('creates many entities when ticking systems', () => {
  const ecs = new Ecs({
    Systems: {
      Spawn: class extends System {
        tick() {
          this.createManyEntities(Array.from({length: 5}).map(() => []));
        }
      },
    },
  });
  ecs.system('Spawn').active = true;
  ecs.create();
  expect(ecs.get(5))
    .to.be.undefined;
  ecs.tick(1);
  expect(ecs.get(5))
    .to.not.be.undefined;
});

test('creates entities when ticking systems', () => {
  const ecs = new Ecs({
    Systems: {
      Spawn: class extends System {
        tick() {
          this.createEntity();
        }
      },
    },
  });
  ecs.system('Spawn').active = true;
  ecs.create();
  expect(ecs.get(2))
    .to.be.undefined;
  ecs.tick(1);
  expect(ecs.get(2))
    .to.not.be.undefined;
});

test('schedules entities to be deleted when ticking systems', () => {
  const ecs = new Ecs({
    Systems: {
      Despawn: class extends System {
        tick() {
          this.destroyEntity(1);
        }
      },
    },
  });
  ecs.system('Despawn').active = true;
  ecs.create();
  ecs.tick(1);
  expect(ecs.get(1))
    .to.be.undefined;
});

test('adds components to and remove components from entities when ticking systems', async () => {
  let promise;
  const ecs = new Ecs({
    Components: {Foo: wrapProperties('Foo', {bar: {type: 'uint8'}})},
    Systems: {
      AddComponent: class extends System {
        static queries() {
          return {
            default: ['Foo'],
          };
        }
        tick() {
          promise = this.insertComponents(1, {Foo: {}});
        }
      },
      RemoveComponent: class extends System {
        static queries() {
          return {
            default: ['Foo'],
          };
        }
        tick() {
          this.removeComponents(1, ['Foo']);
        }
      },
    },
  });
  ecs.system('AddComponent').active = true;
  ecs.create();
  ecs.tick(1);
  await promise;
  expect(Array.from(ecs.system('AddComponent').select('default')).length)
    .to.equal(1);
  expect(ecs.get(1).Foo)
    .to.not.be.undefined;
  ecs.system('AddComponent').active = false;
  ecs.system('RemoveComponent').active = true;
  ecs.tick(1);
  expect(Array.from(ecs.system('RemoveComponent').select('default')).length)
    .to.equal(0);
  expect(ecs.get(1).Foo)
    .to.be.undefined;
});

test('generates diffs for entity creation', async () => {
  const ecs = new Ecs();
  let entity;
  entity = await ecs.create();
  expect(ecs.diff)
    .to.deep.equal({[entity]: {}});
});

test('generates diffs for adding and removing components', async () => {
  const ecs = new Ecs({Components: {Position}});
  let entity;
  entity = await ecs.create();
  ecs.setClean();
  ecs.insert(entity, {Position: {x: 64}});
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: {x: 64}}});
  ecs.setClean();
  expect(ecs.diff)
    .to.deep.equal({});
  ecs.remove(entity, ['Position']);
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: false}});
});

test('generates diffs for empty components', async () => {
  const ecs = new Ecs({Components: {Empty}});
  let entity;
  entity = await ecs.create({Empty: {}});
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Empty: {}}});
  ecs.setClean();
  ecs.remove(entity, ['Empty']);
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Empty: false}});
});

test('generates diffs for entity mutations', async () => {
  const ecs = new Ecs({Components: {Position}});
  let entity;
  entity = await ecs.create({Position: {}});
  ecs.setClean();
  ecs.get(entity).Position.x = 128;
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: {x: 128}}});
  ecs.setClean();
  expect(ecs.diff)
    .to.deep.equal({});
});

test('generates coalesced diffs for components', async () => {
  const ecs = new Ecs({Components: {Position}});
  let entity;
  entity = await ecs.create({Position});
  ecs.remove(entity, ['Position']);
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: false}});
  ecs.insert(entity, {Position: {}});
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: {}}});
});

test('generates coalesced diffs for mutations', async () => {
  const ecs = new Ecs({Components: {Position}});
  let entity;
  entity = await ecs.create({Position});
  ecs.setClean();
  ecs.get(entity).Position.x = 128;
  ecs.get(entity).Position.x = 256;
  ecs.get(entity).Position.x = 512;
  expect(ecs.diff)
    .to.deep.equal({[entity]: {Position: {x: 512}}});
});

test('generates diffs for deletions', async () => {
  const ecs = new Ecs();
  let entity;
  entity = await ecs.create();
  ecs.setClean();
  ecs.destroy(entity);
  expect(ecs.diff)
    .to.deep.equal({[entity]: false});
});

test('applies creation patches', async () => {
  const ecs = new Ecs({Components: {Position}});
  await ecs.apply({16: {Position: {x: 64}}});
  expect(Array.from(ecs.entities).length)
    .to.equal(1);
  expect(ecs.get(16).Position.x)
    .to.equal(64);
});

test('applies update patches', () => {
  const ecs = new Ecs({Components: {Position}});
  ecs.createSpecific(16, {Position: {x: 64}});
  ecs.apply({16: {Position: {x: 128}}});
  expect(Array.from(ecs.entities).length)
    .to.equal(1);
  expect(ecs.get(16).Position.x)
    .to.equal(128);
});

test('applies entity deletion patches', () => {
  const ecs = new Ecs({Components: {Position}});
  ecs.createSpecific(16, {Position: {x: 64}});
  ecs.apply({16: false});
  expect(Array.from(ecs.entities).length)
    .to.equal(0);
});

test('applies component deletion patches', async () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  await ecs.createSpecific(16, {Empty: {}, Position: {x: 64}});
  expect(ecs.get(16).constructor.componentNames)
    .to.deep.equal(['Empty', 'Position']);
  await ecs.apply({16: {Empty: false}});
  expect(ecs.get(16).constructor.componentNames)
    .to.deep.equal(['Position']);
});

test('calculates entity size', () => {
  const ecs = new Ecs({Components: {Empty, Position}});
  ecs.createSpecific(1, {Empty: {}, Position: {}});
  // ID + # of components + Empty + Position + x + y + z
  // 4  + 2               + 2 + 4 + 2 + 4    + 4 + 4 + 4 = 30
  expect(ecs.get(1).size())
    .to.equal(30);
});

test('serializes and deserializes', async () => {
  const ecs = new Ecs({Components: {Empty, Name, Position}});
  ecs.createSpecific(1, {Empty: {}, Position: {x: 64}});
  ecs.createSpecific(16, {Name: {name: 'foobar'}, Position: {x: 128}});
  expect(ecs.toJSON())
    .to.deep.equal({
      entities: {
        1: {Empty: {}, Position: {x: 64}},
        16: {Name: {name: 'foobar'}, Position: {x: 128}},
      },
      systems: [],
    });
  const view = Ecs.serialize(ecs);
  const deserialized = await Ecs.deserialize(
    new Ecs({Components: {Empty, Name, Position}}),
    view,
  );
  expect(Array.from(deserialized.entities).length)
    .to.equal(2);
  expect(deserialized.get(1).constructor.componentNames)
    .to.deep.equal(['Empty', 'Position']);
  expect(deserialized.get(16).constructor.componentNames)
    .to.deep.equal(['Name', 'Position']);
  expect(JSON.stringify(deserialized.get(1)))
    .to.equal(JSON.stringify({Empty: {}, Position: {x: 64}}))
  expect(JSON.stringify(deserialized.get(1).Position))
    .to.equal(JSON.stringify({x: 64}));
  expect(JSON.stringify(deserialized.get(16).Position))
    .to.equal(JSON.stringify({x: 128}));
  expect(deserialized.get(16).Name.name)
    .to.equal('foobar');
});

test('deserializes from compatible ECS', async () => {
  const ecs = new Ecs({Components: {Empty, Name, Position}});
  ecs.createSpecific(1, {Empty: {}, Position: {x: 64}});
  ecs.createSpecific(16, {Name: {name: 'foobar'}, Position: {x: 128}});
  const view = Ecs.serialize(ecs);
  const deserialized = await Ecs.deserialize(
    new Ecs({Components: {Empty, Name}}),
    view,
  );
  expect(deserialized.get(1).toJSON())
    .to.deep.equal({Empty: {}});
  expect(deserialized.get(16).toJSON())
    .to.deep.equal({Name: {name: 'foobar'}});
});

test('creates entities asynchronously', async () => {
  const ecs = new Ecs({Components: {Async}});
  const entity = await ecs.create({Async: {foo: 64}});
  expect(ecs.get(entity).toJSON())
    .to.deep.equal({Async: {foo: 128}});
});

test('inserts components asynchronously', async () => {
  const ecs = new Ecs({Components: {Async}});
  const entity = await ecs.create();
  await ecs.insert(entity, {Async: {foo: 64}});
  expect(ecs.get(entity).toJSON())
    .to.deep.equal({Async: {foo: 128}});
});

test('deserializes asynchronously', async () => {
  const ecs = new Ecs({Components: {Async}});
  await ecs.createSpecific(16, {Async: {foo: 16}});
  const view = Ecs.serialize(ecs);
  const deserialized = await Ecs.deserialize(
    new Ecs({Components: {Async}}),
    view,
  );
  expect(deserialized.get(16).toJSON())
    .to.deep.equal({Async: {foo: 64}});
});