Add test code for Pl_Concatenate

Jay Berkenbilt
1 parent f07e3370
Showing 2 changed files with 39 additions and 0 deletions
libtests/build.mk
libtests/concatenate.cc
@@ -3,6 +3,7 @@ BINS_libtests = \
 	ascii85 \
 	bits \
 	buffer \
+	concatenate \
 	flate \
 	hex \
 	lzw \
+#include <qpdf/Pl_Concatenate.hh>
+#include <qpdf/Pl_Flate.hh>
+#include <qpdf/Pl_Buffer.hh>
+#include <iostream>
+#include <assert.h>
+
+static void pipeStringAndFinish(Pipeline* p, std::string const& str)
+{
+    p->write((unsigned char*)str.c_str(), str.length());
+    p->finish();
+}
+
+int main(int argc, char* argv[])
+{
+    Pl_Buffer b1("compressed");
+    Pl_Flate deflate("compress", &b1, Pl_Flate::a_deflate);
+    Pl_Concatenate concat("concat", &deflate);
+    pipeStringAndFinish(&concat, "-one-");
+    pipeStringAndFinish(&concat, "-two-");
+    concat.manualFinish();
+
+    PointerHolder<Buffer> b1_buf = b1.getBuffer();
+    Pl_Buffer b2("uncompressed");
+    Pl_Flate inflate("uncompress", &b2, Pl_Flate::a_inflate);
+    inflate.write(b1_buf->getBuffer(), b1_buf->getSize());
+    inflate.finish();
+    PointerHolder<Buffer> b2_buf = b2.getBuffer();
+    std::string result((char const*)b2_buf->getBuffer(), b2_buf->getSize());
+    if (result == "-one--two-")
+    {
+        std::cout << "concatenate test passed" << std::endl;
+    }
+    else
+    {
+        std::cout << "concatenate test failed: " << result << std::endl;
+    }
+    return 0;
+}