Userland: Use TestSuite.h in test-compress.

1 files changed, 34 insertions, 62 deletions

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