almet/ihatemoney
1
0
Fork 0
mirror of https://github.com/spiral-project/ihatemoney.git synced 2025-05-06 05:01:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

extract function to extract HTML link

Browse source
This commit is contained in:
Glandos 2021-07-26 00:04:37 +02:00
parent 89c2756dff
commit e25d5dbfd2
2 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
ihatemoney/tests/budget_test.py
View file

@ -8,12 +8,12 @@ from unittest.mock import MagicMock
from urllib.parse import parse_qs, urlencode, urlparse, urlunparse from urllib.parse import parse_qs, urlencode, urlparse, urlunparse
from flask import session from flask import session
from markupsafe import Markup
import pytest import pytest
from werkzeug.security import check_password_hash, generate_password_hash from werkzeug.security import check_password_hash, generate_password_hash
from ihatemoney import models from ihatemoney import models
from ihatemoney.currency_convertor import CurrencyConverter from ihatemoney.currency_convertor import CurrencyConverter
from ihatemoney.tests.common.help_functions import extract_link
from ihatemoney.tests.common.ihatemoney_testcase import IhatemoneyTestCase from ihatemoney.tests.common.ihatemoney_testcase import IhatemoneyTestCase
from ihatemoney.versioning import LoggingMode from ihatemoney.versioning import LoggingMode
@ -109,10 +109,7 @@ class BudgetTestCase(IhatemoneyTestCase):
self.login("raclette") self.login("raclette")
self.post_project("raclette") self.post_project("raclette")
response = self.client.get("/raclette/invite").data.decode("utf-8") response = self.client.get("/raclette/invite").data.decode("utf-8")
base_index = response.find("share the following link") link = extract_link(response, "share the following link")
start = response.find('href="', base_index) + 6
end = response.find('">', base_index)
link = Markup(response[start:end]).unescape()
self.client.get("/exit") self.client.get("/exit")
response = self.client.get(link) response = self.client.get(link)

11
ihatemoney/tests/common/help_functions.py
View file

@ -1,5 +1,16 @@
from markupsafe import Markup
def em_surround(string, regex_escape=False): def em_surround(string, regex_escape=False):
if regex_escape: if regex_escape:
return r'<em class="font-italic">%s<\/em>' % string return r'<em class="font-italic">%s<\/em>' % string
else: else:
return '<em class="font-italic">%s</em>' % string return '<em class="font-italic">%s</em>' % string
def extract_link(data, start_prefix):
base_index = data.find(start_prefix)
start = data.find('href="', base_index) + 6
end = data.find('">', base_index)
link = Markup(data[start:end]).unescape()
return link