test(json-crdt): 💍 add ability to collect fuzzer traces

Author: streamich

src/json-crdt/__tests__/fuzzer/JsonCrdtFuzzer.ts

@@ -5,6 +5,7 @@ import {FuzzerOptions} from './types';
 import {RandomJson} from '../../../json-random/RandomJson';
 import {generateInteger} from './util';
 import {PatchBuilder} from '../../../json-crdt-patch/PatchBuilder';
+import {Patch} from '../../../json-crdt-patch';
 
 export const defaultFuzzerOptions: FuzzerOptions = {
   startingValue: undefined,
@@ -18,12 +19,14 @@ export const defaultFuzzerOptions: FuzzerOptions = {
   concurrentPeers: [1, 6],
   patchesPerPeer: [0, 12],
   testCodecs: true,
+  collectPatches: false,
 };
 
 export class JsonCrdtFuzzer {
   public opts: FuzzerOptions;
   public model: Model;
   public picker: Picker;
+  public patches: Patch[] = [];
 
   constructor(opts: Partial<FuzzerOptions> = {}) {
     this.opts = {...defaultFuzzerOptions, ...opts};
@@ -39,6 +42,7 @@ export class JsonCrdtFuzzer {
     const builder = new PatchBuilder(this.model.clock);
     builder.root(builder.json(json));
     const patch = builder.flush();
+    this.patches.push(patch);
     this.model.applyPatch(patch);
   }
 
@@ -47,6 +51,8 @@ export class JsonCrdtFuzzer {
     const session = new SessionLogical(this, concurrency);
     session.generateEdits();
     session.synchronize();
+    if (this.opts.collectPatches)
+      for (const patches of session.patches) this.patches.push(...patches);
     return session;
   }
 }

src/json-crdt/__tests__/fuzzer/generate-trace.ts

@@ -0,0 +1,39 @@
+// Run: npx ts-node src/json-crdt/__tests__/fuzzer/generate-trace.ts
+
+import {Patch} from '../../../json-crdt-patch';
+import {Model} from '../../model';
+import {JsonCrdtFuzzer} from './JsonCrdtFuzzer';
+import {CborEncoder} from '../../../json-pack/cbor/CborEncoder';
+import {Writer} from '../../../util/buffers/Writer';
+import * as fs from 'fs';
+
+const sessionNum = 100;
+const fuzzer = new JsonCrdtFuzzer({
+  // concurrentPeers: [20, 100],
+  // startingValue: new Uint8Array([1, 2, 3]),
+  collectPatches: true,
+});
+fuzzer.setupModel();
+
+for (let ses = 0; ses < sessionNum; ses++) {
+  fuzzer.executeConcurrentSession();
+}
+
+const patches: Patch[] = [];
+const dupes: Set<string> = new Set();
+for (const patch of fuzzer.patches) {
+  const key = `${patch.getId()?.sid}_${patch.getId()?.time}`;
+  if (dupes.has(key)) continue;
+  dupes.add(key);
+  patches.push(patch);
+}
+
+const model = Model.withLogicalClock();
+model.applyBatch(patches);
+const cborEncoder = new CborEncoder(new Writer());
+fs.writeFileSync(__dirname + '/trace.cbor', cborEncoder.encode(patches.map(p => p.toBinary())));
+
+// console.log(Buffer.from(jsonEncoder.encode(encoded)).toString());
+console.log(model.view());
+console.log(fuzzer.model.view());
+// console.log(model + '');

src/json-crdt/__tests__/fuzzer/types.ts

@@ -34,4 +34,7 @@ export interface FuzzerOptions {
 
   /** Do not generate "__proto__" as a string, so it does not appear as object key. */
   noProtoString?: boolean;
+
+  /** Whether to collect all generated patches. */
+  collectPatches?: boolean;
 }