packages/docs-core/lib/UseCase/SquashAlgorithm.spec.ts

   1 import * as Y from 'yjs'
   2
   3 describe('SquashAlgorithm', () => {
   4   describe('yjs merge', () => {
   5     const harvestUpdates = async (doc: Y.Doc, count: number): Promise<Uint8Array[]> => {
   6       return new Promise((resolve) => {
   7         const updates: Uint8Array[] = []
   8
   9         doc.on('update', (update) => {
  10           updates.push(update)
  11
  12           if (updates.length === count) {
  13             resolve(updates)
  14           }
  15         })
  16       })
  17     }
  18
  19     it('should be able to merge a single update', async () => {
  20       const doc = new Y.Doc()
  21
  22       const updatesPromise = harvestUpdates(doc, 1)
  23
  24       doc.getText().insert(0, 'H')
  25
  26       const updates = await updatesPromise
  27
  28       const squashed = Y.mergeUpdates(updates)
  29
  30       const reassembledDoc = new Y.Doc()
  31       Y.applyUpdate(reassembledDoc, squashed)
  32
  33       expect(reassembledDoc.getText().toString()).toEqual('H')
  34     })
  35
  36     it('should be able to merge multiple updates', async () => {
  37       const doc = new Y.Doc()
  38
  39       const updatesPromise = harvestUpdates(doc, 3)
  40
  41       doc.getText().insert(0, 'H')
  42       doc.getText().insert(1, 'e')
  43       doc.getText().insert(2, 'y')
  44
  45       const updates = await updatesPromise
  46
  47       const squashed = Y.mergeUpdates(updates)
  48
  49       const reassembledDoc = new Y.Doc()
  50       Y.applyUpdate(reassembledDoc, squashed)
  51
  52       expect(reassembledDoc.getText().toString()).toEqual('Hey')
  53     })
  54
  55     it('should be able to merge only a tail subset of updates', async () => {
  56       const doc = new Y.Doc()
  57
  58       const updatesPromise = harvestUpdates(doc, 5)
  59
  60       doc.getText().insert(0, 'H')
  61       doc.getText().insert(1, 'e')
  62       doc.getText().insert(2, 'l')
  63       doc.getText().insert(3, 'l')
  64       doc.getText().insert(4, 'o')
  65
  66       const updates = await updatesPromise
  67
  68       const squashed = Y.mergeUpdates(updates.slice(-2))
  69
  70       const reassembledDoc = new Y.Doc()
  71       for (const update of updates.slice(0, 3)) {
  72         Y.applyUpdate(reassembledDoc, update)
  73       }
  74       Y.applyUpdate(reassembledDoc, squashed)
  75
  76       expect(reassembledDoc.getText().toString()).toEqual('Hello')
  77     })
  78
  79     it('should be able to merge only a tail subset of updates with originals being squashed as well', async () => {
  80       const doc = new Y.Doc()
  81
  82       const updatesPromise = harvestUpdates(doc, 5)
  83
  84       doc.getText().insert(0, 'H')
  85       doc.getText().insert(1, 'e')
  86       doc.getText().insert(2, 'l')
  87       doc.getText().insert(3, 'l')
  88       doc.getText().insert(4, 'o')
  89
  90       const updates = await updatesPromise
  91
  92       const squashed = Y.mergeUpdates(updates.slice(-2))
  93
  94       const reassembledDoc = new Y.Doc()
  95
  96       Y.applyUpdate(reassembledDoc, Y.mergeUpdates(updates.slice(0, 3)))
  97       Y.applyUpdate(reassembledDoc, squashed)
  98
  99       expect(reassembledDoc.getText().toString()).toEqual('Hello')
 100     })
 101   })
 102 })