datahub/datahub-web/@datahub/data-models/tests/unit/utils/entity/dataset/dataset-entity-test.ts

import { module, test } from 'qunit';
import { setupTest } from 'ember-qunit';
import { DatasetEntity } from '@datahub/data-models/entity/dataset/dataset-entity';
import sinonTest from 'ember-sinon-qunit/test-support/test';
import Sinon from 'sinon';
import { BaseEntity } from '@datahub/data-models/entity/base-entity';
import { mockDatasetEntity } from 'dummy/constants/mocks/dataset';

module('Unit | Utility | entity/dataset/dataset-entity', function(hooks) {
  setupTest(hooks);

  test('DatasetEntity class', function(assert) {
    const entity = new DatasetEntity('squirtle_urn');
    assert.ok(entity instanceof BaseEntity, 'Expects entity to be an instance of the BaseEntity');
    assert.equal(entity.displayName, 'datasets', 'expected entity display name to be Datasets');
    assert.equal(entity.kind, 'DatasetEntity', 'expected entity kind to be DatasetEntity');
    assert.equal(entity.urn, 'squirtle_urn', 'expected entity urn to match supplied value squirtle_urn');
  });

  sinonTest('DatasetEntity readEntity', async function(this: SinonTestContext, assert) {
    assert.expect(1);

    const testUrn = 'bulbasaur_urn';
    const testEntity = mockDatasetEntity({ uri: testUrn });
    const requester = this.sandbox.useFakeXMLHttpRequest();
    let request: Sinon.SinonFakeXMLHttpRequest | undefined;

    requester.onCreate = req => {
      request = req;
    };

    const entityRequest = new DatasetEntity(testUrn).readEntity;
    request && request.respond(200, { 'Content-Type': 'application/json' }, JSON.stringify({ dataset: testEntity }));
    const entity = await entityRequest;

    assert.equal(entity.uri, testUrn, 'expected entity urn to match requested urn');
    requester.restore();
  });
});