diff --git a/seleniumbase/fixtures/base_case.py b/seleniumbase/fixtures/base_case.py index 362c86dd..50b35bf6 100755 --- a/seleniumbase/fixtures/base_case.py +++ b/seleniumbase/fixtures/base_case.py @@ -26,7 +26,6 @@ import json import logging import math import os -import pytest import re import sys import time @@ -3185,13 +3184,21 @@ class BaseCase(unittest.TestCase): def __add_pytest_html_extra(self): try: - pytest_html = pytest.config.pluginmanager.getplugin('html') - if self.with_selenium and pytest_html: - driver = self.driver - extra_url = pytest_html.extras.url(driver.current_url) - screenshot = driver.get_screenshot_as_base64() - extra_image = pytest_html.extras.image(screenshot, - name='Screenshot') + if self.with_selenium: + if not self.__last_page_screenshot: + self.__set_last_page_screenshot() + extra_url = {} + extra_url['name'] = 'URL' + extra_url['format'] = 'url' + extra_url['content'] = self.get_current_url() + extra_url['mime_type'] = None + extra_url['extension'] = None + extra_image = {} + extra_image['name'] = 'Screenshot' + extra_image['format'] = 'image' + extra_image['content'] = self.__last_page_screenshot + extra_image['mime_type'] = 'image/png' + extra_image['extension'] = 'png' self._html_report_extra.append(extra_url) self._html_report_extra.append(extra_image) except Exception: