|
| 1 | +import * as sinon from "sinon"; |
| 2 | +import * as chai from "chai"; |
| 3 | +import fetchModule from '../transport/fetch'; |
| 4 | +import { getWithGlobals } from '../transport/request'; |
| 5 | + |
| 6 | +describe("[REST] Stocks", () => { |
| 7 | + chai.should(); |
| 8 | + const mocks = { |
| 9 | + key: 'invalid', |
| 10 | + base: 'https://test.api.polygon.io', |
| 11 | + query: { query1: 'queryVal' }, |
| 12 | + overrideOptions: { |
| 13 | + referrer: 'overrideVal' |
| 14 | + }, |
| 15 | + globalOptions: { |
| 16 | + referrer: 'globalVal1', |
| 17 | + headers: { |
| 18 | + header1: 'headerVal1' |
| 19 | + } |
| 20 | + } |
| 21 | + }; |
| 22 | + let fetchStub; |
| 23 | + const sandbox = sinon.createSandbox(); |
| 24 | + const getPath = (url) => url.slice(mocks.base.length, url.indexOf('?')); |
| 25 | + const setStub = (returnVal, status = 200) => { |
| 26 | + fetchStub?.restore(); |
| 27 | + fetchStub = sandbox.stub(fetchModule, 'fetch').returns( |
| 28 | + Promise.resolve({ json: () => Promise.resolve(returnVal), status } as Response) |
| 29 | + ); |
| 30 | + } |
| 31 | + const setStubs = (returnVal1, returnVal2, status = 200) => { |
| 32 | + fetchStub?.restore(); |
| 33 | + fetchStub = sandbox.stub(fetchModule, 'fetch').onCall(0).returns( |
| 34 | + Promise.resolve({ json: () => Promise.resolve(returnVal1), status } as Response) |
| 35 | + ).onCall(1).returns( |
| 36 | + Promise.resolve({ json: () => Promise.resolve(returnVal2), status } as Response) |
| 37 | + ); |
| 38 | + } |
| 39 | + beforeEach(async () => { |
| 40 | + setStub({ ticks: [], results: [], tickers: [] }); |
| 41 | + }); |
| 42 | + afterEach(() => { |
| 43 | + sandbox.restore(); |
| 44 | + }); |
| 45 | + |
| 46 | + it("Returns the expected result when pagination is not enabled", async () => { |
| 47 | + setStub({ results: [{ a: 1 }] }); |
| 48 | + const requestFn = getWithGlobals(mocks.key, mocks.base, mocks.globalOptions); |
| 49 | + const response = await requestFn("/v2/aggs/ticker/AAPL/range/1/day/2019-01-01/2019-02-01", mocks.query, mocks.overrideOptions); |
| 50 | + response.results[0].a.should.eql(1); |
| 51 | + fetchStub.callCount.should.eql(1); |
| 52 | + getPath(fetchStub.getCalls()[0].args[0]).should.eql( |
| 53 | + "/v2/aggs/ticker/AAPL/range/1/day/2019-01-01/2019-02-01" |
| 54 | + ); |
| 55 | + fetchStub.getCalls()[0].args[0].indexOf(mocks.query.query1).should.be.gt(-1); |
| 56 | + fetchStub.getCalls()[0].args[1].referrer.should.eql(mocks.overrideOptions.referrer); |
| 57 | + fetchStub.getCalls()[0].args[1].headers.header1.should.eql(mocks.globalOptions.headers.header1); |
| 58 | + }); |
| 59 | + |
| 60 | + |
| 61 | + it("Returns the expected result when pagination is enabled", async () => { |
| 62 | + setStubs({ results: [{ a: 1 }], next_url: "/v2/aggs/ticker/AAPL/range/1/day/2019-01-01/2019-02-02" }, { results: [{ b: 2 }], }); |
| 63 | + const requestFn = getWithGlobals(mocks.key, mocks.base, {...mocks.globalOptions, pagination: true}); |
| 64 | + const response = await requestFn("/v2/aggs/ticker/AAPL/range/1/day/2019-01-01/2019-02-01", mocks.query, mocks.overrideOptions); |
| 65 | + response.results.length.should.eql(2); |
| 66 | + response.results[0].a.should.eql(1); |
| 67 | + response.results[1].b.should.eql(2); |
| 68 | + fetchStub.callCount.should.eql(2); |
| 69 | + getPath(fetchStub.getCalls()[0].args[0]).should.eql( |
| 70 | + "/v2/aggs/ticker/AAPL/range/1/day/2019-01-01/2019-02-01" |
| 71 | + ); |
| 72 | + fetchStub.getCalls()[0].args[0].indexOf(mocks.query.query1).should.be.gt(-1); |
| 73 | + fetchStub.getCalls()[0].args[1].referrer.should.eql(mocks.overrideOptions.referrer); |
| 74 | + fetchStub.getCalls()[0].args[1].headers.header1.should.eql(mocks.globalOptions.headers.header1); |
| 75 | + }); |
| 76 | +}); |
0 commit comments