francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding OptionsSetterTest

Browse Source
This commit is contained in:
frankknoll
2023-10-10 08:48:32 +02:00
parent 6790ddd12e
commit 4a77e073ad
2 changed files with 90 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/SymptomsCausedByVaccines/OptionsSetter.py Normal file
View File

@@ -0,0 +1,21 @@
from bs4 import BeautifulSoup
class OptionsSetter:
def setOptions(self, html, options):
soup = self._setBatchcodeOptions(self._parse(html), self._parseOptions(options))
return str(soup)
def _setBatchcodeOptions(self, soup, options):
batchcodeSelect = soup.find(id = "vaccineSelect")
batchcodeSelect.clear()
for option in options:
batchcodeSelect.append(option)
return soup
def _parseOptions(self, options):
return [self._parse(option).option for option in options]
def _parse(self, html):
return BeautifulSoup(html, 'lxml')

69
src/SymptomsCausedByVaccines/OptionsSetterTest.py Normal file
View File

@@ -0,0 +1,69 @@
import unittest
from SymptomsCausedByVaccines.OptionsSetter import OptionsSetter
class OptionsSetterTest(unittest.TestCase):
def test_setOptions(self):
# Given
optionsSetter = OptionsSetter()
# When
htmlActual = optionsSetter.setOptions(
html='''
<html>
<body>
<p>Test<p/>
<select id="vaccineSelect" name="vaccine">
<option value="old1">old1</option>
</select>
</body>
</html>
''',
options=[
'<option value="6VAX-F">6VAX-F</option>',
'<option value="ADEN">ADEN</option>'])
# Then
assertEqualHTML(
htmlActual,
'''
<html>
<body>
<p>Test<p/>
<select id="vaccineSelect" name="vaccine">
<option value="6VAX-F">6VAX-F</option>
<option value="ADEN">ADEN</option>
</select>
</body>
</html>
''')
# FK-TODO: DRY with BatchcodeOptionsSetterTest.assertEqualHTML()
# adapted from https://stackoverflow.com/questions/8006909/pretty-print-assertequal-for-html-strings
def assertEqualHTML(string1, string2, file1='', file2=''):
u'''
Compare two unicode strings containing HTML.
A human friendly diff goes to logging.error() if they
are not equal, and an exception gets raised.
'''
from bs4 import BeautifulSoup as bs
import difflib
def short(mystr):
max = 20
if len(mystr) > max:
return mystr[:max]
return mystr
p = []
for mystr, file in [(string1, file1), (string2, file2)]:
if not isinstance(mystr, str):
raise Exception(u'string ist not unicode: %r %s' %
(short(mystr), file))
soup = bs(mystr, 'lxml')
pretty = soup.prettify()
p.append(pretty)
if p[0] != p[1]:
for line in difflib.unified_diff(p[0].splitlines(), p[1].splitlines(), fromfile=file1, tofile=file2):
print(line)
print(p[0], ' != ', p[1])
raise Exception('Not equal %s %s' % (file1, file2))