libpythonpro/Libpythonpro/tests/test_spam/conftest.py at 9573ed3b536a4b7f8b84aba8200138cc1f1ffcdc · pythonprobr/libpythonpro

Repository navigation

from time import sleep

import pytest

class Sessao:

contador = 0

usuarios = []

def salvar(self, usuario):

Sessao.contador += 1

usuario.id = Sessao.contador

self.usuarios.append(usuario)

def listar(self):

return self.usuarios

def roll_back(self):

self.usuarios.clear()

def fechar(self):

pass

class Conexao:

def __init__(self):

sleep(1)

def gerar_sessao(self):

return Sessao()

def fechar(self):

pass

@pytest.fixture(scope='session')

def conexao():

# Setup

conexao_obj = Conexao()

yield conexao_obj

# Tear Down

conexao_obj.fechar()

@pytest.fixture

def sessao(conexao):

sessao_obj = conexao.gerar_sessao()

yield sessao_obj

sessao_obj.roll_back()

sessao_obj.fechar()