refactoring

2022-11-22 12:40:26 +01:00
parent 134a133da1
commit e87fe0c8ba
21 changed files with 692 additions and 731 deletions
--- a/src/CountryOptionsSetterTest.py
+++ b/src/CountryOptionsSetterTest.py
@@ -0,0 +1,73 @@
+import unittest
+from CountryOptionsSetter import CountryOptionsSetter
+
+class CountryOptionsSetterTest(unittest.TestCase):
+
+    def test_setCountryOptions(self):
+        # Given
+        countryOptionsSetter = CountryOptionsSetter()
+
+        # When
+        htmlActual = countryOptionsSetter.setCountryOptions(
+            html='''
+            <html>
+              <body>
+                <p>Test<p/>
+                <select id="countrySelect" name="country">
+                  <option value="Global" selected>Global</option>
+                  <option value="Afghanistan">Afghanistan</option>
+                  <option value="Albania">Albania</option>
+                  <option value="Algeria">Algeria</option>
+                </select>
+              </body>
+            </html>
+            ''',
+            options=[
+                '<option value="Global" selected>Global</option>',
+                '<option value="Azerbaijan">Azerbaijan</option>',
+                '<option value="Bahrain">Bahrain</option>'])
+
+        # Then
+        assertEqualHTML(
+            htmlActual,
+            '''
+            <html>
+              <body>
+                <p>Test<p/>
+                <select id="countrySelect" name="country">
+                  <option value="Global" selected>Global</option>
+                  <option value="Azerbaijan">Azerbaijan</option>
+                  <option value="Bahrain">Bahrain</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):
+            display(line)
+        display(p[0], ' != ', p[1])
+        raise Exception('Not equal %s %s' % (file1, file2))