From 22b8d29baaa719cd870cac0e90ba65ebe0ebf7a1 Mon Sep 17 00:00:00 2001
From: perrette <mahe.perrette@pik-potsdam.de>
Date: Tue, 27 Jun 2023 13:08:13 +0200
Subject: [PATCH] fix broken test in test_list.py (strip ansi links)

---
 papers/__main__.py | 15 +--------------
 papers/utils.py    | 28 ++++++++++++++++++++++++++++
 tests/test_list.py | 10 +++++-----
 3 files changed, 34 insertions(+), 19 deletions(-)

diff --git a/papers/__main__.py b/papers/__main__.py
index 3653e51..a1b71c7 100644
--- a/papers/__main__.py
+++ b/papers/__main__.py
@@ -18,7 +18,7 @@
 from papers.config import bcolors, Config, search_config, CONFIG_FILE, CONFIG_FILE_LOCAL, DATA_DIR, CONFIG_FILE_LEGACY, BACKUP_DIR
 from papers.duplicate import list_duplicates, list_uniques, edit_entries
 from papers.bib import Biblio, FUZZY_RATIO, DEFAULT_SIMILARITY, entry_filecheck, backupfile as backupfile_func, isvalidkey
-from papers.utils import move, checksum
+from papers.utils import move, checksum, ansi_link as link
 from papers import __version__
 
 
@@ -806,19 +806,6 @@ def parse_keywords(e):
         print(format_entries(entries))
 
 
-def link(uri, label=None):
-    """https://stackoverflow.com/a/71309268/2192272
-    """
-    if label is None:
-        label = uri
-    parameters = ''
-
-    # OSC 8 ; params ; URI ST <name> OSC 8 ;; ST
-    escape_mask = '\033]8;{};{}\033\\{}\033]8;;\033\\'
-
-    return escape_mask.format(parameters, uri, label)
-
-
 def statuscmd(parser, o, config):
     print(config.status(check_files=not o.no_check_files, verbose=o.verbose))
     
diff --git a/papers/utils.py b/papers/utils.py
index de6fc27..6c5a389 100644
--- a/papers/utils.py
+++ b/papers/utils.py
@@ -1,4 +1,5 @@
 import os
+import re
 import shutil
 import hashlib
 
@@ -27,6 +28,33 @@ def strip_colors(s):
     return s
 
 
+def ansi_link(uri, label=None):
+    """https://stackoverflow.com/a/71309268/2192272
+    """
+    if label is None:
+        label = uri
+    parameters = ''
+
+    # OSC 8 ; params ; URI ST <name> OSC 8 ;; ST
+    escape_mask = '\033]8;{};{}\033\\{}\033]8;;\033\\'
+
+    return escape_mask.format(parameters, uri, label)
+
+
+ANSI_LINK_RE = re.compile(r'(?P<ansi_sequence>\033]8;(?P<parameter>.*?);(?P<uri>.*?)\033\\(?P<label>.*?)\033]8;;\033\\)')
+
+def strip_ansi_link(s):
+    for m in ANSI_LINK_RE.findall(s):
+        s = s.replace(m[0], m[3])
+    return s
+
+
+def strip_all(s):
+    s = strip_colors(s)
+    s = strip_ansi_link(s)
+    return s
+
+
 def check_filesdir(folder):
     folder_size = 0
     file_count = 0
diff --git a/tests/test_list.py b/tests/test_list.py
index dfe9a3e..74c978c 100644
--- a/tests/test_list.py
+++ b/tests/test_list.py
@@ -1,6 +1,6 @@
 import bibtexparser
 from tests.common import LocalInstallTest, Biblio
-from papers.utils import strip_colors
+from papers.utils import strip_all
 
 bibtex = """@article{Perrette_2011,
  author = {M. Perrette and A. Yool and G. D. Quartly and E. E. Popova},
@@ -28,13 +28,13 @@ class FormattingTest(ListTest):
 
     def test_format(self):
         out = self.papers(f'list -l', sp_cmd='check_output')
-        self.assertEqual(strip_colors(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean)")
+        self.assertEqual(strip_all(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean)")
 
         out = self.papers(f'list --key-only', sp_cmd='check_output')
         self.assertEqual(out, "Perrette_2011")
 
         out = self.papers(f'list -f month doi', sp_cmd='check_output')
-        self.assertEqual(strip_colors(out), "Perrette_2011: feb 10.5194/bg-8-515-2011")
+        self.assertEqual(strip_all(out), "Perrette_2011: feb 10.5194/bg-8-515-2011")
 
 
 class SearchTest(ListTest):
@@ -151,7 +151,7 @@ def test_add_tag(self):
         self.assertEqual(out, "")
 
         self.papers(f'list --author perrette --add-tag newtag -1')
-        # self.assertEqual(strip_colors(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean | newtag)")
+        # self.assertEqual(strip_all(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean | newtag)")
 
         out = self.papers(f'list --tag newtag -1', sp_cmd='check_output')
-        self.assertEqual(strip_colors(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean | newtag)")
\ No newline at end of file
+        self.assertEqual(strip_all(out), "Perrette_2011: Near-ubiquity of ice-edge blooms in the Arctic (doi:10.5194/bg-8-515-2011, files:2, kiwi | ocean | newtag)")
\ No newline at end of file