Function test

examples/plugin/wasmedge-zlib/main.cpp:299–344

+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Test function and main +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

// Test function and main
// +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
int test() {
  const size_t DataSize = 1 * 1024 * 1024;
  std::vector<uint8_t> Data(DataSize);
  std::generate(Data.begin(), Data.end(), []() -> uint8_t {
    constexpr char Charset[] = "0123456789"
                               "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
                               "abcdefghijklmnopqrstuvwxyz";
    constexpr size_t MaxIndex = (sizeof(Charset) - 1);
    return static_cast<uint8_t>(Charset[rand() % MaxIndex]);
  });

  std::cout << "Compressing Buffer of size : " << DataSize << "B" << std::endl;

  // Test compression
  size_t CompressedSize = DataSize * 2;
  std::vector<uint8_t> Compressed(CompressedSize);
  // Use zlibCustomCompress
  int32_t Result = zlibCustomCompress(Compressed.data(), &CompressedSize,
                                      Data.data(), Data.size());
  if (Result != 0) {
    std::cerr << "Compression failed with error code: " << Result << std::endl;
    return 1;
  }
  Compressed.resize(CompressedSize);

  std::cout << "Decompressing Buffer of size : " << Compressed.size() << "B"
            << std::endl;

  // Test decompression
  size_t DecompressedSize = DataSize;
  std::vector<uint8_t> Decompressed(DecompressedSize);

  // Use zlibCustomUncompress
  Result = zlibCustomUncompress(Decompressed.data(), &DecompressedSize,
                                Compressed.data(), Compressed.size());
  if (Result != 0) {
    std::cerr << "Decompression failed with error code: " << Result
              << std::endl;
    return 1;
  }
  Decompressed.resize(DecompressedSize);

  bool const Success = (Data == Decompressed);
  std::cout << (Success ? "Success" : "Fail") << std::endl;
  return Success ? 0 : 1;
}

int main() { return test(); }