Adding a polyfill for Suite#total()

packages/client/src/serialization.test.ts

@@ -15,11 +15,26 @@ describe("Client serialization", () => {
   });
 
   it("can serialize a Test", () => {
+    const suite = new Suite("root suite");
     const test = new Test("test title");
-    test.parent = new Suite("root suite");
+    suite.addTest(test);
+
     const result = serialize(test);
     const parsed = parse(result);
-    expect(parsed).contain.keys("type", "__mocha_id__");
+    expect(parsed).contain.keys("type", "__mocha_id__", "$$fullTitle");
+    expect(parsed.$$fullTitle).equals("root suite test title");
+  });
+
+  it("can serialize a Suite", () => {
+    const suite = new Suite("root suite");
+    const test = new Test("test title");
+    suite.addTest(test);
+
+    const result = serialize(test.parent);
+    const parsed = parse(result);
+    expect(parsed).contain.keys("type", "__mocha_id__", "$$fullTitle", "$$total");
+    expect(parsed.$$fullTitle).equals("root suite");
+    expect(parsed.$$total).equals(1);
   });
 
   it("can serialize a Test with a parent Suite", () => {
@@ -29,8 +44,7 @@ describe("Client serialization", () => {
       // eslint-disable-next-line no-console
       console.log("Before hook ran");
     });
-    test.parent = suite;
-    suite.tests = [ test ];
+    suite.addTest(test);
     {
       // Serializing the first time should return an object with $type and $properties
       const result = serialize(test);

packages/client/src/serialization.ts

@@ -1,4 +1,5 @@
 import * as flatted from "flatted";
+import { Suite } from "mocha-remote-mocha";
 
 import { extend } from "./debug";
 const debug = extend("serialization");
@@ -17,7 +18,15 @@ function toJSON(value: Record<string, unknown>): Record<string, unknown> {
       stack: stack ? filterStack(stack) : stack,
     };
   } else if (typeof value === "object" && typeof value.serialize === "function") {
-    return value.serialize();
+    const result = value.serialize();
+    if (value instanceof Suite) {
+      // Polyfill missing methods
+      Object.assign(result, {
+        "type": "suite",
+        "$$total": value.total(),
+      });
+    }
+    return result;
   } else {
     return value;
   }

packages/integration-tests/src/basic.test.ts

@@ -102,4 +102,29 @@ describe("basic", () => {
       });
     });
   });
+
+  it("expose the total number of tests on a Suite", async () => {
+    // Create a server - which is supposed to run in Node
+    server = new Server({ port: 0, autoRun: true });
+    await server.start();
+    // Create a client - which is supposed to run where the tests are running
+    client = new Client({
+      autoConnect: false,
+      url: server.url,
+      tests() {
+        it("tests one thing", () => {});
+        it("tests another thing", () => {});
+      }
+    });
+
+    const totalCount = new Promise<number>(resolve => {
+      const runner = server.run(() => {});
+      runner.once("suite", (suite) => {
+        resolve(suite.total());
+      });
+    });
+
+    await client.connect();
+    expect(await totalCount).equals(2);
+  });
 });