socketduct 0.1.20 → 0.1.21

package/test/unit/index.test.js

@@ -13,6 +13,7 @@ import {
   MultiplexedSocketductTransportSet,
   OutboundReverseGateway,
   ReverseGatewayConnection,
+  SocketductRecoveryLookupError,
   SocketductHttpAgent,
   SocketductSocket
 } from "../../src/index.js"
@@ -57,6 +58,7 @@ test("package exports the protocol-v3 pooled connection and relay entrypoints",
 
   assert.equal(multiplexed.MultiplexedSocketductConnection, MultiplexedSocketductConnection)
   assert.equal(multiplexed.MultiplexedSocketductTransportSet, MultiplexedSocketductTransportSet)
+  assert.equal(multiplexed.SocketductRecoveryLookupError, SocketductRecoveryLookupError)
   assert.equal(createMultiplexedSocketductConnection({
     relay: {host: "127.0.0.1", port: 3100, token: "test-token"}
   }) instanceof MultiplexedSocketductConnection, true)

package/test/unit/spool/spool-file.test.js

@@ -63,6 +63,64 @@ test("SessionSpool truncates acknowledged chunks", async () => {
   })
 })
 
+test("SessionSpool truncates an acknowledgement inside a chunk in application bytes", async () => {
+  await withTempDir(async (root) => {
+    const spool = new SessionSpool({rootDir: root})
+    const sessionId = "session_partial_ack"
+    await spool.create({sessionId, ownerId: "user-1"})
+    await spool.appendChunk({sessionId, sequence: 8, data: Buffer.from("abcdefgh")})
+
+    await spool.truncateAcked(sessionId, 3)
+
+    const chunks = await spool.readChunksAfter(sessionId, 3)
+    assert.deepEqual(chunks, [{sequence: 8, data: Buffer.from("defgh")}])
+    assert.equal(spool.getUsedBytes(sessionId), 5)
+  })
+})
+
+test("SessionSpool atomically admits a complete chunk batch before recovery commit", async () => {
+  await withTempDir(async (root) => {
+    const spool = new SessionSpool({rootDir: root, maxBytes: 32})
+    const sessionId = "session_durable_batch"
+    await spool.createNew({sessionId, ownerId: "reverse-client-v3"})
+    await spool.appendChunk({sessionId, ownerId: "reverse-client-v3", sequence: 4, data: Buffer.from("base")})
+    await spool.writeRecoveryState(sessionId, {version: 3, sendSequence: 4})
+
+    await spool.appendChunksAndRecoveryState({
+      sessionId,
+      ownerId: "reverse-client-v3",
+      previousSequence: 4,
+      chunks: [
+        {sequence: 7, data: Buffer.from("one")},
+        {sequence: 10, data: Buffer.from("two")}
+      ],
+      recoveryState: {version: 3, sendSequence: 10}
+    })
+
+    assert.deepEqual((await spool.readChunksAfter(sessionId, 0)).map(({sequence, data}) => [sequence, data.toString()]), [
+      [4, "base"],
+      [7, "one"],
+      [10, "two"]
+    ])
+    assert.deepEqual(spool.readRecoveryStateSync(sessionId, "reverse-client-v3"), {version: 3, sendSequence: 10})
+    assert.equal(spool.getUsedBytes(sessionId), 10)
+  })
+})
+
+test("SessionSpool recovery lookup distinguishes missing state from invalid state", async () => {
+  await withTempDir(async (root) => {
+    const spool = new SessionSpool({rootDir: root})
+    assert.equal(spool.findRecoveryStateSync("missing_session", "edge-v3"), undefined)
+
+    await spool.createNew({sessionId: "present_session", ownerId: "edge-v3"})
+    await spool.writeRecoveryState("present_session", {version: 3})
+    assert.deepEqual(spool.findRecoveryStateSync("present_session", "edge-v3"), {version: 3})
+
+    await writeFile(join(root, "present_session", "recovery.json"), "{invalid\n")
+    assert.throws(() => spool.findRecoveryStateSync("present_session", "edge-v3"), /Invalid Socketduct recovery state/)
+  })
+})
+
 test("SessionSpool rejects path traversal and mismatched metadata", async () => {
   await withTempDir(async (root) => {
     const spool = new SessionSpool({rootDir: root})