Update main_test.js

Author: giwawayu

lab2/main_test.js

@@ -2,35 +2,46 @@ const test = require('node:test');
 const assert = require('assert');
 const { Application, MailSystem } = require('./main');
 
-// Stub: Stub是一個假的相依物件，用來替換真正的相依物件
-const mailSystemStub = {
-  sendMail: () => {} // Stub了sendMail方法
-};
+test('Application sends email when invoked', () => {
+  // Stub: 使用 Stub 來模擬 MailSystem 的行為
+  const mailSystemStub = {
+    sendMail: jest.fn()
+  };
+  
+  const app = new Application(mailSystemStub);
+  app.sendEmail();
 
-// Mock: Mock是一個模擬的相依物件，用來驗證被測試程式的交互作用
-const mailSystemMock = {
-  sendMail: jest.fn() // 使用Jest的Mock函數來模擬sendMail方法
-};
+  // 驗證 sendMail 方法被調用
+  expect(mailSystemStub.sendMail).toHaveBeenCalled();
+});
 
-// Spy: Spy是一種監視函數的調用情況，用來驗證函數是否被正確調用
-const mailSystemSpy = jest.spyOn(mailSystem, 'sendMail');
+test('Application does not send email when disabled', () => {
+  // Mock: 使用 Mock 來模擬 MailSystem 的行為
+  const mailSystemMock = {
+    sendMail: jest.fn()
+  };
 
-test('Application sends email when invoked', () => {
-  const app = new Application(mailSystemStub); // 使用Stub來注入假的相依物件
-  app.sendEmail(); // 觸發應用程式的發送郵件操作
-  // 在這裡進行斷言，驗證是否正確調用了假的相依物件的sendMail方法
-});
+  const app = new Application(mailSystemMock);
+  app.disableEmail();
+  app.sendEmail();
 
-test('Application sends email when invoked - using mock', () => {
-  const app = new Application(mailSystemMock); // 使用Mock來注入模擬的相依物件
-  app.sendEmail(); // 觸發應用程式的發送郵件操作
-  // 在這裡進行斷言，驗證是否正確調用了模擬的相依物件的sendMail方法
-  expect(mailSystemMock.sendMail).toHaveBeenCalled(); // 驗證sendMail方法是否被呼叫
+  // 驗證 sendMail 方法未被調用
+  expect(mailSystemMock.sendMail).not.toHaveBeenCalled();
 });
 
-test('Application sends email when invoked - using spy', () => {
-  const app = new Application(mailSystem); // 直接使用真實的相依物件
-  app.sendEmail(); // 觸發應用程式的發送郵件操作
-  // 在這裡進行斷言，驗證是否正確調用了真實的相依物件的sendMail方法
-  expect(mailSystemSpy).toHaveBeenCalled(); // 使用Spy來驗證sendMail方法是否被呼叫
+test('Application retries sending email on failure', () => {
+  // Spy: 使用 Spy 來監視 MailSystem 的行為
+  const mailSystemSpy = jest.spyOn(MailSystem.prototype, 'sendMail');
+
+  const app = new Application(new MailSystem());
+  app.sendEmail();
+
+  // 模擬 sendMail 失敗
+  mailSystemSpy.mockImplementationOnce(() => { throw new Error('Failed to send email'); });
+
+  // 再次調用 sendEmail 方法，應該會進行重試
+  app.sendEmail();
+
+  // 驗證 sendMail 方法被調用了兩次
+  expect(mailSystemSpy).toHaveBeenCalledTimes(2);
 });