From 14519508e9e02760ddd1a7b2c91ad79968329198 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Fri, 30 Aug 2024 05:55:13 +0300
Subject: Fix URL regex

---
 src/test/java/com/juick/MessageTest.java | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/test/java')

diff --git a/src/test/java/com/juick/MessageTest.java b/src/test/java/com/juick/MessageTest.java
index 14d6175a..6ce01f4a 100644
--- a/src/test/java/com/juick/MessageTest.java
+++ b/src/test/java/com/juick/MessageTest.java
@@ -213,8 +213,8 @@ public class MessageTest {
         assertThat(MessageUtils.formatMessage(msg), is("<blockquote>quote</blockquote>message"));
         String brokenComment = "<!-- read next";
         assertThat(MessageUtils.formatMessage(brokenComment), is("&lt;!-- read next"));
-        String url = "[ya](http://ya.ru)";
-        assertThat(MessageUtils.formatMessage(url), is("<a href=\"http://ya.ru\" rel=\"nofollow\">ya</a>"));
+        String url = "test [ya](https://juick.com/a/1)? hh";
+        assertThat(MessageUtils.formatMessage(url), is("test <a href=\"https://juick.com/a/1\" rel=\"nofollow\">ya</a>? hh"));
         String complexMessage = "У футболистов нет мозгов. Что в России, что в Беларуси:\n" +
                 "\n" +
                 ">Отец футболиста Лухвича, объехавшего пробку по тротуару: «Сына задержали, Infiniti арестовали» https://auto.onliner.by/2019/01/23/probka-9\n" +
@@ -238,4 +238,12 @@ public class MessageTest {
         testMessage.setTags(MessageUtils.parseTags("NSFW test"));
         assertThat(MessageUtils.isSensitive(testMessage), is(true));
     }
+    @Test
+    public void stripHashesTest() {
+        var safeMessage = "And that is a [odd](https://juick.com/a/1)? aaapaa";
+        var nonsafeMessage = "[Here is my link](https://juick.com?hash=12345)";
+        var filteredMessage = "[Here is my link](https://juick.com)";
+        assertThat(MessageUtils.stripNonSafeUrls(safeMessage), is(safeMessage));
+        assertThat(MessageUtils.stripNonSafeUrls(nonsafeMessage), is(filteredMessage));
+    }
 }
-- 
cgit v1.2.3