gitlab-org--gitlab-foss/spec/javascripts/lib/utils/poll_spec.js

176 lines
4.7 KiB
JavaScript
Raw Normal View History

2017-03-21 14:07:02 -04:00
import Poll from '~/lib/utils/poll';
const waitForAllCallsToFinish = (service, waitForCount, successCallback) => {
const timer = () => {
setTimeout(() => {
if (service.fetch.calls.count() === waitForCount) {
successCallback();
} else {
timer();
}
}, 0);
};
timer();
};
function mockServiceCall(service, response, shouldFail = false) {
const action = shouldFail ? Promise.reject : Promise.resolve;
const responseObject = response;
if (!responseObject.headers) responseObject.headers = {};
service.fetch.and.callFake(action.bind(Promise, responseObject));
2017-03-21 14:07:02 -04:00
}
describe('Poll', () => {
const service = jasmine.createSpyObj('service', ['fetch']);
const callbacks = jasmine.createSpyObj('callbacks', ['success', 'error']);
2017-03-21 14:07:02 -04:00
afterEach(() => {
callbacks.success.calls.reset();
callbacks.error.calls.reset();
service.fetch.calls.reset();
2017-03-21 14:07:02 -04:00
});
it('calls the success callback when no header for interval is provided', (done) => {
mockServiceCall(service, { status: 200 });
2017-03-21 14:07:02 -04:00
new Poll({
resource: service,
2017-03-21 14:07:02 -04:00
method: 'fetch',
successCallback: callbacks.success,
errorCallback: callbacks.error,
}).makeRequest();
waitForAllCallsToFinish(service, 1, () => {
2017-03-21 14:07:02 -04:00
expect(callbacks.success).toHaveBeenCalled();
expect(callbacks.error).not.toHaveBeenCalled();
2017-03-21 14:07:02 -04:00
done();
});
2017-03-21 14:07:02 -04:00
});
it('calls the error callback whe the http request returns an error', (done) => {
mockServiceCall(service, { status: 500 }, true);
2017-03-21 14:07:02 -04:00
new Poll({
resource: service,
2017-03-21 14:07:02 -04:00
method: 'fetch',
successCallback: callbacks.success,
errorCallback: callbacks.error,
}).makeRequest();
waitForAllCallsToFinish(service, 1, () => {
2017-03-21 14:07:02 -04:00
expect(callbacks.success).not.toHaveBeenCalled();
expect(callbacks.error).toHaveBeenCalled();
done();
});
2017-03-21 14:07:02 -04:00
});
it('should call the success callback when the interval header is -1', (done) => {
mockServiceCall(service, { status: 200, headers: { 'poll-interval': -1 } });
2017-03-21 14:07:02 -04:00
new Poll({
resource: service,
2017-03-21 14:07:02 -04:00
method: 'fetch',
successCallback: callbacks.success,
errorCallback: callbacks.error,
}).makeRequest().then(() => {
2017-03-21 14:07:02 -04:00
expect(callbacks.success).toHaveBeenCalled();
expect(callbacks.error).not.toHaveBeenCalled();
2017-03-21 14:07:02 -04:00
done();
}).catch(done.fail);
2017-03-21 14:07:02 -04:00
});
it('starts polling when http status is 200 and interval header is provided', (done) => {
mockServiceCall(service, { status: 200, headers: { 'poll-interval': 1 } });
2017-03-21 14:07:02 -04:00
const Polling = new Poll({
2017-03-21 14:07:02 -04:00
resource: service,
method: 'fetch',
data: { page: 1 },
2017-03-21 14:07:02 -04:00
successCallback: callbacks.success,
errorCallback: callbacks.error,
});
Polling.makeRequest();
waitForAllCallsToFinish(service, 2, () => {
Polling.stop();
2017-03-21 14:07:02 -04:00
expect(service.fetch.calls.count()).toEqual(2);
expect(service.fetch).toHaveBeenCalledWith({ page: 1 });
2017-03-21 14:07:02 -04:00
expect(callbacks.success).toHaveBeenCalled();
expect(callbacks.error).not.toHaveBeenCalled();
done();
});
2017-03-21 14:07:02 -04:00
});
describe('stop', () => {
it('stops polling when method is called', (done) => {
mockServiceCall(service, { status: 200, headers: { 'poll-interval': 1 } });
const Polling = new Poll({
resource: service,
method: 'fetch',
data: { page: 1 },
successCallback: () => {
Polling.stop();
},
errorCallback: callbacks.error,
});
spyOn(Polling, 'stop').and.callThrough();
Polling.makeRequest();
waitForAllCallsToFinish(service, 1, () => {
expect(service.fetch.calls.count()).toEqual(1);
expect(service.fetch).toHaveBeenCalledWith({ page: 1 });
expect(Polling.stop).toHaveBeenCalled();
done();
});
});
});
describe('restart', () => {
it('should restart polling when its called', (done) => {
mockServiceCall(service, { status: 200, headers: { 'poll-interval': 1 } });
const Polling = new Poll({
resource: service,
method: 'fetch',
data: { page: 1 },
successCallback: () => {
Polling.stop();
setTimeout(() => {
Polling.restart();
}, 0);
},
errorCallback: callbacks.error,
});
spyOn(Polling, 'stop').and.callThrough();
spyOn(Polling, 'restart').and.callThrough();
Polling.makeRequest();
waitForAllCallsToFinish(service, 2, () => {
Polling.stop();
expect(service.fetch.calls.count()).toEqual(2);
expect(service.fetch).toHaveBeenCalledWith({ page: 1 });
expect(Polling.stop).toHaveBeenCalled();
expect(Polling.restart).toHaveBeenCalled();
done();
});
});
});
2017-03-21 14:07:02 -04:00
});