From 437d6ec0c0fbfc125c89150f1493cdfcd92a3dbf Mon Sep 17 00:00:00 2001
From: brwarner <brwarner2@gmail.com>
Date: Sat, 16 Nov 2024 09:56:12 -0800
Subject: [PATCH] Added a few simple tests for stat generation

---
 src/tests/specs/inkjs/compiler/Stats.spec.ts | 43 ++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 src/tests/specs/inkjs/compiler/Stats.spec.ts

diff --git a/src/tests/specs/inkjs/compiler/Stats.spec.ts b/src/tests/specs/inkjs/compiler/Stats.spec.ts
new file mode 100644
index 00000000..28e65a9b
--- /dev/null
+++ b/src/tests/specs/inkjs/compiler/Stats.spec.ts
@@ -0,0 +1,43 @@
+import { Stats } from "../../../../compiler/Stats";
+import { Compiler } from "../../../../ink";
+
+function getStats(inkSource: string): Stats {
+  const compiler = new Compiler(inkSource);
+  compiler.Compile();
+  const stats = compiler.GenerateStats();
+  expect(stats).not.toBeNull();
+  return stats!;
+}
+
+describe("Stat Generation", () => {
+  it("basic word count", () => {
+    const stats = getStats("this is an ink story.");
+    expect(stats.words).toBe(5);
+  });
+  it("word count doesn't include divert or variable names", () => {
+    const stats = getStats(
+      "VAR MyVariable = 3\n->start\n=== start\nthis is an ink story.\n->END"
+    );
+    expect(stats.words).toBe(5);
+  });
+  it("count functions, knots, and stitches", () => {
+    const stats = getStats(
+      "->start\n=== function myFunc()\n~ return 0\n=== start\n= stitch\nHello world!"
+    );
+    expect(stats.functions).toBe(1);
+    expect(stats.knots).toBe(2);
+    expect(stats.stitches).toBe(1);
+  });
+  it("count diverts", () => {
+    const stats = getStats("->go\n- (go)\n->next\n-(next)");
+    expect(stats.diverts).toBe(2);
+  });
+  it("end counts as a divert", () => {
+    const stats = getStats("->go\n- (go)\n->next\n-(next)\n->END");
+    expect(stats.diverts).toBe(3);
+  });
+  it("count gathers", () => {
+    const stats = getStats("->go\n- (go)\n->next\n-(next)");
+    expect(stats.gathers).toBe(2);
+  });
+});