test-assignment-datawrapper/test/app.e2e-spec.ts

import { setTimeout as wait } from "timers/promises";
import { Test, TestingModule } from '@nestjs/testing';
import { INestApplication } from '@nestjs/common';
import * as request from 'supertest';
import { AppModule } from './../src/app.module';

describe('AppController (e2e)', () => {
  let app: INestApplication;

  beforeEach(async () => {
    const moduleFixture: TestingModule = await Test.createTestingModule({
      imports: [AppModule],
    }).compile();

    app = moduleFixture.createNestApplication();
    await app.init();
  });

  afterEach(async () => {
    await app.close();
  })

  it('stores a job for suricrasia.online and completes it', async () => {
    const supertest = request(app.getHttpServer())

    const jobCreateResponse = await supertest
      .post('/screenshots')
      .send({ pageUrl: 'https://suricrasia.online', imageType: 'png' })

    expect(jobCreateResponse.status).toEqual(201)
    expect(jobCreateResponse.body.jobId).toBeDefined()
    const jobId = jobCreateResponse.body.jobId
    await wait(5000)

    const jobStatusResponse = await supertest.get(`/screenshots/${jobId}`)
    expect(jobStatusResponse.status).toEqual(200)
    expect(jobStatusResponse.body.status).toBeDefined()
    expect(jobStatusResponse.body.status).toEqual('completed')

    const jobResultResponse = await supertest.get(`/screenshots/${jobId}/result`)
    expect(jobResultResponse.status).toEqual(200)
    expect(jobResultResponse.body).toMatchImageSnapshot()
  }, 10000);
});