[encoding] add encode helper function

dmonad · May 25, 2023 · 79c5082 · 79c5082
1 parent 3c8ce61
commit 79c5082
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/encoding.js b/encoding.js
@@ -53,6 +53,15 @@ export class Encoder {
  */
 export const createEncoder = () => new Encoder()
 
+/**
+ * @param {function(Encoder):void} f
+ */
+export const encode = (f) => {
+  const encoder = createEncoder()
+  f(encoder)
+  return toUint8Array(encoder)
+}
+
 /**
  * The current length of the encoded data.
  *

diff --git a/encoding.test.js b/encoding.test.js
@@ -110,9 +110,7 @@ function test (testname, write, read, val, doLog = true) {
  * @param {string} s
  */
 const testVarString = s => {
-  const encoder = encoding.createEncoder()
-  encoding.writeVarString(encoder, s)
-  const decoder = decoding.createDecoder(encoding.toUint8Array(encoder))
+  const decoder = decoding.createDecoder(encoding.encode(encoder => { encoding.writeVarString(encoder, s) }))
   const peeked = decoding.peekVarString(decoder)
   const result = decoding.readVarString(decoder)
   t.compareStrings(s, result)