| 1 | import { LoginService } from '../../src/user/login.service'; |
| 2 | |
| 3 | export function mockSmsService() { |
| 4 | const originalLoginService = jest.spyOn( |
| 5 | LoginService.prototype, |
| 6 | 'postPhoneLoginCode', |
| 7 | ); |
| 8 | originalLoginService.mockImplementation(async () => true); |
| 9 | |
| 10 | const originalVerifyCode = jest.spyOn( |
| 11 | LoginService.prototype, |
| 12 | 'verifyPhoneCode', |
| 13 | ); |
| 14 | originalVerifyCode.mockImplementation( |
| 15 | async (phone, code) => code === '123456', |
| 16 | ); |
| 17 | |
| 18 | return { |
| 19 | restore: () => { |
| 20 | originalLoginService.mockRestore(); |
| 21 | originalVerifyCode.mockRestore(); |
| 22 | }, |
| 23 | }; |
| 24 | } |
| 25 |