Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 7580852b authored by Adam Tauber's avatar Adam Tauber
Browse files

[enh] test utils.prettify_url

parent da13c3b5

searx/tests/test_utils.py

+10 −0
Original line number Diff line number Diff line 
# -*- coding: utf-8 -*-

import mock

from searx.testing import SearxTestCase

from searx import utils
@@ -51,6 +52,15 @@ class TestUtils(SearxTestCase): 
        self.assertIsNotNone(utils.html_to_text(html))

        self.assertEqual(utils.html_to_text(html), "Test text")



    def test_prettify_url(self):

        data = (('https://searx.me/', 'https://searx.me/'),

                (u'https://searx.me/ű', u'https://searx.me/ű'),

                ('https://searx.me/' + (100 * 'a'), 'https://searx.me/[...]aaaaaaaaaaaaaaaaa'),

                (u'https://searx.me/' + (100 * u'ű'), u'https://searx.me/[...]űűűűűűűűűűűűűűűűű'))



        for test_url, expected in data:

            self.assertEqual(utils.prettify_url(test_url, max_length=32), expected)





class TestHTMLTextExtractor(SearxTestCase):

searx/utils.py

+4 −3
Original line number Diff line number Diff line
@@ -222,9 +222,10 @@ def dict_subset(d, properties): 
    return result





def prettify_url(url):

    if len(url) > 74:

        return u'{0}[...]{1}'.format(url[:35], url[-35:])

def prettify_url(url, max_length=74):

    if len(url) > max_length:

        chunk_len = max_length / 2 + 1

        return u'{0}[...]{1}'.format(url[:chunk_len], url[-chunk_len:])

    else:

        return url