79 lines
2.6 KiB
C++
79 lines
2.6 KiB
C++
|
#include "compression.hh"
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
namespace nix {
|
||
|
|
|
||
|
|
/* ----------------------------------------------------------------------------
|
||
|
|
* compress / decompress
|
||
|
|
* --------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
TEST(compress, compressWithUnknownMethod) {
|
||
|
|
ASSERT_THROW(compress("invalid-method", "something-to-compress"), UnknownCompressionMethod);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(compress, noneMethodDoesNothingToTheInput) {
|
||
|
|
ref<std::string> o = compress("none", "this-is-a-test");
|
||
|
|
|
||
|
|
ASSERT_EQ(*o, "this-is-a-test");
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(decompress, decompressXzCompressed) {
|
||
|
|
auto method = "xz";
|
||
|
|
auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf";
|
||
|
|
ref<std::string> o = decompress(method, *compress(method, str));
|
||
|
|
|
||
|
|
ASSERT_EQ(*o, str);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(decompress, decompressBzip2Compressed) {
|
||
|
|
auto method = "bzip2";
|
||
|
|
auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf";
|
||
|
|
ref<std::string> o = decompress(method, *compress(method, str));
|
||
|
|
|
||
|
|
ASSERT_EQ(*o, str);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(decompress, decompressBrCompressed) {
|
||
|
|
auto method = "br";
|
||
|
|
auto str = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf";
|
||
|
|
ref<std::string> o = decompress(method, *compress(method, str));
|
||
|
|
|
||
|
|
ASSERT_EQ(*o, str);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(decompress, decompressInvalidInputThrowsCompressionError) {
|
||
|
|
auto method = "bzip2";
|
||
|
|
auto str = "this is a string that does not qualify as valid bzip2 data";
|
||
|
|
|
||
|
|
ASSERT_THROW(decompress(method, str), CompressionError);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* ----------------------------------------------------------------------------
|
||
|
|
* compression sinks
|
||
|
|
* --------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
TEST(makeCompressionSink, noneSinkDoesNothingToInput) {
|
||
|
|
StringSink strSink;
|
||
|
|
auto inputString = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf";
|
||
|
|
auto sink = makeCompressionSink("none", strSink);
|
||
|
|
(*sink)(inputString);
|
||
|
|
sink->finish();
|
||
|
|
|
||
|
|
ASSERT_STREQ((*strSink.s).c_str(), inputString);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(makeCompressionSink, compressAndDecompress) {
|
||
|
|
StringSink strSink;
|
||
|
|
auto inputString = "slfja;sljfklsa;jfklsjfkl;sdjfkl;sadjfkl;sdjf;lsdfjsadlf";
|
||
|
|
auto decompressionSink = makeDecompressionSink("bzip2", strSink);
|
||
|
|
auto sink = makeCompressionSink("bzip2", *decompressionSink);
|
||
|
|
|
||
|
|
(*sink)(inputString);
|
||
|
|
sink->finish();
|
||
|
|
decompressionSink->finish();
|
||
|
|
|
||
|
|
ASSERT_STREQ((*strSink.s).c_str(), inputString);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|