Update main_test.js

Author: as10968574

lab2/main_test.js

@@ -1,80 +1,68 @@
 const test = require('node:test');
 const assert = require('assert');
 const fs = require('fs');
-const { Application, MailSystem } = require('./main');
-const sinon = require('sinon');
-
-// Mock fs.readFile
-sinon.stub(fs, 'readFile').callsFake((file, options, callback) => {
-    callback(null, 'Guan\nChen\nGala\n');
+test.mock.method(fs, 'readFile', (file, options, callback) => {
+  callback(null, 'Guan\nChen\nGala');
 });
+const { Application, MailSystem } = require('./main');
 
-test("MailSystem write", () => {
-    const mailSystem = new MailSystem();
-    assert.strictEqual(mailSystem.write("Guan"), "Congrats, Guan!");
+// TODO: write your tests here
+// Remember to use Stub, Mock, and Spy when necessary
+test("Test MailSystem's write", () => {
+    const myClass = new MailSystem()
+    assert.strictEqual( myClass.write("Guan"),"Congrats, Guan!")
 });
 
-test("MailSystem send", () => {
-    const mailSystem = new MailSystem();
-    const originalMathRandom = Math.random;
-
-    Math.random = () => 0.6;
-    const context = mailSystem.write("Guan");
-    assert.strictEqual(mailSystem.send("Guan", context), true);
-
-    Math.random = () => 0.4;
-    assert.strictEqual(mailSystem.send("Chen", context), false);
-
-    Math.random = originalMathRandom;
+test("Test MailSystem's send(name, context)", () => {
+  Math.random = () => 0.6
+    const mailSystem = new MailSystem()
+    const context = mailSystem.write("Guan")
+    const success = mailSystem.send("Guan",context)
+    assert.strictEqual(success,true)
+    Math.random = () => 0.5
+    assert.strictEqual(mailSystem.send("Chen",context),false)
 });
 
-test("Application getNames", async () => {
-    const application = new Application();
-    const [people, selected] = await application.getNames();
-    // Remove any empty strings from the array
-    assert.deepStrictEqual(people.filter(name => name), ["Guan", "Chen", "Gala"]);
-    assert.deepStrictEqual(selected, []);
+test("Test Application's getNames", async () => {
+  const application = new Application
+  const [people, selected] = await application.getNames()
+  assert.deepStrictEqual(people, ["Guan", "Chen", "Gala"])
+  assert.deepStrictEqual(selected, [])
 });
 
-test("Application getRandomPerson", () => {
-    const application = new Application();
-    application.people = ["Guan", "Chen", "Gala"];
-    Math.random = () => 0;
-    assert.deepStrictEqual(application.getRandomPerson(), "Guan");
-    Math.random = () => 0.6;
-    assert.deepStrictEqual(application.getRandomPerson(), "Chen");
-    Math.random = () => 0.9;
-    assert.deepStrictEqual(application.getRandomPerson(), "Gala");
+test("Test Application's getRandomPerson()", (t) => {
+  const application = new Application
+  application.people = ["Guan", "Chen", "Gala"]
+  Math.random = () => 0
+  assert.deepStrictEqual(application.getRandomPerson(), "Guan");
+  Math.random = () => 0.6
+  assert.deepStrictEqual(application.getRandomPerson(), "Chen");
+  Math.random = () => 0.9
+  assert.deepStrictEqual(application.getRandomPerson(), "Gala");
 });
 
-test("Application selectNextPerson", async () => {
-    const application = new Application();
-    await application.getNames();
-    application.selected = ["Guan"];
-    let count = 1;
-    const getRandomPersonStub = sinon.stub(application, 'getRandomPerson').callsFake(() => {
-        const name_list = application.people;
-        if (count < name_list.length) {
-            return name_list[count++];
-        } else {
-            return null;
-        }
-    });
-
-    assert.strictEqual(application.selectNextPerson(), "Chen");
-    assert.strictEqual(application.selectNextPerson(), "Gala");
-    assert.strictEqual(application.selectNextPerson(), null);
-
-    getRandomPersonStub.restore();
+test("Test Application's selectNextPerson",  async() => {
+  const application = new Application
+  const name_list = await application.getNames();
+  application.selected = ["Guan"]
+  let count = 0;
+  test.mock.method(application, 'getRandomPerson', () => {
+      if (count <= name_list.length) { 
+          return name_list[0][count++]; 
+      }
+  });
+  assert.strictEqual(application.selectNextPerson(), "Chen")
+  assert.strictEqual(application.selectNextPerson(), "Gala")
+  assert.strictEqual(application.selectNextPerson(), null)
 });
 
-test("Application notifySelected", () => {
-    const application = new Application();
-    application.people = ["Guan", "Chen", "Gala"];
-    application.selected = ["Guan", "Chen", "Gala"];
-    sinon.spy(application.mailSystem, 'write');
-    sinon.spy(application.mailSystem, 'send');
-    application.notifySelected();
-    assert.strictEqual(application.mailSystem.write.callCount, application.people.length);
-    assert.strictEqual(application.mailSystem.send.callCount, application.people.length);
+test("Test Application's notifySelected() ",  () => {
+  const application = new Application  
+  application.people = ["Guan", "Chen", "Gala"]
+  application.selected = ["Guan", "Chen", "Gala"]
+  application.mailSystem.write = test.mock.fn(application.mailSystem.write)
+  application.mailSystem.send = test.mock.fn(application.mailSystem.send)
+  application.notifySelected()
+  assert.strictEqual(application.mailSystem.send.mock.calls.length, application.people.length);
+  assert.strictEqual(application.mailSystem.write.mock.calls.length, application.people.length); 
 });