diff --git a/tests/__init__.py b/tests/__init__.py index c63b04e..c835b2d 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -428,6 +428,18 @@ class SimplePagureTest(unittest.TestCase): return output.get_data(as_text=True).split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] + def get_wtforms_version(self): + """Returns the wtforms version as a tuple.""" + import wtforms + wtforms_v = wtforms.__version__.split('.') + for idx, val in enumerate(wtforms_v): + try: + val = int(val) + except ValueError: + pass + wtforms_v[idx] = val + return tuple(wtforms_v) + def assertURLEqual(self, url_1, url_2): url_parsed_1 = list(urlparse(url_1)) url_parsed_1[4] = parse_qs(url_parsed_1[4]) diff --git a/tests/test_pagure_flask_ui_app.py b/tests/test_pagure_flask_ui_app.py index 9557498..c346bae 100644 --- a/tests/test_pagure_flask_ui_app.py +++ b/tests/test_pagure_flask_ui_app.py @@ -637,9 +637,15 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', + output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -687,9 +693,15 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -943,9 +960,15 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', + output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -961,12 +984,18 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', + output_text) + else: + self.assertIn( + '', output_text) self.assertIn( - '', output_text) - self.assertIn( - '\n You must always have at least one email', - output_text) + '\n You must always have at ' + 'least one email', output_text) user.username = 'pingou' with tests.user_set(self.app.application, user): @@ -976,9 +1005,15 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', + output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -1055,9 +1090,14 @@ class PagureFlaskApptests(tests.Modeltests): self.assertEqual(output.status_code, 200) output_text = output.get_data(as_text=True) self.assertIn("Add new email", output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) user.username = 'pingou' with tests.user_set(self.app.application, user): @@ -1065,9 +1105,14 @@ class PagureFlaskApptests(tests.Modeltests): self.assertEqual(output.status_code, 200) output_text = output.get_data(as_text=True) self.assertIn("Add new email", output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -1181,9 +1226,15 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', + output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -1277,9 +1328,14 @@ class PagureFlaskApptests(tests.Modeltests): self.assertIn( '
\n Basic Information\n' '
', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] diff --git a/tests/test_pagure_flask_ui_plugins_mail.py b/tests/test_pagure_flask_ui_plugins_mail.py index 9cd0ff3..e24b5c6 100644 --- a/tests/test_pagure_flask_ui_plugins_mail.py +++ b/tests/test_pagure_flask_ui_plugins_mail.py @@ -109,10 +109,16 @@ class PagureFlaskPluginMailtests(tests.SimplePagureTest): self.assertIn('

Mail settings

', output_text) self.assertNotIn( '\n Hook activated', output_text) - self.assertIn( - '\n' - 'This field is required.', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '\n' + 'This field is required.', output_text) + else: + self.assertIn( + '\n' + 'This field is required.', output_text) self.assertIn( '', output_text) diff --git a/tests/test_pagure_flask_ui_plugins_noff.py b/tests/test_pagure_flask_ui_plugins_noff.py index 8924506..1233c4e 100644 --- a/tests/test_pagure_flask_ui_plugins_noff.py +++ b/tests/test_pagure_flask_ui_plugins_noff.py @@ -45,9 +45,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertIn( '

Block non fast-forward pushes settings

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertTrue( '' in output_text) @@ -67,9 +72,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertIn( '

Block non fast-forward pushes settings

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertTrue( '' in output_text) @@ -99,9 +109,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertIn( '

Block non fast-forward pushes settings

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertTrue( '' in output_text) @@ -126,9 +141,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertNotIn( '\n Hook activated', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertTrue( '' in output_text) @@ -165,9 +185,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertIn( '

Block non fast-forward pushes settings

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertIn( '', output_text) @@ -199,9 +224,14 @@ class PagureFlaskPluginNoFFtests(tests.SimplePagureTest): self.assertIn( '

Block non fast-forward pushes settings

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertIn( '', output_text) diff --git a/tests/test_pagure_flask_ui_plugins_pagure_ci.py b/tests/test_pagure_flask_ui_plugins_pagure_ci.py index 344a986..0dfd04c 100644 --- a/tests/test_pagure_flask_ui_plugins_pagure_ci.py +++ b/tests/test_pagure_flask_ui_plugins_pagure_ci.py @@ -179,14 +179,29 @@ class PagureFlaskPluginPagureCItests(tests.SimplePagureTest): self.assertIn('

Pagure CI settings

', output_text) self.assertFalse( '\n Hook activated' in output_text) - self.assertIn( - '' - '\nThis field is required.', - output_text) - self.assertIn( - '' - '\nThis field is required.', - output_text) + + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '' + '\nThis field is required.', + output_text) + self.assertIn( + '' + '\nThis field is required.', + output_text) + else: + self.assertIn( + '' + '\nThis field is required.', + output_text) + self.assertIn( + '' + '\nThis field is required.', + output_text) self.assertIn( '', output_text) diff --git a/tests/test_pagure_flask_ui_repo.py b/tests/test_pagure_flask_ui_repo.py index ad42375..82ba0b9 100644 --- a/tests/test_pagure_flask_ui_repo.py +++ b/tests/test_pagure_flask_ui_repo.py @@ -4505,9 +4505,14 @@ index 0000000..fb7093d self.assertIn( 'Settings - test - Pagure', output_text) self.assertIn('

Settings for test

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) csrf_token = output_text.split( 'name="csrf_token" type="hidden" value="')[1].split('">')[0] @@ -4542,11 +4547,19 @@ index 0000000..fb7093d self.assertIn( 'Settings - test - Pagure', output_text) self.assertIn('

Settings for test

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertIn( '\n Default branch updated ' 'to feature', output_text) @@ -4564,11 +4577,19 @@ index 0000000..fb7093d self.assertIn( 'Settings - test - Pagure', output_text) self.assertIn('

Settings for test

', output_text) - self.assertIn( - '', output_text) + if self.get_wtforms_version() >= (2, 2): + self.assertIn( + '', output_text) + else: + self.assertIn( + '', output_text) self.assertIn( '\n Default branch updated ' 'to master', output_text)