72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import unittest
|
|
from BatchcodeOptionsSetter import BatchcodeOptionsSetter
|
|
|
|
class BatchcodeOptionsSetterTest(unittest.TestCase):
|
|
|
|
def test_setBatchcodeOptions(self):
|
|
# Given
|
|
batchcodeOptionsSetter = BatchcodeOptionsSetter()
|
|
|
|
# When
|
|
htmlActual = batchcodeOptionsSetter.setBatchcodeOptions(
|
|
html='''
|
|
<html>
|
|
<body>
|
|
<p>Test<p/>
|
|
<select id="batchCodeSelect" name="batchCode">
|
|
<option value="old1">old1</option>
|
|
<option value="old2">old2</option>
|
|
</select>
|
|
</body>
|
|
</html>
|
|
''',
|
|
options=[
|
|
'<option value="FE6208">FE6208</option>',
|
|
'<option value="039K20A">039K20A</option>',
|
|
'<option value="FF3318">FF3318</option>'])
|
|
|
|
# Then
|
|
assertEqualHTML(
|
|
htmlActual,
|
|
'''
|
|
<html>
|
|
<body>
|
|
<p>Test<p/>
|
|
<select id="batchCodeSelect" name="batchCode">
|
|
<option value="FE6208">FE6208</option>
|
|
<option value="039K20A">039K20A</option>
|
|
<option value="FF3318">FF3318</option>
|
|
</select>
|
|
</body>
|
|
</html>
|
|
''')
|
|
|
|
# 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))
|