61 lines
1.7 KiB
JavaScript
61 lines
1.7 KiB
JavaScript
|
import {resource} from '@avocado/resource';
|
||
|
import {Trait, traits} from '@avocado/traits';
|
||
|
import {Latus} from '@latus/core';
|
||
|
import {expect} from 'chai';
|
||
|
|
||
|
const {name} = require('../package.json');
|
||
|
|
||
|
describe(name, () => {
|
||
|
let latus;
|
||
|
let Entity;
|
||
|
beforeEach(async () => {
|
||
|
latus = Latus.mock([
|
||
|
['@avocado/entity', `${__dirname}/../src`],
|
||
|
'@avocado/resource',
|
||
|
'@avocado/traits',
|
||
|
]);
|
||
|
await Promise.all(latus.invokeFlat('@latus/core/starting'));
|
||
|
({fromResourceType: {Entity}} = resource(latus));
|
||
|
});
|
||
|
it('has sane defaults', () => {
|
||
|
const entity = new Entity();
|
||
|
expect(entity.allTraitInstances()).to.deep.equal({});
|
||
|
expect(entity.allTraitTypes()).to.deep.equal([]);
|
||
|
});
|
||
|
it('can add and remove traits', async () => {
|
||
|
const entity = new Entity();
|
||
|
const TestTrait = class extends Trait {
|
||
|
|
||
|
static get type() {
|
||
|
return 'TestTrait';
|
||
|
}
|
||
|
|
||
|
};
|
||
|
traits(latus).fromType['TestTrait'] = TestTrait;
|
||
|
const trait = await TestTrait.load();
|
||
|
entity.addTrait([TestTrait, {}]);
|
||
|
expect(entity.is('TestTrait')).to.equal(true);
|
||
|
entity.removeTrait('TestTrait');
|
||
|
expect(entity.is('TestTrait')).to.equal(false);
|
||
|
});
|
||
|
it('can add traits asynchronously', async () => {
|
||
|
const DELAY = 30;
|
||
|
class AsyncTrait extends Trait {
|
||
|
|
||
|
static async extendJson(json) {
|
||
|
const extended = await super.extendJson(json);
|
||
|
await new Promise((resolve) => setTimeout(resolve, DELAY));
|
||
|
return extended;
|
||
|
}
|
||
|
}
|
||
|
traits(latus).fromType['Async'] = AsyncTrait;
|
||
|
let start = Date.now();
|
||
|
const entity = await Entity.load({
|
||
|
traits: {
|
||
|
Async: {},
|
||
|
},
|
||
|
});
|
||
|
expect(Date.now() - start).to.be.at.least(DELAY);
|
||
|
});
|
||
|
});
|