starting to add select2 box for batch codes
This commit is contained in:
71
src/BatchcodeOptionsSetterTest.py
Normal file
71
src/BatchcodeOptionsSetterTest.py
Normal file
@@ -0,0 +1,71 @@
|
||||
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))
|
||||
Reference in New Issue
Block a user