aws-sdk-vitest-mock 0.4.0 → 0.6.0

package/index.js

@@ -1,347 +1,461 @@
-import { vi as b } from "vitest";
-import { readFileSync as R } from "node:fs";
-import N from "node:path";
-import { Readable as W } from "node:stream";
-import { m as Y } from "./matchers-C6AtmwWz.js";
+import { vi as j } from "vitest";
+import { c as F } from "./matchers-DmTtFk31.js";
+import { m as ge } from "./matchers-DmTtFk31.js";
+import { readFileSync as O } from "node:fs";
+import I from "node:path";
+import { Readable as M } from "node:stream";
 class d extends Error {
-  constructor(e, s, c, n) {
-    super(e), this.name = "AwsError", this.code = s, this.statusCode = c, this.retryable = n;
+  constructor(t, o, i, n) {
+    super(t), this.name = "AwsError", this.code = o, this.statusCode = i, this.retryable = n;
   }
 }
-const C = (t) => {
-  const e = t ? `The specified key does not exist. Key: ${t}` : "The specified key does not exist.";
-  return new d(e, "NoSuchKey", 404, !1);
-}, $ = (t) => {
-  const e = t ? `The specified bucket does not exist. Bucket: ${t}` : "The specified bucket does not exist.";
-  return new d(e, "NoSuchBucket", 404, !1);
-}, D = (t) => {
-  const e = t ? `Access Denied for resource: ${t}` : "Access Denied";
-  return new d(e, "AccessDenied", 403, !1);
-}, M = (t) => {
-  const e = t ? `Requested resource not found: ${t}` : "Requested resource not found";
-  return new d(e, "ResourceNotFoundException", 400, !1);
-}, T = () => new d(
+const T = (e) => new d(
+  e ? `The specified key does not exist. Key: ${e}` : "The specified key does not exist.",
+  "NoSuchKey",
+  404,
+  !1
+), K = (e) => new d(
+  e ? `The specified bucket does not exist. Bucket: ${e}` : "The specified bucket does not exist.",
+  "NoSuchBucket",
+  404,
+  !1
+), A = (e) => new d(
+  e ? `Access Denied for resource: ${e}` : "Access Denied",
+  "AccessDenied",
+  403,
+  !1
+), B = (e) => new d(
+  e ? `Requested resource not found: ${e}` : "Requested resource not found",
+  "ResourceNotFoundException",
+  400,
+  !1
+), J = () => new d(
   "The conditional request failed",
   "ConditionalCheckFailedException",
   400,
   !1
-), I = () => new d("Rate exceeded", "Throttling", 400, !0), O = () => new d(
+), _ = () => new d("Rate exceeded", "Throttling", 400, !0), q = () => new d(
   "We encountered an internal error. Please try again.",
   "InternalServerError",
   500,
   !0
-);
-function k() {
-  return {
-    enabled: !1,
-    log(t, e) {
-      this.enabled && (e === void 0 ? console.log(`[aws-sdk-vitest-mock](Debug) ${t}`) : console.log(`[aws-sdk-vitest-mock](Debug) ${t}`, e));
-    }
-  };
-}
-function y(t) {
-  t.enabled = !0;
-}
-function w(t) {
-  t.enabled = !1;
-}
-function F(t) {
-  const e = N.resolve(t), s = R(e, "utf8");
-  return t.endsWith(".json") ? JSON.parse(s) : s;
-}
-function L(t, e = {}) {
-  const { pageSize: s = 10, tokenKey: c = "NextToken", itemsKey: n = "Items" } = e;
-  if (t.length === 0)
+), z = (e) => {
+  try {
+    return JSON.stringify(e, void 0, 2);
+  } catch {
+    return typeof e == "object" && e !== null ? "[Complex Object]" : typeof e == "string" ? e : typeof e == "number" || typeof e == "boolean" ? String(e) : "[Non-serializable value]";
+  }
+}, v = (e, t) => {
+  const o = F.magenta("aws-sdk-vitest-mock(debug):");
+  if (t === void 0)
+    console.log(`${o} ${e}`);
+  else {
+    const i = z(t);
+    console.log(`${o} ${e}
+${i}`);
+  }
+}, L = (e = !1) => ({
+  enabled: e,
+  explicitlySet: !1,
+  log(t, o) {
+    this.enabled && v(t, o);
+  },
+  logDirect(t, o) {
+    v(t, o);
+  }
+}), w = (e) => {
+  e.enabled = !0, e.explicitlySet = !0;
+}, D = (e) => {
+  e.enabled = !1, e.explicitlySet = !0;
+}, V = (e) => {
+  const t = I.resolve(e), o = O(t, "utf8");
+  return e.endsWith(".json") ? JSON.parse(o) : o;
+}, U = (e, t = {}) => {
+  const { pageSize: o = 10, tokenKey: i = "NextToken", itemsKey: n = "Items" } = t;
+  if (e.length === 0)
     return [{ [n]: [] }];
-  const o = [];
-  for (let r = 0; r < t.length; r += s) {
-    const u = t.slice(r, r + s), i = r + s < t.length, l = { [n]: u };
-    if (i) {
-      const a = l;
-      a[c] = `token-${r + s}`;
+  const s = [];
+  for (let r = 0; r < e.length; r += o) {
+    const c = e.slice(r, r + o), l = r + o < e.length, u = { [n]: c };
+    if (l) {
+      const a = u, g = c[c.length - 1];
+      a[i] = g;
     }
-    o.push(l);
+    s.push(u);
   }
-  return o;
-}
-function K() {
-  return typeof process < "u" && process.versions?.node ? "node" : typeof process < "u" && process.versions?.bun ? "bun" : "browser";
-}
-function A(t) {
-  const e = typeof t == "string" ? Buffer.from(t, "utf8") : Buffer.from(t);
-  let s = !1;
-  return new W({
+  return s;
+}, G = () => typeof process < "u" && process.versions?.node ? "node" : typeof process < "u" && process.versions?.bun ? "bun" : "browser", H = (e) => {
+  const t = typeof e == "string" ? Buffer.from(e, "utf8") : Buffer.from(e);
+  let o = !1;
+  return new M({
     read() {
-      s || (this.push(e), this.push(null), s = !0);
+      o || (this.push(t), this.push(null), o = !0);
     }
   });
-}
-function B(t) {
-  let e;
-  return typeof t == "string" ? e = new TextEncoder().encode(t) : t instanceof Buffer ? e = new Uint8Array(t) : e = t, new ReadableStream({
-    start(s) {
-      s.enqueue(e), s.close();
+}, Q = (e) => {
+  let t;
+  return typeof e == "string" ? t = new TextEncoder().encode(e) : e instanceof Buffer ? t = new Uint8Array(e) : t = e, new ReadableStream({
+    start(o) {
+      o.enqueue(t), o.close();
     }
   });
+}, S = (e) => {
+  const t = G();
+  return t === "node" || t === "bun" ? H(e) : Q(e);
+};
+let $ = !1;
+function ce(e) {
+  $ = e;
 }
-function h(t) {
-  const e = K();
-  return e === "node" || e === "bun" ? A(t) : B(t);
+function p(e) {
+  return e.explicitlySet ? e.enabled : $;
 }
-function v(t, e) {
-  return Object.keys(e).every((s) => {
-    const c = e[s], n = t[s];
-    return c && typeof c == "object" && !Array.isArray(c) ? typeof n != "object" || n === null ? !1 : v(
+function N(e, t) {
+  return Object.keys(t).every((o) => {
+    const i = t[o], n = e[o];
+    return i && typeof i == "object" && !Array.isArray(i) ? typeof n != "object" || n === null ? !1 : N(
       n,
-      c
-    ) : n === c;
+      i
+    ) : n === i;
   });
 }
-function j(t, e) {
-  if (t === e) return !0;
-  if (typeof t != "object" || t === null || typeof e != "object" || e === null)
-    return t === e;
-  const s = Object.keys(t), c = Object.keys(e);
-  return s.length !== c.length ? !1 : c.every((n) => {
-    if (!Object.prototype.hasOwnProperty.call(t, n)) return !1;
-    const o = t, r = e, u = o[n], i = r[n];
-    return typeof u == "object" && u !== null && typeof i == "object" && i !== null ? j(u, i) : u === i;
+function x(e, t) {
+  if (e === t) return !0;
+  if (typeof e != "object" || e === null || typeof t != "object" || t === null)
+    return e === t;
+  const o = Object.keys(e), i = Object.keys(t);
+  return o.length !== i.length ? !1 : i.every((n) => {
+    if (!Object.prototype.hasOwnProperty.call(e, n)) return !1;
+    const s = e, r = t, c = s[n], l = r[n];
+    return typeof c == "object" && c !== null && typeof l == "object" && l !== null ? x(c, l) : c === l;
   });
 }
-function x(t) {
-  return async function(e) {
-    const s = () => this;
-    t.debugLogger.log(
-      `Received command: ${e.constructor.name}`,
-      e.input
+function X(e, t, o) {
+  const i = t.map((s, r) => {
+    const c = s.matcher ? JSON.stringify(s.matcher, void 0, 2) : "any input", l = s.strict ? " (strict mode)" : "";
+    return `  Mock #${r + 1}: ${c}${l}`;
+  }).join(`
+`), n = JSON.stringify(o, void 0, 2);
+  return new Error(
+    `No matching mock found for ${e}.
+
+Found ${t.length} mock(s) but none matched the input.
+
+Configured mocks:
+${i}
+
+Received input:
+${n}
+
+Tip: Enable debug mode with enableDebug() for detailed matching information.`
+  );
+}
+function Y(e, t) {
+  const o = JSON.stringify(t, void 0, 2);
+  return new Error(
+    `No mock configured for command: ${e}.
+
+Received input:
+${o}
+
+Did you forget to call mockClient.on(${e})?`
+  );
+}
+function Z(e, t) {
+  return e.findIndex((o) => o.strict ? o.matcher && x(t, o.matcher) : !o.matcher || N(t, o.matcher));
+}
+function E(e) {
+  return async function(t) {
+    const o = () => this, i = p(e.debugLogger), n = t.constructor.name;
+    i && e.debugLogger.logDirect(
+      `Received command: ${n}`,
+      t.input
     );
-    const c = t.map.get(
-      e.constructor
+    const s = e.map.get(
+      t.constructor
     );
-    if (c) {
-      t.debugLogger.log(
-        `Found ${c.length} mock(s) for ${e.constructor.name}`
-      );
-      const n = c.findIndex((o) => o.strict ? o.matcher && j(e.input, o.matcher) : !o.matcher || v(e.input, o.matcher));
-      if (n === -1)
-        t.debugLogger.log(
-          `No matching mock found for ${e.constructor.name}`,
-          e.input
-        );
-      else {
-        const o = c[n];
-        if (!o)
-          throw new Error(`Mock at index ${n} not found`);
-        return t.debugLogger.log(
-          `Using mock at index ${n} for ${e.constructor.name}`
-        ), o.once && (c.splice(n, 1), t.debugLogger.log(
-          `Removed one-time mock for ${e.constructor.name}`
-        )), o.handler(
-          e.input,
-          s()
-        );
-      }
-    } else
-      t.debugLogger.log(
-        `No mocks configured for ${e.constructor.name}`
-      );
-    throw new Error(
-      `No mock configured for command: ${e.constructor.name}`
+    if (!s)
+      throw i && e.debugLogger.logDirect(
+        `No mocks configured for ${n}`
+      ), Y(n, t.input);
+    i && e.debugLogger.logDirect(
+      `Found ${s.length} mock(s) for ${n}`
     );
+    const r = Z(s, t.input);
+    if (r === -1)
+      throw i && e.debugLogger.logDirect(
+        `No matching mock found for ${n}`,
+        t.input
+      ), X(n, s, t.input);
+    const c = s[r];
+    if (!c)
+      throw new Error(`Mock at index ${r} not found`);
+    return i && e.debugLogger.logDirect(
+      `Using mock at index ${r} for ${n}`
+    ), c.once && (s.splice(r, 1), i && e.debugLogger.logDirect(
+      `Removed one-time mock for ${n}`
+    )), c.handler(t.input, o());
   };
 }
-function S(t, e, s, c = {}) {
-  const n = (r, u) => {
-    const i = {
-      matcher: s,
+function C(e, t, o, i = {}) {
+  const n = (r, c, l) => {
+    const u = {
+      matcher: o,
       handler: r,
-      once: u,
-      strict: !!c.strict
-    }, l = t.map.get(e) ?? [];
-    if (u) {
-      const a = l.findIndex((f) => !f.once);
-      a === -1 ? l.push(i) : l.splice(a, 0, i), t.map.set(
-        e,
-        l
+      once: c,
+      strict: !!i.strict
+    }, a = e.map.get(t) ?? [], g = p(e.debugLogger);
+    if (c) {
+      const f = a.findIndex((h) => !h.once);
+      f === -1 ? a.push(u) : a.splice(f, 0, u), e.map.set(
+        t,
+        a
+      ), g && e.debugLogger.logDirect(
+        `Configured ${l}Once for ${t.name}`,
+        o ? { matcher: o, strict: !!i.strict } : void 0
       );
     } else {
-      const a = l.filter(
-        (f) => f.once || JSON.stringify(f.matcher) !== JSON.stringify(s)
+      const f = a.filter(
+        (h) => h.once || JSON.stringify(h.matcher) !== JSON.stringify(o)
       );
-      a.push(i), t.map.set(
-        e,
-        a
+      f.push(u), e.map.set(
+        t,
+        f
+      ), g && e.debugLogger.logDirect(
+        `Configured ${l} for ${t.name}`,
+        o ? { matcher: o, strict: !!i.strict } : void 0
       );
     }
-  }, o = {
+  }, s = {
     resolves(r) {
-      return n(() => Promise.resolve(r), !1), o;
+      return n(() => Promise.resolve(r), !1, "resolves"), s;
     },
     rejects(r) {
-      return n(() => {
-        const u = typeof r == "string" ? new Error(r) : r;
-        return Promise.reject(u);
-      }, !1), o;
+      return n(
+        () => {
+          const c = typeof r == "string" ? new Error(r) : r;
+          return Promise.reject(c);
+        },
+        !1,
+        "rejects"
+      ), s;
     },
     callsFake(r) {
-      return n(r, !1), o;
+      return n(r, !1, "callsFake"), s;
     },
     resolvesOnce(r) {
-      return n(() => Promise.resolve(r), !0), o;
+      return n(() => Promise.resolve(r), !0, "resolves"), s;
     },
     rejectsOnce(r) {
-      return n(() => {
-        const u = typeof r == "string" ? new Error(r) : r;
-        return Promise.reject(u);
-      }, !0), o;
+      return n(
+        () => {
+          const c = typeof r == "string" ? new Error(r) : r;
+          return Promise.reject(c);
+        },
+        !0,
+        "rejects"
+      ), s;
     },
     callsFakeOnce(r) {
-      return n(r, !0), o;
+      return n(r, !0, "callsFake"), s;
     },
     resolvesStream(r) {
       return n(
-        () => Promise.resolve({ Body: h(r) }),
-        !1
-      ), o;
+        () => Promise.resolve({ Body: S(r) }),
+        !1,
+        "resolvesStream"
+      ), s;
     },
     resolvesStreamOnce(r) {
       return n(
-        () => Promise.resolve({ Body: h(r) }),
-        !0
-      ), o;
+        () => Promise.resolve({ Body: S(r) }),
+        !0,
+        "resolvesStream"
+      ), s;
     },
-    resolvesWithDelay(r, u) {
-      const i = (l) => {
-        setTimeout(() => l(r), u);
+    resolvesWithDelay(r, c) {
+      const l = (u) => {
+        setTimeout(() => u(r), c);
       };
-      return n(() => new Promise(i), !1), o;
+      return n(() => new Promise(l), !1, "resolvesWithDelay"), s;
     },
-    rejectsWithDelay(r, u) {
-      const i = typeof r == "string" ? new Error(r) : r, l = (a, f) => {
-        setTimeout(() => f(i), u);
+    rejectsWithDelay(r, c) {
+      const l = typeof r == "string" ? new Error(r) : r, u = (a, g) => {
+        setTimeout(() => g(l), c);
       };
-      return n(() => new Promise(l), !1), o;
+      return n(() => new Promise(u), !1, "rejectsWithDelay"), s;
     },
     rejectsWithNoSuchKey(r) {
-      return n(() => Promise.reject(C(r)), !1), o;
+      return n(
+        () => Promise.reject(T(r)),
+        !1,
+        "rejectsWithNoSuchKey"
+      ), s;
     },
     rejectsWithNoSuchBucket(r) {
-      return n(() => Promise.reject($(r)), !1), o;
+      return n(
+        () => Promise.reject(K(r)),
+        !1,
+        "rejectsWithNoSuchBucket"
+      ), s;
     },
     rejectsWithAccessDenied(r) {
-      return n(() => Promise.reject(D(r)), !1), o;
+      return n(
+        () => Promise.reject(A(r)),
+        !1,
+        "rejectsWithAccessDenied"
+      ), s;
     },
     rejectsWithResourceNotFound(r) {
       return n(
-        () => Promise.reject(M(r)),
-        !1
-      ), o;
+        () => Promise.reject(B(r)),
+        !1,
+        "rejectsWithResourceNotFound"
+      ), s;
     },
     rejectsWithConditionalCheckFailed() {
       return n(
-        () => Promise.reject(T()),
-        !1
-      ), o;
+        () => Promise.reject(J()),
+        !1,
+        "rejectsWithConditionalCheckFailed"
+      ), s;
     },
     rejectsWithThrottling() {
-      return n(() => Promise.reject(I()), !1), o;
+      return n(
+        () => Promise.reject(_()),
+        !1,
+        "rejectsWithThrottling"
+      ), s;
     },
     rejectsWithInternalServerError() {
-      return n(() => Promise.reject(O()), !1), o;
+      return n(
+        () => Promise.reject(q()),
+        !1,
+        "rejectsWithInternalServerError"
+      ), s;
     },
-    resolvesPaginated(r, u = {}) {
-      const i = L(r, u);
-      let l = 0;
-      return n((a) => {
-        const f = u.tokenKey || "NextToken", E = u.inputTokenKey || f, p = a[E];
-        if (p) {
-          const g = /token-(\d+)/.exec(p);
-          if (g && g[1]) {
-            const P = g[1];
-            l = Math.floor(
-              Number.parseInt(P, 10) / (u.pageSize || 10)
-            );
-          }
-        } else
-          l = 0;
-        const m = (
-          // eslint-disable-next-line security/detect-object-injection
-          i[l] || // eslint-disable-next-line unicorn/prefer-at -- TypeScript target doesn't support Array.at() method
-          i[i.length - 1] || i[0]
-        );
-        if (!m)
-          throw new Error("No paginated responses available");
-        return l = Math.min(l + 1, i.length - 1), Promise.resolve(m);
-      }, !1), o;
+    resolvesPaginated(r, c = {}) {
+      const l = U(r, c);
+      let u = 0;
+      return e.debugLogger.log(
+        `Configured resolvesPaginated for ${t.name}`,
+        { pageSize: c.pageSize, itemsCount: r.length }
+      ), n(
+        (a) => {
+          const g = c.tokenKey || "NextToken", f = c.inputTokenKey || g, b = a[f];
+          if (b != null) {
+            const R = c.itemsKey || "Items";
+            let k = 0;
+            for (const W of l) {
+              const m = W[R];
+              if (m && m.length > 0) {
+                const P = m[m.length - 1];
+                if (JSON.stringify(P) === JSON.stringify(b)) {
+                  u = k + 1;
+                  break;
+                }
+              }
+              k++;
+            }
+          } else
+            u = 0;
+          const y = (
+            // eslint-disable-next-line security/detect-object-injection
+            l[u] || // eslint-disable-next-line unicorn/prefer-at -- TypeScript target doesn't support Array.at() method
+            l[l.length - 1] || l[0]
+          );
+          if (!y)
+            throw new Error("No paginated responses available");
+          return u = Math.min(u + 1, l.length - 1), Promise.resolve(y);
+        },
+        !1,
+        "resolvesPaginated"
+      ), s;
     },
     resolvesFromFile(r) {
-      return n(() => {
-        const u = F(r);
-        return Promise.resolve(u);
-      }, !1), o;
+      return e.debugLogger.log(
+        `Configured resolvesFromFile for ${t.name}`,
+        { filePath: r }
+      ), n(
+        () => {
+          const c = V(r);
+          return Promise.resolve(c);
+        },
+        !1,
+        "resolvesFromFile"
+      ), s;
     }
   };
-  return o;
+  return s;
 }
-const G = (t) => {
-  const e = {
+const ie = (e) => {
+  const t = {
     map: /* @__PURE__ */ new WeakMap(),
-    debugLogger: k()
-  }, s = t.prototype, c = b.spyOn(s, "send").mockImplementation(x(e));
+    debugLogger: L()
+  }, o = e.prototype, i = j.spyOn(o, "send").mockImplementation(E(t));
   return {
     client: void 0,
-    on: (o, r, u) => S(
-      e,
-      o,
+    on: (s, r, c) => C(
+      t,
+      s,
       r,
-      u
+      c
     ),
     reset: () => {
-      c.mockClear(), e.map = /* @__PURE__ */ new WeakMap();
+      p(t.debugLogger) && t.debugLogger.logDirect(
+        "Clearing call history (mocks preserved)"
+      ), i.mockClear();
     },
     restore: () => {
-      c.mockRestore(), e.map = /* @__PURE__ */ new WeakMap();
+      p(t.debugLogger) && t.debugLogger.logDirect(
+        "Restoring original client behavior and clearing all mocks"
+      ), i.mockRestore(), t.map = /* @__PURE__ */ new WeakMap();
     },
-    calls: () => c.mock.calls.map((o) => o[0]),
-    __rawCalls: () => c.mock.calls,
+    calls: () => i.mock.calls.map((s) => s[0]),
+    __rawCalls: () => i.mock.calls,
     enableDebug: () => {
-      y(e.debugLogger);
+      w(t.debugLogger);
     },
     disableDebug: () => {
-      w(e.debugLogger);
+      D(t.debugLogger);
     }
   };
-}, H = (t) => {
-  const e = {
+}, le = (e) => {
+  const t = {
     map: /* @__PURE__ */ new WeakMap(),
-    debugLogger: k()
-  }, s = b.spyOn(t, "send").mockImplementation(x(e));
+    debugLogger: L()
+  }, o = j.spyOn(e, "send").mockImplementation(E(t));
   return {
-    client: t,
-    on: (n, o, r) => S(
-      e,
+    client: e,
+    on: (n, s, r) => C(
+      t,
       n,
-      o,
+      s,
       r
     ),
     reset: () => {
-      s.mockClear(), e.map = /* @__PURE__ */ new WeakMap();
+      p(t.debugLogger) && t.debugLogger.logDirect(
+        "Clearing call history (mocks preserved) for client instance"
+      ), o.mockClear();
     },
     restore: () => {
-      s.mockRestore(), e.map = /* @__PURE__ */ new WeakMap();
+      p(t.debugLogger) && t.debugLogger.logDirect(
+        "Restoring original client behavior and clearing all mocks for client instance"
+      ), o.mockRestore(), t.map = /* @__PURE__ */ new WeakMap();
     },
-    calls: () => s.mock.calls.map((n) => n[0]),
-    __rawCalls: () => s.mock.calls,
+    calls: () => o.mock.calls.map((n) => n[0]),
+    __rawCalls: () => o.mock.calls,
     enableDebug: () => {
-      y(e.debugLogger);
+      w(t.debugLogger);
     },
     disableDebug: () => {
-      w(e.debugLogger);
+      D(t.debugLogger);
     }
   };
 };
 export {
-  Y as matchers,
-  G as mockClient,
-  H as mockClientInstance
+  ge as matchers,
+  ie as mockClient,
+  le as mockClientInstance,
+  ce as setGlobalDebug
 };