commit b4ed7d2b8fb5699024bee6317c4dc47ecfd32240
author Jeremiah Griffin <jzgriffin@google.com> Mon Jul 31 15:20:54 2023 -0700
committer Jeremiah Griffin <jzgriffin@google.com> Tue Aug 08 09:46:01 2023 -0700
tree 1a129ce76ad87f5add83a929e4286bc181e3eb8c
parent efccf39ab2d865fe569c898f6be268d8345f64c7

tests: Add NDK program zlib tests

Test: berberis_run_host_tests
Bug: 276830819
Change-Id: I60b609e7bfe1a539585b73f5848cd630bc7a5b77

tests/ndk_program_tests/Android.bp

diff --git a/tests/ndk_program_tests/Android.bp b/tests/ndk_program_tests/Android.bp
index a00e171..3d3bc9e 100644
--- a/tests/ndk_program_tests/Android.bp
+++ b/tests/ndk_program_tests/Android.bp
@@ -25,6 +25,7 @@
         "div_test.cc",
         "string_test.cc",
         "time_test.cc",
+        "zlib_test.cc",
     ],
 }
 
@@ -45,10 +46,12 @@
 cc_test {
     name: "berberis_ndk_program_tests",
     defaults: ["berberis_ndk_program_tests_defaults"],
+    shared_libs: ["libz"],
 }
 
 cc_test {
     name: "berberis_ndk_program_tests_static",
     defaults: ["berberis_ndk_program_tests_defaults"],
+    static_libs: ["libz"],
     static_executable: true,
 }

tests/ndk_program_tests/zlib_test.cc

diff --git a/tests/ndk_program_tests/zlib_test.cc b/tests/ndk_program_tests/zlib_test.cc
new file mode 100644
index 0000000..53392fd
--- /dev/null
+++ b/tests/ndk_program_tests/zlib_test.cc
@@ -0,0 +1,70 @@
+/*
+ * Copyright (C) 2023 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "gtest/gtest.h"
+
+#include <string.h>
+#include <zlib.h>
+
+const char kTestString[] = "compressed string";
+
+void* ZAlloc(void* opaque, uint32_t items, uint32_t size) {
+  *reinterpret_cast<int*>(opaque) |= 1;
+  return calloc(items, size);
+}
+
+void ZFree(void* opaque, void* address) {
+  *reinterpret_cast<int*>(opaque) |= 2;
+  free(address);
+}
+
+TEST(ZLib, Deflate) {
+  char input[1024];
+  char output[1024];
+
+  z_stream encode_stream;
+  memset(&encode_stream, 0, sizeof(encode_stream));
+  strncpy(input, kTestString, sizeof(input));
+  encode_stream.next_in = reinterpret_cast<Bytef*>(input);
+  encode_stream.avail_in = strlen(input) + 1;
+  encode_stream.next_out = reinterpret_cast<Bytef*>(output);
+  encode_stream.avail_out = sizeof(output);
+  encode_stream.zalloc = ZAlloc;
+  encode_stream.zfree = ZFree;
+  int opaque = 0;
+  encode_stream.opaque = &opaque;
+  ASSERT_EQ(deflateInit(&encode_stream, Z_BEST_COMPRESSION), Z_OK);
+  ASSERT_EQ(deflate(&encode_stream, Z_FINISH), Z_STREAM_END);
+  ASSERT_EQ(deflateEnd(&encode_stream), Z_OK);
+  EXPECT_EQ(opaque, 3);
+
+  z_stream decode_stream;
+  memset(&decode_stream, 0, sizeof(decode_stream));
+  decode_stream.next_in = reinterpret_cast<Bytef*>(output);
+  decode_stream.avail_in = encode_stream.total_out;
+  decode_stream.next_out = reinterpret_cast<Bytef*>(input);
+  decode_stream.avail_out = sizeof(input);
+  opaque = 0;
+  decode_stream.zalloc = ZAlloc;
+  decode_stream.zfree = ZFree;
+  decode_stream.opaque = &opaque;
+  ASSERT_EQ(inflateInit(&decode_stream), Z_OK);
+  ASSERT_EQ(inflate(&decode_stream, Z_FINISH), Z_STREAM_END);
+  ASSERT_EQ(inflateEnd(&decode_stream), Z_OK);
+  EXPECT_EQ(decode_stream.total_out, sizeof(kTestString));
+  EXPECT_EQ(strncmp(input, kTestString, sizeof(input)), 0);
+  EXPECT_EQ(opaque, 3);
+}