import { mockDeep, mockReset } from 'jest-mock-extended'; import { PrismaService } from 'src/prisma/prisma.service'; import { TeamEnvironment } from './team-environments.model'; import { TeamEnvironmentsService } from './team-environments.service'; import { JSON_INVALID, TEAM_ENVIRONMENT_NOT_FOUND, TEAM_ENVIRONMENT_SHORT_NAME, } from 'src/errors'; const mockPrisma = mockDeep(); const mockPubSub = { publish: jest.fn().mockResolvedValue(null), }; // eslint-disable-next-line @typescript-eslint/ban-ts-comment // @ts-ignore const teamEnvironmentsService = new TeamEnvironmentsService( mockPrisma, mockPubSub as any, ); const teamEnvironment = { id: '123', name: 'test', teamID: 'abc123', variables: [{}], }; beforeEach(() => { mockReset(mockPrisma); mockPubSub.publish.mockClear(); }); describe('TeamEnvironmentsService', () => { describe('getTeamEnvironment', () => { test('should successfully return a TeamEnvironment with valid ID', async () => { mockPrisma.teamEnvironment.findFirstOrThrow.mockResolvedValueOnce( teamEnvironment, ); const result = await teamEnvironmentsService.getTeamEnvironment( teamEnvironment.id, ); expect(result).toEqualRight(teamEnvironment); }); test('should throw TEAM_ENVIRONMENT_NOT_FOUND with invalid ID', async () => { mockPrisma.teamEnvironment.findFirstOrThrow.mockRejectedValueOnce( 'RejectOnNotFound', ); const result = await teamEnvironmentsService.getTeamEnvironment( teamEnvironment.id, ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_NOT_FOUND); }); }); describe('createTeamEnvironment', () => { test('should successfully create and return a new team environment given valid inputs', async () => { mockPrisma.teamEnvironment.create.mockResolvedValue(teamEnvironment); const result = await teamEnvironmentsService.createTeamEnvironment( teamEnvironment.name, teamEnvironment.teamID, JSON.stringify(teamEnvironment.variables), ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify(teamEnvironment.variables), }); }); test('should throw TEAM_ENVIRONMENT_SHORT_NAME if input TeamEnvironment name is invalid', async () => { const result = await teamEnvironmentsService.createTeamEnvironment( '12', teamEnvironment.teamID, JSON.stringify(teamEnvironment.variables), ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_SHORT_NAME); }); test('should send pubsub message to "team_environment//created" if team environment is created successfully', async () => { mockPrisma.teamEnvironment.create.mockResolvedValue(teamEnvironment); const result = await teamEnvironmentsService.createTeamEnvironment( teamEnvironment.name, teamEnvironment.teamID, JSON.stringify(teamEnvironment.variables), ); expect(mockPubSub.publish).toHaveBeenCalledWith( `team_environment/${teamEnvironment.teamID}/created`, { ...teamEnvironment, variables: JSON.stringify(teamEnvironment.variables), }, ); }); }); describe('deleteTeamEnvironment', () => { test('should successfully delete a TeamEnvironment with a valid ID', async () => { mockPrisma.teamEnvironment.delete.mockResolvedValueOnce(teamEnvironment); const result = await teamEnvironmentsService.deleteTeamEnvironment( teamEnvironment.id, ); expect(result).toEqualRight(true); }); test('should throw TEAM_ENVIRONMMENT_NOT_FOUND if given id is invalid', async () => { mockPrisma.teamEnvironment.delete.mockRejectedValue('RecordNotFound'); const result = await teamEnvironmentsService.deleteTeamEnvironment( 'invalidid', ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_NOT_FOUND); }); test('should send pubsub message to "team_environment//deleted" if team environment is deleted successfully', async () => { mockPrisma.teamEnvironment.delete.mockResolvedValueOnce(teamEnvironment); const result = await teamEnvironmentsService.deleteTeamEnvironment( teamEnvironment.id, ); expect(mockPubSub.publish).toHaveBeenCalledWith( `team_environment/${teamEnvironment.teamID}/deleted`, { ...teamEnvironment, variables: JSON.stringify(teamEnvironment.variables), }, ); }); }); describe('updateVariablesInTeamEnvironment', () => { test('should successfully add new variable to a team environment', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce({ ...teamEnvironment, variables: [{ key: 'value' }], }); const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, teamEnvironment.name, JSON.stringify([{ key: 'value' }]), ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify([{ key: 'value' }]), }); }); test('should successfully add new variable to already existing list of variables in a team environment', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce({ ...teamEnvironment, variables: [{ key: 'value' }, { key_2: 'value_2' }], }); const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, teamEnvironment.name, JSON.stringify([{ key: 'value' }, { key_2: 'value_2' }]), ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify([{ key: 'value' }, { key_2: 'value_2' }]), }); }); test('should successfully edit existing variables in a team environment', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce({ ...teamEnvironment, variables: [{ key: '1234' }], }); const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, teamEnvironment.name, JSON.stringify([{ key: '1234' }]), ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify([{ key: '1234' }]), }); }); test('should successfully edit name of an existing team environment', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce({ ...teamEnvironment, variables: [{ key: '123' }], }); const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, teamEnvironment.name, JSON.stringify([{ key: '123' }]), ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify([{ key: '123' }]), }); }); test('should throw TEAM_ENVIRONMENT_SHORT_NAME if input TeamEnvironment name is invalid', async () => { const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, '12', JSON.stringify([{ key: 'value' }]), ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_SHORT_NAME); }); test('should throw TEAM_ENVIRONMMENT_NOT_FOUND if provided id is invalid', async () => { mockPrisma.teamEnvironment.update.mockRejectedValue('RecordNotFound'); const result = await teamEnvironmentsService.updateTeamEnvironment( 'invalidid', teamEnvironment.name, JSON.stringify(teamEnvironment.variables), ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_NOT_FOUND); }); test('should send pubsub message to "team_environment//updated" if team environment is updated successfully', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce(teamEnvironment); const result = await teamEnvironmentsService.updateTeamEnvironment( teamEnvironment.id, teamEnvironment.name, JSON.stringify([{ key: 'value' }]), ); expect(mockPubSub.publish).toHaveBeenCalledWith( `team_environment/${teamEnvironment.teamID}/updated`, { ...teamEnvironment, variables: JSON.stringify(teamEnvironment.variables), }, ); }); }); describe('deleteAllVariablesFromTeamEnvironment', () => { test('should successfully delete all variables in a team environment', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce(teamEnvironment); const result = await teamEnvironmentsService.deleteAllVariablesFromTeamEnvironment( teamEnvironment.id, ); expect(result).toEqualRight({ ...teamEnvironment, variables: JSON.stringify([{}]), }); }); test('should throw TEAM_ENVIRONMMENT_NOT_FOUND if provided id is invalid', async () => { mockPrisma.teamEnvironment.update.mockRejectedValue('RecordNotFound'); const result = await teamEnvironmentsService.deleteAllVariablesFromTeamEnvironment( 'invalidid', ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_NOT_FOUND); }); test('should send pubsub message to "team_environment//updated" if team environment is updated successfully', async () => { mockPrisma.teamEnvironment.update.mockResolvedValueOnce(teamEnvironment); const result = await teamEnvironmentsService.deleteAllVariablesFromTeamEnvironment( teamEnvironment.id, ); expect(mockPubSub.publish).toHaveBeenCalledWith( `team_environment/${teamEnvironment.teamID}/updated`, { ...teamEnvironment, variables: JSON.stringify([{}]), }, ); }); }); describe('createDuplicateEnvironment', () => { test('should successfully duplicate an existing team environment', async () => { mockPrisma.teamEnvironment.findFirstOrThrow.mockResolvedValueOnce( teamEnvironment, ); mockPrisma.teamEnvironment.create.mockResolvedValueOnce({ id: 'newid', ...teamEnvironment, }); const result = await teamEnvironmentsService.createDuplicateEnvironment( teamEnvironment.id, ); expect(result).toEqualRight({ id: 'newid', ...teamEnvironment, variables: JSON.stringify(teamEnvironment.variables), }); }); test('should throw TEAM_ENVIRONMMENT_NOT_FOUND if provided id is invalid', async () => { mockPrisma.teamEnvironment.findFirstOrThrow.mockRejectedValue( 'NotFoundError', ); const result = await teamEnvironmentsService.createDuplicateEnvironment( teamEnvironment.id, ); expect(result).toEqualLeft(TEAM_ENVIRONMENT_NOT_FOUND); }); test('should send pubsub message to "team_environment//created" if team environment is updated successfully', async () => { mockPrisma.teamEnvironment.findFirstOrThrow.mockResolvedValueOnce( teamEnvironment, ); mockPrisma.teamEnvironment.create.mockResolvedValueOnce({ id: 'newid', ...teamEnvironment, }); const result = await teamEnvironmentsService.createDuplicateEnvironment( teamEnvironment.id, ); expect(mockPubSub.publish).toHaveBeenCalledWith( `team_environment/${teamEnvironment.teamID}/created`, { id: 'newid', ...teamEnvironment, variables: JSON.stringify([{}]), }, ); }); }); describe('totalEnvsInTeam', () => { test('should resolve right and return a total team envs count ', async () => { mockPrisma.teamEnvironment.count.mockResolvedValueOnce(2); const result = await teamEnvironmentsService.totalEnvsInTeam('id1'); expect(mockPrisma.teamEnvironment.count).toHaveBeenCalledWith({ where: { teamID: 'id1', }, }); expect(result).toEqual(2); }); test('should resolve left and return an error when no team envs found', async () => { mockPrisma.teamEnvironment.count.mockResolvedValueOnce(0); const result = await teamEnvironmentsService.totalEnvsInTeam('id1'); expect(mockPrisma.teamEnvironment.count).toHaveBeenCalledWith({ where: { teamID: 'id1', }, }); expect(result).toEqual(0); }); }); });