test: Fix monitor tests artifact mock and timeouts

__tests__/monitor.test.js

@@ -11,25 +11,30 @@ vi.mock("@actions/core", () => ({
   endGroup: vi.fn(),
 }));
 
-// ── mock @actions/artifact so tests never attempt real uploads ──────────────
+import * as artifact from "@actions/artifact";
 vi.mock("@actions/artifact", () => ({
-  DefaultArtifactClient: vi.fn().mockImplementation(() => ({
-    uploadArtifact: vi.fn().mockResolvedValue({}),
-  })),
+  DefaultArtifactClient: class {
+    uploadArtifact() {
+      return Promise.resolve({});
+    }
+  },
 }));
 
 import * as core from "@actions/core";
 import fs from "fs";
 import os from "os";
 import path from "path";
+import { DefaultArtifactClient } from "@actions/artifact";
 
 import {
   isOfficialServer,
   buildSubscribePath,
   detectTraceback,
   detectWarning,
+  outputMultiline,
   buildProgressMessage,
   writeLogFile,
+  uploadLogArtifacts,
   handleConsoleEvent,
   monitorConsole,
 } from "../monitor.js";
@@ -123,6 +128,33 @@ describe("detectWarning", () => {
   });
 });
 
+describe("outputMultiline", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("prefixes lines with shard when provided", () => {
+    outputMultiline("line1\nline2", "info", "shard0");
+    expect(core.info).toHaveBeenCalledWith("[shard0] line1");
+    expect(core.info).toHaveBeenCalledWith("[shard0] line2");
+  });
+
+  it("does not prefix when shard is missing", () => {
+    outputMultiline("line1", "info");
+    expect(core.info).toHaveBeenCalledWith("line1");
+  });
+
+  it("uses core.warning for level=warning", () => {
+    outputMultiline("warn", "warning", "s0");
+    expect(core.warning).toHaveBeenCalledWith("[s0] warn");
+  });
+
+  it("uses core.error for level=error", () => {
+    outputMultiline("err", "error", "s0");
+    expect(core.error).toHaveBeenCalledWith("[s0] err");
+  });
+});
+
 describe("buildProgressMessage", () => {
   it("formats correctly at 0 elapsed", () => {
     expect(buildProgressMessage(0, 50)).toBe("[Monitor] 0/50 ticks elapsed...");
@@ -141,48 +173,86 @@ describe("buildProgressMessage", () => {
   });
 });
 
+describe("uploadLogArtifacts", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("instantiates DefaultArtifactClient and calls uploadArtifact", async () => {
+    const spy = vi.spyOn(
+      artifact.DefaultArtifactClient.prototype,
+      "uploadArtifact",
+    );
+    await uploadLogArtifacts(
+      ["/path/to/shard0_console_log.txt"],
+      "custom-name",
+    );
+    expect(spy).toHaveBeenCalledWith(
+      "custom-name",
+      ["/path/to/shard0_console_log.txt"],
+      "/path/to",
+    );
+  });
+
+  it("does nothing if filePaths is empty", async () => {
+    const spy = vi.spyOn(
+      artifact.DefaultArtifactClient.prototype,
+      "uploadArtifact",
+    );
+    await uploadLogArtifacts([]);
+    expect(spy).not.toHaveBeenCalled();
+  });
+});
+
 // ────────────────────────────────────────────────────────────────────────────
 // handleConsoleEvent
 // ────────────────────────────────────────────────────────────────────────────
 
 describe("handleConsoleEvent", () => {
   let state;
-  let stdoutBuffer;
+  let shardBuffers;
 
   beforeEach(() => {
     vi.clearAllMocks();
     state = { sawTraceback: false, sawErrorLog: false, sawWarningLog: false };
-    stdoutBuffer = [];
+    shardBuffers = {};
   });
 
   const makeEvent = (log = [], results = [], error = null) => ({
     data: { messages: { log, results }, error },
   });
 
-  it("calls core.info for each stdout line when logToFile=false", () => {
-    const event = makeEvent(["line1", "line2"]);
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
-    expect(core.info).toHaveBeenCalledWith("line1");
-    expect(core.info).toHaveBeenCalledWith("line2");
-    expect(stdoutBuffer).toHaveLength(0);
+  it("calls core.info for each stdout line with shard prefix when logToFile=false", () => {
+    const event = {
+      data: {
+        shard: "shard0",
+        messages: { log: ["line1"], results: ["line2"] },
+      },
+    };
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
+    expect(core.info).toHaveBeenCalledWith("[shard0] line1");
+    expect(core.info).toHaveBeenCalledWith("[shard0] line2");
+    expect(Object.keys(shardBuffers)).toHaveLength(0);
   });
 
   it("buffers stdout when logToFile=true; does not call core.info", () => {
-    const event = makeEvent(["line1", "line2"]);
-    handleConsoleEvent(event, { logToFile: true }, stdoutBuffer, state);
+    const event = {
+      data: { shard: "shard0", messages: { log: ["line1", "line2"] } },
+    };
+    handleConsoleEvent(event, { logToFile: true }, shardBuffers, state);
     expect(core.info).not.toHaveBeenCalled();
-    expect(stdoutBuffer).toEqual(["line1", "line2"]);
+    expect(shardBuffers["shard0"]).toEqual(["line1", "line2"]);
   });
 
   it("includes results lines in output", () => {
     const event = makeEvent([], ["result1"]);
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
     expect(core.info).toHaveBeenCalledWith("result1");
   });
 
   it("calls core.error when error field is non-empty", () => {
     const event = makeEvent([], [], "Script crashed");
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
     expect(core.error).toHaveBeenCalledWith("Script crashed");
     expect(state.sawErrorLog).toBe(true);
   });
@@ -190,58 +260,73 @@ describe("handleConsoleEvent", () => {
   it("sets state.sawTraceback when error contains stack frames", () => {
     const error = "TypeError: boom\n    at Object.<anonymous> (main:1:1)";
     const event = makeEvent([], [], error);
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
     expect(state.sawTraceback).toBe(true);
     expect(state.sawErrorLog).toBe(true);
   });
 
   it("does not set sawTraceback for plain error without stack frames", () => {
     const event = makeEvent([], [], "Script error: low CPU");
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
     expect(state.sawTraceback).toBe(false);
     expect(state.sawErrorLog).toBe(true);
   });
 
   it("sets state.sawWarningLog and calls core.warning for warn lines", () => {
     const event = makeEvent(["<font color='orange'>low energy</font>"]);
-    handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: false }, shardBuffers, state);
     expect(state.sawWarningLog).toBe(true);
     expect(core.warning).toHaveBeenCalled();
   });
 
   it("calls core.warning regardless of logToFile setting", () => {
     const event = makeEvent(["<font color='orange'>warn line</font>"]);
-    handleConsoleEvent(event, { logToFile: true }, stdoutBuffer, state);
+    handleConsoleEvent(event, { logToFile: true }, shardBuffers, state);
     expect(core.warning).toHaveBeenCalled();
   });
 
   it("handles missing messages gracefully (no crash on empty event)", () => {
     const event = { data: {} };
     expect(() =>
-      handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state),
+      handleConsoleEvent(event, { logToFile: false }, shardBuffers, state),
     ).not.toThrow();
   });
 
   it("handles completely empty event gracefully", () => {
     const event = {};
     expect(() =>
-      handleConsoleEvent(event, { logToFile: false }, stdoutBuffer, state),
+      handleConsoleEvent(event, { logToFile: false }, shardBuffers, state),
     ).not.toThrow();
   });
 
   it("filters messages by shard when targetShard is provided", () => {
     const event = { data: { shard: "shard0", messages: { log: ["msg0"] } } };
-    handleConsoleEvent(event, { shard: "shard1" }, stdoutBuffer, state);
+    handleConsoleEvent(event, { shard: "shard1" }, shardBuffers, state);
     expect(core.info).not.toHaveBeenCalled();
 
-    handleConsoleEvent(event, { shard: "shard0" }, stdoutBuffer, state);
-    expect(core.info).toHaveBeenCalledWith("msg0");
+    handleConsoleEvent(event, { shard: "shard0" }, shardBuffers, state);
+    expect(core.info).toHaveBeenCalledWith("[shard0] msg0");
   });
 
   it("does not filter when targetShard is undefined", () => {
     const event = { data: { shard: "shard0", messages: { log: ["msg0"] } } };
-    handleConsoleEvent(event, {}, stdoutBuffer, state);
-    expect(core.info).toHaveBeenCalledWith("msg0");
+    handleConsoleEvent(event, {}, shardBuffers, state);
+    expect(core.info).toHaveBeenCalledWith("[shard0] msg0");
+  });
+
+  it("buffers messages separately for different shards", () => {
+    const event0 = { data: { shard: "shard0", messages: { log: ["msg0"] } } };
+    const event1 = { data: { shard: "shard1", messages: { log: ["msg1"] } } };
+    handleConsoleEvent(event0, { logToFile: true }, shardBuffers, state);
+    handleConsoleEvent(event1, { logToFile: true }, shardBuffers, state);
+    expect(shardBuffers["shard0"]).toEqual(["msg0"]);
+    expect(shardBuffers["shard1"]).toEqual(["msg1"]);
+  });
+
+  it("uses 'default' key when shard is missing in event", () => {
+    const event = { data: { messages: { log: ["msg"] } } };
+    handleConsoleEvent(event, { logToFile: true }, shardBuffers, state);
+    expect(shardBuffers["default"]).toEqual(["msg"]);
   });
 });
 
@@ -519,4 +604,40 @@ describe("monitorConsole", () => {
     expect(result.sawTraceback).toBe(true);
     expect(result.sawErrorLog).toBe(true);
   });
+
+  it("creates separate log files for different shards when logToFile=true", async () => {
+    const api = buildMockApi({ ticks: [100, 101, 102, 103] });
+
+    setTimeout(() => {
+      api._fireConsole({
+        shard: "shard0",
+        messages: { log: ["msg0"], results: [] },
+        error: null,
+      });
+      api._fireConsole({
+        shard: "shard1",
+        messages: { log: ["msg1"], results: [] },
+        error: null,
+      });
+    }, 50);
+
+    // Verify uploadArtifact was called with two files
+    const spy = vi.spyOn(
+      artifact.DefaultArtifactClient.prototype,
+      "uploadArtifact",
+    );
+    await monitorConsole(api, {
+      ...BASE_OPTS,
+      logToFile: true,
+    });
+
+    expect(spy).toHaveBeenCalledWith(
+      "screeps-console-log",
+      expect.arrayContaining([
+        expect.stringContaining("shard0_console_log.txt"),
+        expect.stringContaining("shard1_console_log.txt"),
+      ]),
+      expect.any(String),
+    );
+  });
 });