From 7f266701734940122e1e65398f56f386b3465caf Mon Sep 17 00:00:00 2001
From: hagen <hagen@mail.i2p>
Date: Tue, 28 Jun 2016 00:00:00 +0000
Subject: [PATCH] + tests/test-base-64.cpp

---
 tests/Makefile         |  7 +++++--
 tests/test-base-64.cpp | 45 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 2 deletions(-)
 create mode 100644 tests/test-base-64.cpp

diff --git a/tests/Makefile b/tests/Makefile
index 957d4632..ef30c631 100644
--- a/tests/Makefile
+++ b/tests/Makefile
@@ -1,11 +1,14 @@
 CXXFLAGS += -Wall -Wextra -pedantic -O0 -g -std=c++11 -D_GLIBCXX_USE_NANOSLEEP=1
 
 TESTS = test-http-url test-http-req test-http-res test-http-url_decode \
-  test-http-merge_chunked
+  test-http-merge_chunked test-base-64
 
 all: $(TESTS) run
 
-test-http-%: test-http-%.cpp ../HTTP.cpp
+test-http-%: ../HTTP.cpp test-http-%.cpp
+	$(CXX) $(CXXFLAGS) $(NEEDED_CXXFLAGS) $(INCFLAGS) -o $@ $^
+
+test-base-%: ../Base.cpp test-base-%.cpp
 	$(CXX) $(CXXFLAGS) $(NEEDED_CXXFLAGS) $(INCFLAGS) -o $@ $^
 
 run: $(TESTS)
diff --git a/tests/test-base-64.cpp b/tests/test-base-64.cpp
new file mode 100644
index 00000000..d8d0ce2f
--- /dev/null
+++ b/tests/test-base-64.cpp
@@ -0,0 +1,45 @@
+#include <cassert>
+#include <string.h>
+
+#include "../Base.h"
+
+using namespace i2p::data;
+
+int main() {
+  const char *in = "test";
+  size_t in_len = strlen(in);
+  char out[16];
+
+  /* bytes -> b64 */
+  assert(ByteStreamToBase64(NULL, 0, NULL, 0) == 0);
+  assert(ByteStreamToBase64(NULL, 0, out, sizeof(out)) == 0);
+
+  assert(Base64EncodingBufferSize(2)  == 4);
+  assert(Base64EncodingBufferSize(4)  == 8);
+  assert(Base64EncodingBufferSize(6)  == 8);
+  assert(Base64EncodingBufferSize(7)  == 12);
+  assert(Base64EncodingBufferSize(9)  == 12);
+  assert(Base64EncodingBufferSize(10) == 16);
+  assert(Base64EncodingBufferSize(12) == 16);
+  assert(Base64EncodingBufferSize(13) == 20);
+
+  assert(ByteStreamToBase64((uint8_t *) in, in_len, out, sizeof(out)) == 8);
+  assert(memcmp(out, "dGVzdA==", 8) == 0);
+
+  /* b64 -> bytes */
+  assert(Base64ToByteStream(NULL, 0, NULL, 0) == 0);
+  assert(Base64ToByteStream(NULL, 0, (uint8_t *) out, sizeof(out)) == 0);
+
+  in = "dGVzdA=="; /* valid b64 */
+  assert(Base64ToByteStream(in, strlen(in), (uint8_t *) out, sizeof(out)) == 4);
+  assert(memcmp(out, "test", 4) == 0);
+
+  in = "dGVzdA=";  /* invalid b64 : not padded */
+  assert(Base64ToByteStream(in, strlen(in), (uint8_t *) out, sizeof(out)) == 0);
+
+  in = "dG/z.A=="; /* invalid b64 : char not from alphabet */
+//  assert(Base64ToByteStream(in, strlen(in), (uint8_t *) out, sizeof(out)) == 0);
+//  ^^^ fails, current implementation not checks acceptable symbols
+
+  return 0;
+}