1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 03:47:35 +00:00

Userland: Use TestSuite.h in test-compress.

This commit is contained in:
asynts 2020-08-20 18:17:19 +02:00 committed by Andreas Kling
parent d2121ab7c7
commit aa25fb8875

View file

@ -24,79 +24,51 @@
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/ */
#include <AK/TestSuite.h>
#include <LibCompress/Deflate.h> #include <LibCompress/Deflate.h>
#include <LibCompress/Zlib.h> #include <LibCompress/Zlib.h>
#include <LibCore/ArgsParser.h>
auto main(int argc, char** argv) -> int bool compare(ReadonlyBytes lhs, ReadonlyBytes rhs)
{ {
const char* mode = nullptr; if (lhs.size() != rhs.size())
const char* type = nullptr; return false;
Core::ArgsParser parser; for (size_t idx = 0; idx < lhs.size(); ++idx) {
parser.add_positional_argument(type, "Type of algorithm to apply (Only Zlib and DEFLATE is present at the moment)", "type", Core::ArgsParser::Required::No); if (lhs[idx] != rhs[idx])
parser.add_positional_argument(mode, "Mode to operate in (compress|decompress; Only decompress is valid at the moment)", "mode", Core::ArgsParser::Required::No); return false;
parser.parse(argc, argv);
if (type == nullptr) {
type = "deflate";
} }
if (mode == nullptr) { return true;
mode = "decompress"; }
}
StringView mode_sv { mode }; TEST_CASE(deflate_decompress_compressed_block)
StringView type_sv { type }; {
if (mode_sv == "decompress") { const u8 compressed[] = {
if (type_sv == "deflate") { 0x0B, 0xC9, 0xC8, 0x2C, 0x56, 0x00, 0xA2, 0x44, 0x85, 0xE2, 0xCC, 0xDC,
// Deflated bytes for the string "This is a simple text file :)" 0x82, 0x9C, 0x54, 0x85, 0x92, 0xD4, 0x8A, 0x12, 0x85, 0xB4, 0x4C, 0x20,
u8 data_bytes[] = {
0x0B, 0xC9, 0xC8, 0x2C,
0x56, 0x00, 0xA2, 0x44,
0x85, 0xE2, 0xCC, 0xDC,
0x82, 0x9C, 0x54, 0x85,
0x92, 0xD4, 0x8A, 0x12,
0x85, 0xB4, 0x4C, 0x20,
0xCB, 0x4A, 0x13, 0x00 0xCB, 0x4A, 0x13, 0x00
}; };
auto deflated = Compress::DeflateStream::decompress_all({ data_bytes, 4 * 7 }); const u8 uncompressed[] = "This is a simple text file :)";
auto decompressed = String((const char*)deflated.data(), deflated.size());
if (decompressed.equals_ignoring_case("This is a simple text file :)")) { const auto decompressed = Compress::DeflateStream::decompress_all({ compressed, sizeof(compressed) });
printf("Test PASSED"); EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.span()));
return 0; }
} else {
printf("Test FAILED");
return 1;
}
}
if (type_sv == "zlib") { TEST_CASE(zlib_simple_decompress)
// zlib bytes for the string "This is a simple text file :)" {
u8 data_bytes[] = { const u8 compressed[] = {
0x78, 0x01, 0x01, 0x1D, 0x00, 0xE2, 0xFF, 0x54, 0x78, 0x01, 0x01, 0x1D, 0x00, 0xE2, 0xFF, 0x54, 0x68, 0x69, 0x73, 0x20,
0x68, 0x69, 0x73, 0x20, 0x69, 0x73, 0x20, 0x61, 0x69, 0x73, 0x20, 0x61, 0x20, 0x73, 0x69, 0x6D, 0x70, 0x6C, 0x65, 0x20,
0x20, 0x73, 0x69, 0x6D, 0x70, 0x6C, 0x65, 0x20, 0x74, 0x65, 0x78, 0x74, 0x20, 0x66, 0x69, 0x6C, 0x65, 0x20, 0x3A, 0x29,
0x74, 0x65, 0x78, 0x74, 0x20, 0x66, 0x69, 0x6C, 0x99, 0x5E, 0x09, 0xE8
0x65, 0x20, 0x3A, 0x29, 0x99, 0x5E, 0x09, 0xE8
}; };
auto deflater = Compress::Zlib({ data_bytes, 8 * 5 }); const u8 uncompressed[] = "This is a simple text file :)";
auto deflated = deflater.decompress();
auto decompressed = String((const char*)deflated.data(), deflated.size());
if (decompressed.equals_ignoring_case("This is a simple text file :)")) { const auto decompressed = Compress::Zlib { { compressed, sizeof(compressed) } }.decompress();
printf("Test PASSED"); EXPECT(compare({ uncompressed, sizeof(uncompressed) - 1 }, decompressed.span()));
return 0;
} else {
printf("Test FAILED");
return 1;
}
}
}
printf("Unknown arguments passed to test!");
return 1;
} }
TEST_MAIN(Compress)