qpdf/libtests/concatenate.cc

41 lines
1.2 KiB
C++

#include <qpdf/Pl_Concatenate.hh>
#include <qpdf/Pl_Flate.hh>
#include <qpdf/Pl_Buffer.hh>
#include <qpdf/QUtil.hh>
#include <iostream>
#include <assert.h>
static void pipeStringAndFinish(Pipeline* p, std::string const& str)
{
p->write(QUtil::unsigned_char_pointer(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(reinterpret_cast<char*>(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;
}