import MockAdapter from 'axios-mock-adapter'; import { s__ } from '~/locale'; import axios from '~/lib/utils/axios_utils'; import { resolvers } from '~/environments/graphql/resolvers'; import environmentToRollback from '~/environments/graphql/queries/environment_to_rollback.query.graphql'; import environmentToDelete from '~/environments/graphql/queries/environment_to_delete.query.graphql'; import environmentToStopQuery from '~/environments/graphql/queries/environment_to_stop.query.graphql'; import createMockApollo from 'helpers/mock_apollo_helper'; import pollIntervalQuery from '~/environments/graphql/queries/poll_interval.query.graphql'; import isEnvironmentStoppingQuery from '~/environments/graphql/queries/is_environment_stopping.query.graphql'; import pageInfoQuery from '~/environments/graphql/queries/page_info.query.graphql'; import { TEST_HOST } from 'helpers/test_constants'; import { environmentsApp, resolvedEnvironmentsApp, resolvedEnvironment, folder, resolvedFolder, } from './mock_data'; const ENDPOINT = `${TEST_HOST}/environments`; describe('~/frontend/environments/graphql/resolvers', () => { let mockResolvers; let mock; let mockApollo; let localState; beforeEach(() => { mockResolvers = resolvers(ENDPOINT); mock = new MockAdapter(axios); mockApollo = createMockApollo(); localState = mockApollo.defaultClient.localState; }); afterEach(() => { mock.reset(); }); describe('environmentApp', () => { it('should fetch environments and map them to frontend data', async () => { const cache = { writeQuery: jest.fn() }; const scope = 'available'; mock .onGet(ENDPOINT, { params: { nested: true, scope, page: 1 } }) .reply(200, environmentsApp, {}); const app = await mockResolvers.Query.environmentApp(null, { scope, page: 1 }, { cache }); expect(app).toEqual(resolvedEnvironmentsApp); expect(cache.writeQuery).toHaveBeenCalledWith({ query: pollIntervalQuery, data: { interval: undefined }, }); }); it('should set the poll interval when there is one', async () => { const cache = { writeQuery: jest.fn() }; const scope = 'stopped'; const interval = 3000; mock .onGet(ENDPOINT, { params: { nested: true, scope, page: 1 } }) .reply(200, environmentsApp, { 'poll-interval': interval, }); await mockResolvers.Query.environmentApp(null, { scope, page: 1 }, { cache }); expect(cache.writeQuery).toHaveBeenCalledWith({ query: pollIntervalQuery, data: { interval }, }); }); it('should set page info if there is any', async () => { const cache = { writeQuery: jest.fn() }; const scope = 'stopped'; mock .onGet(ENDPOINT, { params: { nested: true, scope, page: 1 } }) .reply(200, environmentsApp, { 'x-next-page': '2', 'x-page': '1', 'X-Per-Page': '2', 'X-Prev-Page': '', 'X-TOTAL': '37', 'X-Total-Pages': '5', }); await mockResolvers.Query.environmentApp(null, { scope, page: 1 }, { cache }); expect(cache.writeQuery).toHaveBeenCalledWith({ query: pageInfoQuery, data: { pageInfo: { total: 37, perPage: 2, previousPage: NaN, totalPages: 5, nextPage: 2, page: 1, __typename: 'LocalPageInfo', }, }, }); }); it('should not set page info if there is none', async () => { const cache = { writeQuery: jest.fn() }; const scope = 'stopped'; mock .onGet(ENDPOINT, { params: { nested: true, scope, page: 1 } }) .reply(200, environmentsApp, {}); await mockResolvers.Query.environmentApp(null, { scope, page: 1 }, { cache }); expect(cache.writeQuery).toHaveBeenCalledWith({ query: pageInfoQuery, data: { pageInfo: { __typename: 'LocalPageInfo', nextPage: NaN, page: NaN, perPage: NaN, previousPage: NaN, total: NaN, totalPages: NaN, }, }, }); }); }); describe('folder', () => { it('should fetch the folder url passed to it', async () => { mock.onGet(ENDPOINT, { params: { per_page: 3, scope: 'available' } }).reply(200, folder); const environmentFolder = await mockResolvers.Query.folder(null, { environment: { folderPath: ENDPOINT }, scope: 'available', }); expect(environmentFolder).toEqual(resolvedFolder); }); }); describe('stopEnvironment', () => { it('should post to the stop environment path', async () => { mock.onPost(ENDPOINT).reply(200); const client = { writeQuery: jest.fn() }; const environment = { stopPath: ENDPOINT }; await mockResolvers.Mutation.stopEnvironment(null, { environment }, { client }); expect(mock.history.post).toContainEqual( expect.objectContaining({ url: ENDPOINT, method: 'post' }), ); expect(client.writeQuery).toHaveBeenCalledWith({ query: isEnvironmentStoppingQuery, variables: { environment }, data: { isEnvironmentStopping: true }, }); }); it('should set is stopping to false if stop fails', async () => { mock.onPost(ENDPOINT).reply(500); const client = { writeQuery: jest.fn() }; const environment = { stopPath: ENDPOINT }; await mockResolvers.Mutation.stopEnvironment(null, { environment }, { client }); expect(mock.history.post).toContainEqual( expect.objectContaining({ url: ENDPOINT, method: 'post' }), ); expect(client.writeQuery).toHaveBeenCalledWith({ query: isEnvironmentStoppingQuery, variables: { environment }, data: { isEnvironmentStopping: false }, }); }); }); describe('rollbackEnvironment', () => { it('should post to the retry environment path', async () => { mock.onPost(ENDPOINT).reply(200); await mockResolvers.Mutation.rollbackEnvironment(null, { environment: { retryUrl: ENDPOINT }, }); expect(mock.history.post).toContainEqual( expect.objectContaining({ url: ENDPOINT, method: 'post' }), ); }); }); describe('deleteEnvironment', () => { it('should DELETE to the delete environment path', async () => { mock.onDelete(ENDPOINT).reply(200); await mockResolvers.Mutation.deleteEnvironment(null, { environment: { deletePath: ENDPOINT }, }); expect(mock.history.delete).toContainEqual( expect.objectContaining({ url: ENDPOINT, method: 'delete' }), ); }); }); describe('cancelAutoStop', () => { it('should post to the auto stop path', async () => { mock.onPost(ENDPOINT).reply(200); await mockResolvers.Mutation.cancelAutoStop(null, { autoStopUrl: ENDPOINT }); expect(mock.history.post).toContainEqual( expect.objectContaining({ url: ENDPOINT, method: 'post' }), ); }); }); describe('setEnvironmentToRollback', () => { it('should write the given environment to the cache', () => { localState.client.writeQuery = jest.fn(); mockResolvers.Mutation.setEnvironmentToRollback( null, { environment: resolvedEnvironment }, localState, ); expect(localState.client.writeQuery).toHaveBeenCalledWith({ query: environmentToRollback, data: { environmentToRollback: resolvedEnvironment }, }); }); }); describe('setEnvironmentToDelete', () => { it('should write the given environment to the cache', () => { localState.client.writeQuery = jest.fn(); mockResolvers.Mutation.setEnvironmentToDelete( null, { environment: resolvedEnvironment }, localState, ); expect(localState.client.writeQuery).toHaveBeenCalledWith({ query: environmentToDelete, data: { environmentToDelete: resolvedEnvironment }, }); }); }); describe('setEnvironmentToStop', () => { it('should write the given environment to the cache', () => { localState.client.writeQuery = jest.fn(); mockResolvers.Mutation.setEnvironmentToStop( null, { environment: resolvedEnvironment }, localState, ); expect(localState.client.writeQuery).toHaveBeenCalledWith({ query: environmentToStopQuery, data: { environmentToStop: resolvedEnvironment }, }); }); }); describe('action', () => { it('should POST to the given path', async () => { mock.onPost(ENDPOINT).reply(200); const errors = await mockResolvers.Mutation.action(null, { action: { playPath: ENDPOINT } }); expect(errors).toEqual({ __typename: 'LocalEnvironmentErrors', errors: [] }); }); it('should return a nice error message on fail', async () => { mock.onPost(ENDPOINT).reply(500); const errors = await mockResolvers.Mutation.action(null, { action: { playPath: ENDPOINT } }); expect(errors).toEqual({ __typename: 'LocalEnvironmentErrors', errors: [s__('Environments|An error occurred while making the request.')], }); }); }); });