90 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
			
		
		
	
	
			90 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
import AxiosMockAdapter from 'axios-mock-adapter';
 | 
						|
import axios from '~/lib/utils/axios_utils';
 | 
						|
import pollUntilComplete from '~/lib/utils/poll_until_complete';
 | 
						|
import httpStatusCodes from '~/lib/utils/http_status';
 | 
						|
import { TEST_HOST } from 'helpers/test_constants';
 | 
						|
 | 
						|
const endpoint = `${TEST_HOST}/foo`;
 | 
						|
const mockData = 'mockData';
 | 
						|
const pollInterval = 1234;
 | 
						|
const pollIntervalHeader = {
 | 
						|
  'Poll-Interval': pollInterval,
 | 
						|
};
 | 
						|
 | 
						|
describe('pollUntilComplete', () => {
 | 
						|
  let mock;
 | 
						|
 | 
						|
  beforeEach(() => {
 | 
						|
    mock = new AxiosMockAdapter(axios);
 | 
						|
  });
 | 
						|
 | 
						|
  afterEach(() => {
 | 
						|
    mock.restore();
 | 
						|
  });
 | 
						|
 | 
						|
  describe('given an immediate success response', () => {
 | 
						|
    beforeEach(() => {
 | 
						|
      mock.onGet(endpoint).replyOnce(httpStatusCodes.OK, mockData);
 | 
						|
    });
 | 
						|
 | 
						|
    it('resolves with the response', () =>
 | 
						|
      pollUntilComplete(endpoint).then(({ data }) => {
 | 
						|
        expect(data).toBe(mockData);
 | 
						|
      }));
 | 
						|
  });
 | 
						|
 | 
						|
  describe(`given the endpoint returns NO_CONTENT with a Poll-Interval before succeeding`, () => {
 | 
						|
    beforeEach(() => {
 | 
						|
      mock
 | 
						|
        .onGet(endpoint)
 | 
						|
        .replyOnce(httpStatusCodes.NO_CONTENT, undefined, pollIntervalHeader)
 | 
						|
        .onGet(endpoint)
 | 
						|
        .replyOnce(httpStatusCodes.OK, mockData);
 | 
						|
    });
 | 
						|
 | 
						|
    it('calls the endpoint until it succeeds, and resolves with the response', () =>
 | 
						|
      Promise.all([
 | 
						|
        pollUntilComplete(endpoint).then(({ data }) => {
 | 
						|
          expect(data).toBe(mockData);
 | 
						|
          expect(mock.history.get).toHaveLength(2);
 | 
						|
        }),
 | 
						|
 | 
						|
        // To ensure the above pollUntilComplete() promise is actually
 | 
						|
        // fulfilled, we must explictly run the timers forward by the time
 | 
						|
        // indicated in the headers *after* each previous request has been
 | 
						|
        // fulfilled.
 | 
						|
        axios
 | 
						|
          // wait for initial NO_CONTENT response to be fulfilled
 | 
						|
          .waitForAll()
 | 
						|
          .then(() => {
 | 
						|
            jest.advanceTimersByTime(pollInterval);
 | 
						|
          }),
 | 
						|
      ]));
 | 
						|
  });
 | 
						|
 | 
						|
  describe('given the endpoint returns an error status', () => {
 | 
						|
    const errorMessage = 'error message';
 | 
						|
 | 
						|
    beforeEach(() => {
 | 
						|
      mock.onGet(endpoint).replyOnce(httpStatusCodes.NOT_FOUND, errorMessage);
 | 
						|
    });
 | 
						|
 | 
						|
    it('rejects with the error response', () =>
 | 
						|
      pollUntilComplete(endpoint).catch(error => {
 | 
						|
        expect(error.response.data).toBe(errorMessage);
 | 
						|
      }));
 | 
						|
  });
 | 
						|
 | 
						|
  describe('given params', () => {
 | 
						|
    const params = { foo: 'bar' };
 | 
						|
    beforeEach(() => {
 | 
						|
      mock.onGet(endpoint, { params }).replyOnce(httpStatusCodes.OK, mockData);
 | 
						|
    });
 | 
						|
 | 
						|
    it('requests the expected URL', () =>
 | 
						|
      pollUntilComplete(endpoint, { params }).then(({ data }) => {
 | 
						|
        expect(data).toBe(mockData);
 | 
						|
      }));
 | 
						|
  });
 | 
						|
});
 |