Py_code

@pytest.fixture(scope='session')
def UD(request, driver):
    """
    Фикстура создает несколько УД для работы, с набором переданных услуг.
    """
    # Для использования фикстуры надо авторизоваться
    server = SERVER
    user = LOGIN
    password = PASSWORD

    session = Session()
    session.auth_cas(server=server, user=user, password=password)
    session.headers.update({'x-lpu-id': X_LPU_ID})

    def add_service(service = None):
        """
        Этот метод вызывается для добавления пациента в очередь "на удаление".
        Пациенты храняться в виде словаря, ибо удобно.
        """
        if not service:
            raise NotEnoughDataException('Не передан один из обязательных параметров! \nservice: {}'.format(service))

        patient = PatientAPI()
        document = DocumentAPI()

        patient.create_patient()
        document.edit(block='services_main', code=service)
        document.create_ud(patient.id)


    def fin():
        document.delete()
        patient.delete()

    request.addfinalizer(fin)

    return add_patient