Fix report tests for pytest-html >= 2.0.0 (#240)

Co-authored-by: PhilippSelenium <[email protected]>

Author: PhilippSelenium

testing/test_report.py

@@ -5,16 +5,24 @@
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 import re
+from distutils.version import LooseVersion
 
 import pytest
+from pytest_html import __version__ as PYTEST_HTML_VERSION
 
 pytestmark = pytest.mark.nondestructive
 
 URL_LINK = '<a class="url" href="{0}/" target="_blank">URL</a>'
-SCREENSHOT_LINK_REGEX = (
-    '<a class="image" href=".*" target="_blank">Screenshot</a>'
-)  # noqa
-SCREENSHOT_REGEX = '<div class="image"><a href=".*"><img src=".*"/></a></div>'
+
+if LooseVersion(PYTEST_HTML_VERSION) < LooseVersion("2.0.0"):
+    SCREENSHOT_LINK_REGEX = '<a class="image" href=".*" target="_blank">Screenshot</a>'
+    SCREENSHOT_REGEX = '<div class="image"><a href=".*"><img src=".*"/></a></div>'
+else:
+    SCREENSHOT_LINK_REGEX = '<a class="image" href=".*" target="_blank"></a>'
+    SCREENSHOT_REGEX = (
+        '<div class="image"><a class="image" href=".*" target="_blank"></a></div>'
+    )
+
 LOGS_REGEX = '<a class="text" href=".*" target="_blank">.* Log</a>'
 HTML_REGEX = '<a class="text" href=".*" target="_blank">HTML</a>'