BitBuilder

1 files changed, 58 insertions, 0 deletions

diff --git a/test/gleam/bit_string_builder_test.gleam b/test/gleam/bit_string_builder_test.gleam
new file mode 100644
index 0000000..43e010a
--- /dev/null
+++ b/test/gleam/bit_string_builder_test.gleam
@@ -0,0 +1,58 @@
+import gleam/should
+import gleam/bit_builder
+
+pub fn builder_test() {
+  let data = bit_builder.from_bit_string(<<1>>)
+    |> bit_builder.append(<<2>>)
+    |> bit_builder.append(<<3>>)
+    |> bit_builder.prepend(<<0>>)
+
+  data
+  |> bit_builder.to_bit_string
+  |> should.equal(<<0, 1, 2, 3>>)
+
+  data
+  |> bit_builder.byte_size
+  |> should.equal(4)
+}
+
+pub fn builder_with_strings_test() {
+  let data = bit_builder.from_bit_string(<<1>>)
+    |> bit_builder.append_string("2")
+    |> bit_builder.append_string("3")
+    |> bit_builder.prepend_string("0")
+
+  data
+  |> bit_builder.to_bit_string
+  |> should.equal(<<"0":utf8, 1, "2":utf8, "3":utf8>>)
+
+  data
+  |> bit_builder.byte_size
+  |> should.equal(4)
+}
+
+pub fn builder_with_builders_test() {
+  let data = bit_builder.from_bit_string(<<1>>)
+    |> bit_builder.append_builder(bit_builder.from_bit_string(<<2>>))
+    |> bit_builder.append_builder(bit_builder.from_bit_string(<<3>>))
+    |> bit_builder.prepend_builder(bit_builder.from_bit_string(<<0>>))
+
+  data
+  |> bit_builder.to_bit_string
+  |> should.equal(<<0, 1, 2, 3>>)
+
+  data
+  |> bit_builder.byte_size
+  |> should.equal(4)
+}
+
+pub fn concat_test() {
+  [
+    bit_builder.from_bit_string(<<1, 2>>),
+    bit_builder.from_bit_string(<<3, 4>>),
+    bit_builder.from_bit_string(<<5, 6>>),
+  ]
+  |> bit_builder.concat
+  |> bit_builder.to_bit_string
+  |> should.equal(<<1, 2, 3, 4, 5, 6>>)
+}