nwaku/tests/common/test_base64_codec.nim

{.used.}

import std/strutils, stew/[results, byteutils], testutils/unittests
import waku/common/base64

suite "Waku Common - stew base64 wrapper":
  const TestData =
    @[
      # Test vectors from RFC 4648
      # See: https://datatracker.ietf.org/doc/html/rfc4648#section-10
      ("", Base64String("")),
      ("f", Base64String("Zg==")),
      ("fo", Base64String("Zm8=")),
      ("foo", Base64String("Zm9v")),
      ("foob", Base64String("Zm9vYg==")),
      ("fooba", Base64String("Zm9vYmE=")),
      ("foobar", Base64String("Zm9vYmFy")),

      # Custom test vectors
      ("\x01", Base64String("AQ==")),
      ("\x13", Base64String("Ew==")),
      ("\x01\x02\x03\x04", Base64String("AQIDBA==")),
    ]

  for (plaintext, encoded) in TestData:
    test "encode into base64 (" & escape(plaintext) & " -> \"" & string(encoded) & "\")":
      ## Given
      let data = plaintext

      ## When
      let encodedData = base64.encode(data)

      ## Then
      check:
        encodedData == encoded

    test "decode from base64 (\"" & string(encoded) & "\" -> " & escape(plaintext) & ")":
      ## Given
      let data = encoded

      ## When
      let decodedRes = base64.decode(data)

      ## Then
      check:
        decodedRes.isOk()

      let decoded = decodedRes.tryGet()
      check:
        decoded == toBytes(plaintext)