backend unit test coverage to 80%

This commit is contained in:
jackiettran
2026-01-19 19:22:01 -05:00
parent d4362074f5
commit 1923ffc251
8 changed files with 3183 additions and 7 deletions

View File

@@ -9,6 +9,8 @@ const mockStripeRefundsRetrieve = jest.fn();
const mockStripePaymentIntentsCreate = jest.fn();
const mockStripeCustomersCreate = jest.fn();
const mockStripeCheckoutSessionsCreate = jest.fn();
const mockStripeAccountSessionsCreate = jest.fn();
const mockStripePaymentMethodsRetrieve = jest.fn();
jest.mock('stripe', () => {
return jest.fn(() => ({
@@ -25,6 +27,9 @@ jest.mock('stripe', () => {
accountLinks: {
create: mockStripeAccountLinksCreate
},
accountSessions: {
create: mockStripeAccountSessionsCreate
},
transfers: {
create: mockStripeTransfersCreate
},
@@ -37,15 +42,20 @@ jest.mock('stripe', () => {
},
customers: {
create: mockStripeCustomersCreate
},
paymentMethods: {
retrieve: mockStripePaymentMethodsRetrieve
}
}));
});
const mockLoggerError = jest.fn();
const mockLoggerWarn = jest.fn();
const mockLoggerInfo = jest.fn();
jest.mock('../../../utils/logger', () => ({
error: mockLoggerError,
info: jest.fn(),
warn: jest.fn(),
info: mockLoggerInfo,
warn: mockLoggerWarn,
withRequestId: jest.fn(() => ({
error: jest.fn(),
info: jest.fn(),
@@ -53,6 +63,23 @@ jest.mock('../../../utils/logger', () => ({
})),
}));
// Mock User model
const mockUserFindOne = jest.fn();
const mockUserUpdate = jest.fn();
jest.mock('../../../models', () => ({
User: {
findOne: mockUserFindOne
}
}));
// Mock email services
const mockSendAccountDisconnectedEmail = jest.fn();
jest.mock('../../../services/email', () => ({
payment: {
sendAccountDisconnectedEmail: mockSendAccountDisconnectedEmail
}
}));
const StripeService = require('../../../services/stripeService');
describe('StripeService', () => {
@@ -1158,4 +1185,500 @@ describe('StripeService', () => {
);
});
});
describe('createAccountSession', () => {
it('should create account session successfully', async () => {
const mockSession = {
object: 'account_session',
client_secret: 'acct_sess_secret_123',
expires_at: Date.now() + 3600
};
mockStripeAccountSessionsCreate.mockResolvedValue(mockSession);
const result = await StripeService.createAccountSession('acct_123456789');
expect(mockStripeAccountSessionsCreate).toHaveBeenCalledWith({
account: 'acct_123456789',
components: {
account_onboarding: { enabled: true }
}
});
expect(result).toEqual(mockSession);
});
it('should handle account session creation errors', async () => {
const stripeError = new Error('Account not found');
mockStripeAccountSessionsCreate.mockRejectedValue(stripeError);
await expect(StripeService.createAccountSession('invalid_account'))
.rejects.toThrow('Account not found');
expect(mockLoggerError).toHaveBeenCalledWith(
'Error creating account session',
expect.objectContaining({
error: stripeError.message,
})
);
});
it('should handle invalid account ID', async () => {
const stripeError = new Error('Invalid account ID format');
mockStripeAccountSessionsCreate.mockRejectedValue(stripeError);
await expect(StripeService.createAccountSession(null))
.rejects.toThrow('Invalid account ID format');
});
});
describe('getPaymentMethod', () => {
it('should retrieve payment method successfully', async () => {
const mockPaymentMethod = {
id: 'pm_123456789',
type: 'card',
card: {
brand: 'visa',
last4: '4242',
exp_month: 12,
exp_year: 2025
},
customer: 'cus_123456789'
};
mockStripePaymentMethodsRetrieve.mockResolvedValue(mockPaymentMethod);
const result = await StripeService.getPaymentMethod('pm_123456789');
expect(mockStripePaymentMethodsRetrieve).toHaveBeenCalledWith('pm_123456789');
expect(result).toEqual(mockPaymentMethod);
});
it('should handle payment method retrieval errors', async () => {
const stripeError = new Error('Payment method not found');
mockStripePaymentMethodsRetrieve.mockRejectedValue(stripeError);
await expect(StripeService.getPaymentMethod('pm_invalid'))
.rejects.toThrow('Payment method not found');
expect(mockLoggerError).toHaveBeenCalledWith(
'Error retrieving payment method',
expect.objectContaining({
error: stripeError.message,
paymentMethodId: 'pm_invalid'
})
);
});
it('should handle null payment method ID', async () => {
const stripeError = new Error('Invalid payment method ID');
mockStripePaymentMethodsRetrieve.mockRejectedValue(stripeError);
await expect(StripeService.getPaymentMethod(null))
.rejects.toThrow('Invalid payment method ID');
});
});
describe('isAccountDisconnectedError', () => {
it('should return true for account_invalid error code', () => {
const error = { code: 'account_invalid', message: 'Account is invalid' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return true for platform_api_key_expired error code', () => {
const error = { code: 'platform_api_key_expired', message: 'API key expired' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return true for error message containing "cannot transfer"', () => {
const error = { code: 'some_code', message: 'You cannot transfer to this account' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return true for error message containing "not connected"', () => {
const error = { code: 'some_code', message: 'This account is not connected to your platform' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return true for error message containing "no longer connected"', () => {
const error = { code: 'some_code', message: 'This account is no longer connected' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return true for error message containing "account has been deauthorized"', () => {
const error = { code: 'some_code', message: 'The account has been deauthorized' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(true);
});
it('should return false for unrelated error codes', () => {
const error = { code: 'card_declined', message: 'Card was declined' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(false);
});
it('should return false for unrelated error messages', () => {
const error = { code: 'some_code', message: 'Insufficient funds in account' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(false);
});
it('should handle error with no message', () => {
const error = { code: 'some_code' };
expect(StripeService.isAccountDisconnectedError(error)).toBe(false);
});
it('should handle error with undefined message', () => {
const error = { code: 'some_code', message: undefined };
expect(StripeService.isAccountDisconnectedError(error)).toBe(false);
});
});
describe('handleDisconnectedAccount', () => {
beforeEach(() => {
mockUserFindOne.mockReset();
mockSendAccountDisconnectedEmail.mockReset();
});
it('should clear user stripe connection data', async () => {
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: 'John',
lastName: 'Doe',
update: mockUserUpdate.mockResolvedValue(true)
};
mockUserFindOne.mockResolvedValue(mockUser);
mockSendAccountDisconnectedEmail.mockResolvedValue(true);
await StripeService.handleDisconnectedAccount('acct_123456789');
expect(mockUserFindOne).toHaveBeenCalledWith({
where: { stripeConnectedAccountId: 'acct_123456789' }
});
expect(mockUserUpdate).toHaveBeenCalledWith({
stripeConnectedAccountId: null,
stripePayoutsEnabled: false
});
});
it('should send account disconnected email', async () => {
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: 'John',
lastName: 'Doe',
update: mockUserUpdate.mockResolvedValue(true)
};
mockUserFindOne.mockResolvedValue(mockUser);
mockSendAccountDisconnectedEmail.mockResolvedValue(true);
await StripeService.handleDisconnectedAccount('acct_123456789');
expect(mockSendAccountDisconnectedEmail).toHaveBeenCalledWith('test@example.com', {
ownerName: 'John',
hasPendingPayouts: true,
pendingPayoutCount: 1
});
});
it('should do nothing when user not found', async () => {
mockUserFindOne.mockResolvedValue(null);
await StripeService.handleDisconnectedAccount('acct_nonexistent');
expect(mockUserFindOne).toHaveBeenCalledWith({
where: { stripeConnectedAccountId: 'acct_nonexistent' }
});
expect(mockUserUpdate).not.toHaveBeenCalled();
expect(mockSendAccountDisconnectedEmail).not.toHaveBeenCalled();
});
it('should handle email sending errors gracefully', async () => {
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: 'John',
lastName: 'Doe',
update: mockUserUpdate.mockResolvedValue(true)
};
mockUserFindOne.mockResolvedValue(mockUser);
mockSendAccountDisconnectedEmail.mockRejectedValue(new Error('Email service down'));
// Should not throw
await expect(StripeService.handleDisconnectedAccount('acct_123456789'))
.resolves.not.toThrow();
// Should have logged the error
expect(mockLoggerError).toHaveBeenCalledWith(
'Failed to clean up disconnected account',
expect.objectContaining({
accountId: 'acct_123456789'
})
);
});
it('should handle user update errors gracefully', async () => {
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: 'John',
lastName: 'Doe',
update: jest.fn().mockRejectedValue(new Error('Database error'))
};
mockUserFindOne.mockResolvedValue(mockUser);
// Should not throw
await expect(StripeService.handleDisconnectedAccount('acct_123456789'))
.resolves.not.toThrow();
expect(mockLoggerError).toHaveBeenCalledWith(
'Failed to clean up disconnected account',
expect.objectContaining({
accountId: 'acct_123456789'
})
);
});
it('should use lastName as fallback when firstName is not available', async () => {
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: null,
lastName: 'Doe',
update: mockUserUpdate.mockResolvedValue(true)
};
mockUserFindOne.mockResolvedValue(mockUser);
mockSendAccountDisconnectedEmail.mockResolvedValue(true);
await StripeService.handleDisconnectedAccount('acct_123456789');
expect(mockSendAccountDisconnectedEmail).toHaveBeenCalledWith('test@example.com', {
ownerName: 'Doe',
hasPendingPayouts: true,
pendingPayoutCount: 1
});
});
});
describe('createTransfer - disconnected account handling', () => {
beforeEach(() => {
mockUserFindOne.mockReset();
mockSendAccountDisconnectedEmail.mockReset();
mockLoggerWarn.mockReset();
});
it('should call handleDisconnectedAccount when account_invalid error occurs', async () => {
const disconnectedError = new Error('The account has been deauthorized');
disconnectedError.code = 'account_invalid';
mockStripeTransfersCreate.mockRejectedValue(disconnectedError);
const mockUser = {
id: 123,
email: 'test@example.com',
firstName: 'John',
update: mockUserUpdate.mockResolvedValue(true)
};
mockUserFindOne.mockResolvedValue(mockUser);
mockSendAccountDisconnectedEmail.mockResolvedValue(true);
await expect(StripeService.createTransfer({
amount: 50.00,
destination: 'acct_disconnected'
})).rejects.toThrow('The account has been deauthorized');
// Wait for async handleDisconnectedAccount to complete
await new Promise(resolve => setTimeout(resolve, 10));
expect(mockLoggerWarn).toHaveBeenCalledWith(
'Transfer failed - account appears disconnected',
expect.objectContaining({
destination: 'acct_disconnected',
errorCode: 'account_invalid'
})
);
});
it('should still throw the original error after cleanup', async () => {
const disconnectedError = new Error('Cannot transfer to this account');
disconnectedError.code = 'account_invalid';
mockStripeTransfersCreate.mockRejectedValue(disconnectedError);
mockUserFindOne.mockResolvedValue(null);
await expect(StripeService.createTransfer({
amount: 50.00,
destination: 'acct_disconnected'
})).rejects.toThrow('Cannot transfer to this account');
});
it('should log warning for disconnected account errors', async () => {
const disconnectedError = new Error('This account is no longer connected');
disconnectedError.code = 'some_error';
disconnectedError.type = 'StripeInvalidRequestError';
mockStripeTransfersCreate.mockRejectedValue(disconnectedError);
mockUserFindOne.mockResolvedValue(null);
await expect(StripeService.createTransfer({
amount: 50.00,
destination: 'acct_disconnected'
})).rejects.toThrow('This account is no longer connected');
expect(mockLoggerWarn).toHaveBeenCalledWith(
'Transfer failed - account appears disconnected',
expect.objectContaining({
destination: 'acct_disconnected'
})
);
});
it('should not call handleDisconnectedAccount for non-disconnection errors', async () => {
const normalError = new Error('Insufficient balance');
normalError.code = 'insufficient_balance';
mockStripeTransfersCreate.mockRejectedValue(normalError);
await expect(StripeService.createTransfer({
amount: 50.00,
destination: 'acct_123'
})).rejects.toThrow('Insufficient balance');
expect(mockLoggerWarn).not.toHaveBeenCalledWith(
'Transfer failed - account appears disconnected',
expect.any(Object)
);
});
});
describe('chargePaymentMethod - additional cases', () => {
it('should handle authentication_required error and return requires_action', async () => {
const authError = new Error('Authentication required');
authError.code = 'authentication_required';
authError.payment_intent = {
id: 'pi_requires_auth',
client_secret: 'pi_requires_auth_secret'
};
mockStripePaymentIntentsCreate.mockRejectedValue(authError);
const result = await StripeService.chargePaymentMethod(
'pm_123',
50.00,
'cus_123'
);
expect(result.status).toBe('requires_action');
expect(result.requiresAction).toBe(true);
expect(result.paymentIntentId).toBe('pi_requires_auth');
expect(result.clientSecret).toBe('pi_requires_auth_secret');
});
it('should handle us_bank_account payment method type', async () => {
const mockPaymentIntent = {
id: 'pi_bank',
status: 'succeeded',
client_secret: 'pi_bank_secret',
created: Date.now() / 1000,
latest_charge: {
payment_method_details: {
type: 'us_bank_account',
us_bank_account: {
last4: '6789',
bank_name: 'Test Bank'
}
}
}
};
mockStripePaymentIntentsCreate.mockResolvedValue(mockPaymentIntent);
const result = await StripeService.chargePaymentMethod(
'pm_bank_123',
50.00,
'cus_123'
);
expect(result.status).toBe('succeeded');
expect(result.paymentMethod).toEqual({
type: 'bank',
brand: 'bank_account',
last4: '6789'
});
});
it('should handle unknown payment method type', async () => {
const mockPaymentIntent = {
id: 'pi_unknown',
status: 'succeeded',
client_secret: 'pi_unknown_secret',
created: Date.now() / 1000,
latest_charge: {
payment_method_details: {
type: 'crypto_wallet'
}
}
};
mockStripePaymentIntentsCreate.mockResolvedValue(mockPaymentIntent);
const result = await StripeService.chargePaymentMethod(
'pm_crypto_123',
50.00,
'cus_123'
);
expect(result.status).toBe('succeeded');
expect(result.paymentMethod).toEqual({
type: 'crypto_wallet',
brand: 'crypto_wallet',
last4: null
});
});
it('should handle payment with no charge details', async () => {
const mockPaymentIntent = {
id: 'pi_no_charge',
status: 'succeeded',
client_secret: 'pi_no_charge_secret',
created: Date.now() / 1000,
latest_charge: null
};
mockStripePaymentIntentsCreate.mockResolvedValue(mockPaymentIntent);
const result = await StripeService.chargePaymentMethod(
'pm_123',
50.00,
'cus_123'
);
expect(result.status).toBe('succeeded');
expect(result.paymentMethod).toBeNull();
});
it('should handle card with missing details gracefully', async () => {
const mockPaymentIntent = {
id: 'pi_card_no_details',
status: 'succeeded',
client_secret: 'pi_card_secret',
created: Date.now() / 1000,
latest_charge: {
payment_method_details: {
type: 'card',
card: {} // Missing brand and last4
}
}
};
mockStripePaymentIntentsCreate.mockResolvedValue(mockPaymentIntent);
const result = await StripeService.chargePaymentMethod(
'pm_123',
50.00,
'cus_123'
);
expect(result.status).toBe('succeeded');
expect(result.paymentMethod).toEqual({
type: 'card',
brand: 'card',
last4: '****'
});
});
});
});