adding OptionsSetterTest

2023-10-10 08:48:32 +02:00
parent 6790ddd12e
commit 4a77e073ad
2 changed files with 90 additions and 0 deletions
--- a/src/SymptomsCausedByVaccines/OptionsSetter.py
+++ b/src/SymptomsCausedByVaccines/OptionsSetter.py
@@ -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')
--- a/src/SymptomsCausedByVaccines/OptionsSetterTest.py
+++ b/src/SymptomsCausedByVaccines/OptionsSetterTest.py
@@ -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))