Триггеры

from dsl import *
from trigger import BaseTrigger

class AuthTrigger(BaseTrigger):
    """
    Авторизация на сайте при помощи cookies
    """
    def __init__(self):
        self._cookies = {
            "name": "user", "value": "234d36e1de92e831d66bd788e1974bc0"
        }
    
    def setup_test(self, browser, test_id, test_scope, test_args):
        """
        Перед выполнением теста переходим на страницу BASE_URL, очищаем cookies и       
        устанавливаем cookies определённые в конструкторе
        """
        browser.get_url(CONFIG.BASE_URL)
        browser.del_cookies()
        browser.set_cookies(self._cookies)

# "подключаем" триггер
CONFIG.TRIGGER = AuthTrigger

# Если URL страницы после выполнения команды поменялся, то проверяем элемент с id RightSide_Advertisement, если он есть, то скрываем его.

class QaDemoTrigger(BaseTrigger):
    def setup_test(self, browser, test_id, test_scope, test_args):
        """
        в начале теста запоминаем начальный URL
        """
        self._start_url = browser.url
    
    def setup_command(self, browser, command, test_scope):
        """
        сохраняем текущий URL, если он равен начальному, то присваиваем пустую строку
        """
        self._url = browser.url
        if self._url == self._start_url:
            self._url = ""
    
    def teardown_command(self, browser, command, test_scope):
        """
        Если URL после выполнения команды поменялся, то проверяем и скрываем #RightSide_Advertisement
        """
        if not self._url == browser.url:
            return (
                # если #RightSide_Advertisement виден, то скрываем его
                DATA_ATTR("#RightSide_Advertisement", "style", "display:none", execute=~(NOT("#RightSide_Advertisement", no_displayed=True)))
                # ждём исчезновения элемента #RightSide_Advertisement
                , NOT("#RightSide_Advertisement", no_displayed=True)
            )

CONFIG.TRIGGER = QaDemoTrigger