open-nomad/ui/tests/unit/models/task-group-test.js

import { get } from '@ember/object';
import { module, test } from 'qunit';
import { setupTest } from 'ember-qunit';

import { run } from '@ember/runloop';

const sum = (list, key) => list.reduce((sum, item) => sum + get(item, key), 0);

module('Unit | Model | task-group', function(hooks) {
  setupTest(hooks);

  test("should expose reserved resource stats as aggregates of each task's reserved resources", function(assert) {
    const taskGroup = run(() => this.owner.lookup('service:store').createRecord('task-group', {
      name: 'group-example',
      tasks: [
        {
          name: 'task-one',
          driver: 'docker',
          reservedMemory: 512,
          reservedCPU: 500,
          reservedDisk: 1024,
        },
        {
          name: 'task-two',
          driver: 'docker',
          reservedMemory: 256,
          reservedCPU: 1000,
          reservedDisk: 512,
        },
        {
          name: 'task-three',
          driver: 'docker',
          reservedMemory: 1024,
          reservedCPU: 1500,
          reservedDisk: 4096,
        },
        {
          name: 'task-four',
          driver: 'docker',
          reservedMemory: 2048,
          reservedCPU: 500,
          reservedDisk: 128,
        },
      ],
    }));

    assert.equal(
      taskGroup.get('reservedCPU'),
      sum(taskGroup.get('tasks'), 'reservedCPU'),
      'reservedCPU is an aggregate sum of task CPU reservations'
    );
    assert.equal(
      taskGroup.get('reservedMemory'),
      sum(taskGroup.get('tasks'), 'reservedMemory'),
      'reservedMemory is an aggregate sum of task memory reservations'
    );
    assert.equal(
      taskGroup.get('reservedDisk'),
      sum(taskGroup.get('tasks'), 'reservedDisk'),
      'reservedDisk is an aggregate sum of task disk reservations'
    );
  });
});