diff options
Diffstat (limited to 'spec/frontend/observability/client_spec.js')
-rw-r--r-- | spec/frontend/observability/client_spec.js | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/spec/frontend/observability/client_spec.js b/spec/frontend/observability/client_spec.js new file mode 100644 index 00000000000..239d7adf986 --- /dev/null +++ b/spec/frontend/observability/client_spec.js @@ -0,0 +1,66 @@ +import MockAdapter from 'axios-mock-adapter'; +import { buildClient } from '~/observability/client'; +import axios from '~/lib/utils/axios_utils'; + +jest.mock('~/lib/utils/axios_utils'); + +describe('buildClient', () => { + let client; + let axiosMock; + + const tracingUrl = 'https://example.com/tracing'; + + beforeEach(() => { + axiosMock = new MockAdapter(axios); + jest.spyOn(axios, 'get'); + + client = buildClient({ + tracingUrl, + provisioningUrl: 'https://example.com/provisioning', + }); + }); + + afterEach(() => { + axiosMock.restore(); + }); + + describe('fetchTraces', () => { + it('should fetch traces from the tracing URL', async () => { + const mockTraces = [ + { id: 1, spans: [{ duration_nano: 1000 }, { duration_nano: 2000 }] }, + { id: 2, spans: [{ duration_nano: 2000 }] }, + ]; + + axiosMock.onGet(tracingUrl).reply(200, { + traces: mockTraces, + }); + + const result = await client.fetchTraces(); + + expect(axios.get).toHaveBeenCalledTimes(1); + expect(axios.get).toHaveBeenCalledWith(tracingUrl, { + withCredentials: true, + }); + expect(result).toEqual([ + { id: 1, spans: [{ duration_nano: 1000 }, { duration_nano: 2000 }], duration: 3 }, + { id: 2, spans: [{ duration_nano: 2000 }], duration: 2 }, + ]); + }); + + it('rejects if traces are missing', () => { + axiosMock.onGet(tracingUrl).reply(200, {}); + + return expect(client.fetchTraces()).rejects.toThrow( + 'traces are missing/invalid in the response', + ); + }); + + it('rejects if traces are invalid', () => { + axiosMock.onGet(tracingUrl).reply(200, { traces: 'invalid' }); + + return expect(client.fetchTraces()).rejects.toThrow( + 'traces are missing/invalid in the response', + ); + }); + }); +}); |