@hegeldev/hegel 0.1.0

package/dist/testCase.js

@@ -0,0 +1,186 @@
+/**
+ * TestCase: per-test-case state passed explicitly to test functions.
+ *
+ * Provides draw(), assume(), note(), span management, and the Collection
+ * class for server-managed collection sizing.
+ *
+ * @packageDocumentation
+ */
+import { inspect } from "node:util";
+export class StopTestError extends Error {
+    constructor() {
+        super("Server ran out of data (StopTest)");
+        this.name = "StopTestError";
+    }
+}
+export class AssumeError extends Error {
+    constructor() {
+        super("Assumption rejected");
+        this.name = "AssumeError";
+    }
+}
+// ---------------------------------------------------------------------------
+// Span labels
+// ---------------------------------------------------------------------------
+export const Labels = {
+    LIST: 1,
+    LIST_ELEMENT: 2,
+    SET: 3,
+    SET_ELEMENT: 4,
+    MAP: 5,
+    MAP_ENTRY: 6,
+    TUPLE: 7,
+    ONE_OF: 8,
+    OPTIONAL: 9,
+    FIXED_DICT: 10,
+    FLAT_MAP: 11,
+    FILTER: 12,
+    MAPPED: 13,
+    SAMPLED_FROM: 14,
+    ENUM_VARIANT: 15,
+};
+export class TestCase {
+    _dataSource;
+    _isLastRun;
+    drawCount = 0;
+    spanDepth = 0;
+    /** @internal */
+    constructor(dataSource, isLastRun) {
+        this._dataSource = dataSource;
+        this._isLastRun = isLastRun;
+    }
+    /** @internal */
+    dataSource() {
+        return this._dataSource;
+    }
+    get isLastRun() {
+        return this._isLastRun;
+    }
+    get testAborted() {
+        return this._dataSource.testAborted();
+    }
+    /**
+     * Draw a value from a generator.
+     */
+    draw(generator) {
+        const value = generator.doDraw(this);
+        if (this.spanDepth === 0) {
+            this.drawCount++;
+            if (this._isLastRun) {
+                console.error(`var draw_${this.drawCount} = ${inspect(value, { depth: null })};`);
+            }
+        }
+        return value;
+    }
+    /**
+     * Draw a value from a generator without recording it in the output.
+     * Unlike draw(), this does not print the value during the final replay.
+     */
+    drawSilent(generator) {
+        return generator.doDraw(this);
+    }
+    /**
+     * Reject the current test case if the condition is false.
+     */
+    assume(condition) {
+        if (!condition) {
+            throw new AssumeError();
+        }
+    }
+    /**
+     * Note a message that will be displayed during the final replay.
+     */
+    note(message) {
+        if (this._isLastRun) {
+            console.error(message);
+        }
+    }
+    /**
+     * Start a shrinking span with the given label.
+     */
+    startSpan(label) {
+        this.spanDepth++;
+        try {
+            this._dataSource.startSpan(label);
+        }
+        catch (e) {
+            this.spanDepth--;
+            throw e;
+        }
+    }
+    /**
+     * Stop the current shrinking span.
+     */
+    stopSpan(discard = false) {
+        this.spanDepth--;
+        try {
+            this._dataSource.stopSpan(discard);
+        }
+        catch {
+            // Ignore errors during stop_span (matches Rust: `let _ = ...`)
+        }
+    }
+}
+/**
+ * Send a generate command to the data source and return the raw result.
+ * Throws StopTestError if the data source runs out of data.
+ * @internal
+ */
+export function generateRaw(tc, schema) {
+    return tc.dataSource().generate(schema);
+}
+/**
+ * Server-managed collection sizing.
+ *
+ * The server determines how many elements to generate based on
+ * min_size, max_size, and shrinking state.
+ */
+export class Collection {
+    dataSource;
+    minSize;
+    maxSize;
+    collectionId = null;
+    finished = false;
+    constructor(tc, minSize, maxSize) {
+        this.dataSource = tc.dataSource();
+        this.minSize = minSize;
+        this.maxSize = maxSize;
+    }
+    ensureInitialized() {
+        if (this.collectionId === null) {
+            this.collectionId = this.dataSource.newCollection(this.minSize, this.maxSize);
+        }
+        return this.collectionId;
+    }
+    /**
+     * Ask the server whether to produce another element.
+     */
+    more() {
+        if (this.finished) {
+            return false;
+        }
+        const collectionId = this.ensureInitialized();
+        let result;
+        try {
+            result = this.dataSource.collectionMore(collectionId);
+        }
+        catch (e) {
+            this.finished = true;
+            throw e;
+        }
+        if (!result) {
+            this.finished = true;
+        }
+        return result;
+    }
+    /**
+     * Reject the last element (don't count towards size budget).
+     */
+    reject(why) {
+        if (this.finished)
+            return;
+        const collectionId = this.ensureInitialized();
+        this.dataSource.collectionReject(collectionId, why);
+    }
+}
+//# sourceMappingURL=testCase.js.map

package/dist/testCase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testCase.js","sourceRoot":"","sources":["../src/testCase.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,OAAO,aAAc,SAAQ,KAAK;IACtC;QACE,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;IAC9B,CAAC;CACF;AAED,MAAM,OAAO,WAAY,SAAQ,KAAK;IACpC;QACE,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;IAC5B,CAAC;CACF;AAED,8EAA8E;AAC9E,cAAc;AACd,8EAA8E;AAE9E,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,IAAI,EAAE,CAAC;IACP,YAAY,EAAE,CAAC;IACf,GAAG,EAAE,CAAC;IACN,WAAW,EAAE,CAAC;IACd,GAAG,EAAE,CAAC;IACN,SAAS,EAAE,CAAC;IACZ,KAAK,EAAE,CAAC;IACR,MAAM,EAAE,CAAC;IACT,QAAQ,EAAE,CAAC;IACX,UAAU,EAAE,EAAE;IACd,QAAQ,EAAE,EAAE;IACZ,MAAM,EAAE,EAAE;IACV,MAAM,EAAE,EAAE;IACV,YAAY,EAAE,EAAE;IAChB,YAAY,EAAE,EAAE;CACR,CAAC;AAwBX,MAAM,OAAO,QAAQ;IACX,WAAW,CAAa;IACxB,UAAU,CAAU;IACpB,SAAS,GAAG,CAAC,CAAC;IACd,SAAS,GAAG,CAAC,CAAC;IAEtB,gBAAgB;IAChB,YAAY,UAAsB,EAAE,SAAkB;QACpD,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAED,gBAAgB;IAChB,UAAU;QACR,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,IAAI,CAAI,SAA2B;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,IAAI,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,OAAO,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,SAAS,MAAM,OAAO,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC;YACpF,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;OAGG;IACH,UAAU,CAAI,SAA2B;QACvC,OAAO,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,SAAkB;QACvB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,WAAW,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,OAAe;QAClB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,KAAa;QACrB,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC;YACH,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,OAAO,GAAG,KAAK;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC;YACH,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,+DAA+D;QACjE,CAAC;IACH,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,UAAU,WAAW,CAAC,EAAY,EAAE,MAA+B;IACvE,OAAO,EAAE,CAAC,UAAU,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC1C,CAAC;AAED;;;;;GAKG;AACH,MAAM,OAAO,UAAU;IACb,UAAU,CAAa;IACvB,OAAO,CAAS;IAChB,OAAO,CAAqB;IAC5B,YAAY,GAAkB,IAAI,CAAC;IACnC,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAAY,EAAY,EAAE,OAAe,EAAE,OAAgB;QACzD,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;YAC/B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,IAAI;QACF,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC9C,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;YACrB,MAAM,CAAC,CAAC;QACV,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACvB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,GAAY;QACjB,IAAI,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC9C,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;IACtD,CAAC;CACF"}