diff --git a/py/selenium/webdriver/common/bidi/log.py b/py/selenium/webdriver/common/bidi/log.py index 52097d536a902..036ebd24a031f 100644 --- a/py/selenium/webdriver/common/bidi/log.py +++ b/py/selenium/webdriver/common/bidi/log.py @@ -68,3 +68,12 @@ def from_json(cls, json): stacktrace=json["stackTrace"], type_=json["type"], ) + + +class LogLevel: + """Represents log level.""" + + DEBUG = "debug" + INFO = "info" + WARN = "warn" + ERROR = "error" diff --git a/py/test/selenium/webdriver/common/bidi_script_tests.py b/py/test/selenium/webdriver/common/bidi_script_tests.py index f974d60770a88..9030227ed196f 100644 --- a/py/test/selenium/webdriver/common/bidi_script_tests.py +++ b/py/test/selenium/webdriver/common/bidi_script_tests.py @@ -15,6 +15,7 @@ # specific language governing permissions and limitations # under the License. +from selenium.webdriver.common.bidi.log import LogLevel from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait @@ -31,7 +32,7 @@ def test_logs_console_messages(driver, pages): WebDriverWait(driver, 5).until(lambda _: log_entries) log_entry = log_entries[0] - assert log_entry.level == "info" + assert log_entry.level == LogLevel.INFO assert log_entry.method == "log" assert log_entry.text == "Hello, world!" assert log_entry.type_ == "console" @@ -55,7 +56,7 @@ def log_error(entry): assert len(log_entries) == 1 log_entry = log_entries[0] - assert log_entry.level == "error" + assert log_entry.level == LogLevel.ERROR assert log_entry.method == "error" assert log_entry.text == "I am console error" assert log_entry.type_ == "console" @@ -105,7 +106,7 @@ def test_javascript_error_messages(driver, pages): log_entry = log_entries[0] assert log_entry.text == "Error: Not working" - assert log_entry.level == "error" + assert log_entry.level == LogLevel.ERROR assert log_entry.type_ == "javascript"