Tests: Add some basic public suffix tests

Author: tcl3

Tests/LibURL/CMakeLists.txt

@@ -1,6 +1,7 @@
 set(URL_TEST_SOURCES
     TestURL.cpp
     TestURLPatternConstructorStringParser.cpp
+    TestPublicSuffix.cpp
 )
 
 foreach(source IN LISTS URL_TEST_SOURCES)

Tests/LibURL/TestPublicSuffix.cpp

@@ -0,0 +1,49 @@
+/*
+ * Copyright (c) 2025, Tim Ledbetter <tim.ledbetter@ladybird.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <LibTest/TestCase.h>
+
+#include <LibURL/PublicSuffixData.h>
+
+TEST_CASE(is_public_suffix)
+{
+    auto* public_suffix_data = URL::PublicSuffixData::the();
+
+    EXPECT(public_suffix_data->is_public_suffix("com"sv));
+    EXPECT(public_suffix_data->is_public_suffix("com.br"sv));
+
+    EXPECT(!public_suffix_data->is_public_suffix(""sv));
+    EXPECT(!public_suffix_data->is_public_suffix("."sv));
+    EXPECT(!public_suffix_data->is_public_suffix(".."sv));
+    EXPECT(!public_suffix_data->is_public_suffix("/"sv));
+    EXPECT(!public_suffix_data->is_public_suffix("not-a-public-suffix.com"sv));
+    EXPECT(!public_suffix_data->is_public_suffix("com."sv));
+    EXPECT(!public_suffix_data->is_public_suffix("com/"sv));
+    EXPECT(!public_suffix_data->is_public_suffix("/com"sv));
+    EXPECT(!public_suffix_data->is_public_suffix("not-a-public-suffix"sv));
+    EXPECT(!public_suffix_data->is_public_suffix(" com"sv));
+    EXPECT(!public_suffix_data->is_public_suffix("com "sv));
+}
+
+TEST_CASE(get_public_suffix)
+{
+    auto* public_suffix_data = URL::PublicSuffixData::the();
+
+    EXPECT_EQ(public_suffix_data->get_public_suffix(""sv), OptionalNone {});
+    EXPECT_EQ(public_suffix_data->get_public_suffix("."sv), OptionalNone {});
+    EXPECT_EQ(public_suffix_data->get_public_suffix(".."sv), OptionalNone {});
+    EXPECT_EQ(public_suffix_data->get_public_suffix(" "sv), OptionalNone {});
+    EXPECT_EQ(public_suffix_data->get_public_suffix("/"sv), OptionalNone {});
+    EXPECT_EQ(public_suffix_data->get_public_suffix("not-a-public-suffix"sv), OptionalNone {});
+
+    EXPECT_EQ(public_suffix_data->get_public_suffix("com"sv), "com"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix("not-a-public-suffix.com"sv), "com"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix("com."sv), "com"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix(".com."sv), "com"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix("..com."sv), "com"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix("com.br"sv), "com.br"sv);
+    EXPECT_EQ(public_suffix_data->get_public_suffix("not-a-public-suffix.com.br"sv), "com.br"sv);
+}