open-nomad/ui/tests/unit/serializers/job-plan-test.js

import { module, test } from 'qunit';
import { setupTest } from 'ember-qunit';
import JobPlanModel from 'nomad-ui/models/job-plan';

module('Unit | Serializer | JobPlan', function (hooks) {
  setupTest(hooks);
  hooks.beforeEach(function () {
    this.store = this.owner.lookup('service:store');
    this.subject = () => this.store.serializerFor('job-plan');
  });

  const normalizationTestCases = [
    {
      name: 'Normal',
      in: {
        ID: 'test-plan',
        Diff: {
          Arbitrary: 'Value',
        },
        FailedTGAllocs: {
          taskGroup: {
            NodesAvailable: 10,
          },
        },
      },
      out: {
        data: {
          id: 'test-plan',
          type: 'job-plan',
          attributes: {
            diff: {
              Arbitrary: 'Value',
            },
            failedTGAllocs: [
              {
                name: 'taskGroup',
                nodesAvailable: 10,
              },
            ],
          },
          relationships: {
            preemptions: {
              data: [],
            },
          },
        },
      },
    },

    {
      name: 'Dots in task names',
      in: {
        ID: 'test-plan',
        Diff: {
          Arbitrary: 'Value',
        },
        FailedTGAllocs: {
          'one.two': {
            NodesAvailable: 10,
          },
          'three.four': {
            NodesAvailable: 25,
          },
        },
      },
      out: {
        data: {
          id: 'test-plan',
          type: 'job-plan',
          attributes: {
            diff: {
              Arbitrary: 'Value',
            },
            failedTGAllocs: [
              {
                name: 'one.two',
                nodesAvailable: 10,
              },
              {
                name: 'three.four',
                nodesAvailable: 25,
              },
            ],
          },
          relationships: {
            preemptions: {
              data: [],
            },
          },
        },
      },
    },

    {
      name: 'With preemptions',
      in: {
        ID: 'test-plan',
        Diff: {
          Arbitrary: 'Value',
        },
        FailedTGAllocs: {
          task: {
            NodesAvailable: 10,
          },
        },
        Annotations: {
          PreemptedAllocs: [
            { ID: 'preemption-one-allocation' },
            { ID: 'preemption-two-allocation' },
          ],
        },
      },
      out: {
        data: {
          id: 'test-plan',
          type: 'job-plan',
          attributes: {
            diff: {
              Arbitrary: 'Value',
            },
            failedTGAllocs: [
              {
                name: 'task',
                nodesAvailable: 10,
              },
            ],
          },
          relationships: {
            preemptions: {
              data: [
                { id: 'preemption-one-allocation', type: 'allocation' },
                { id: 'preemption-two-allocation', type: 'allocation' },
              ],
            },
          },
        },
      },
    },
  ];

  normalizationTestCases.forEach((testCase) => {
    test(`normalization: ${testCase.name}`, async function (assert) {
      assert.deepEqual(
        this.subject().normalize(JobPlanModel, testCase.in),
        testCase.out
      );
    });
  });
});