fix send mail

Author: SapphireLinXu

lab2/main_test.js

@@ -21,11 +21,15 @@ test('MailSystem should write mail correctly', (t) => {
 // });
 
 test('MailSystem should send mail correctly', (t) => {
-    const sendMock = t.mock.fn(MailSystem.prototype.send);
-    const success = sendMock('Alice', 'Congrats, Alice!');
-    assert.strictEqual(typeof success, 'boolean');
+    t.mock.method(Math, 'random').mock.mockImplementation(() => 0.6);
+    const success = MailSystem.prototype.send('Alice', 'Congrats, Alice!');
+    assert.strictEqual(success, true);
+    t.mock.method(Math, 'random').mock.mockImplementation(() => 0.4);
+    const failure = MailSystem.prototype.send('Alice', 'Congrats, Alice!');
+    assert.strictEqual(failure, false);
 });
 
+
 test('Application should read file correctly', async (t) => {
 
     fs.writeFileSync(TestData, 'Alice\nBob\nCharlie', 'utf8');
@@ -71,21 +75,21 @@ test('Application should select next person', async (t) => {
 });
 
 test('Application should notify person', async (t) => {
+    const nameList = ['Alice', 'Bob', 'Charlie'];
     t.mock.method(Application.prototype, 'getNames', async () => {
-        return [['Alice', 'Bob', 'Charlie'], []];
+        return [nameList, []];
     });
     const app = new Application();
     await app.getNames();
 
-    app.selectNextPerson();
-    console.log(app.selected);
+    app.selected = nameList;
     const writeMock = t.mock.method(MailSystem.prototype, 'write', () => {
         return true;
     });
     const sendMock = t.mock.method(MailSystem.prototype, 'send', () => {
         return true;
     });
     app.notifySelected();
-    assert.strictEqual(writeMock.mock.callCount(), 1);
-    assert.strictEqual(sendMock.mock.callCount(), 1);
+    assert.strictEqual(writeMock.mock.callCount(), app.selected.length);
+    assert.strictEqual(sendMock.mock.callCount(), app.selected.length);
 });