Aula de implementação de conexão, Sessão e usuário

Libpythonpro/spam/db.py

@@ -0,0 +1,25 @@
+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):
+        pass
+
+    def fechar(self):
+        pass
+
+
+class Conexao:
+    def gerar_sessao(self):
+        return Sessao()
+
+    def fechar(self):
+        pass

Libpythonpro/spam/modelos.py

@@ -0,0 +1,4 @@
+class Usuario:
+    def __init__(self, nome):
+        self.nome = nome
+        self.id = None

Libpythonpro/spam/test_spam/enviador_de_email.py

@@ -0,0 +1,9 @@
+class Enviador:
+    def enviar(self, remetente, destinatário, assunto, corpo):
+        if '@' not in remetente:
+            raise EmailInvalido(f'Email de Remetente Inválido:{remetente}')
+        return remetente
+
+
+class EmailInvalido(Exception):
+    pass

Libpythonpro/spam/test_spam/testes_usuarios.py

@@ -0,0 +1,25 @@
+from Libpythonpro.spam.db import Conexao
+from Libpythonpro.spam.modelos import Usuario
+
+
+def test_salvar_usuario():
+    conexao = Conexao()
+    sessao = conexao.gerar_sessao()
+    usuario= Usuario(nome='Carlos')
+    sessao.salvar(usuario)
+    assert isinstance(usuario.id, int)
+    sessao.roll_back()
+    sessao.fechar()
+    conexao.fechar()
+
+
+def test_listar_usuario():
+    conexao = Conexao()
+    sessao = conexao.gerar_sessao()
+    usuarios = [Usuario(nome='Carlos'), Usuario(nome='Givani')]
+    for usuario in usuarios:
+        sessao.salvar(usuario)
+    assert usuarios == sessao.listar()
+    sessao.roll_back()
+    sessao.fechar()
+    conexao.fechar()

Libpythonpro/tests/test_spam/test_enviador_de_email.py

@@ -0,0 +1,38 @@
+import pytest
+
+from Libpythonpro.spam.test_spam.enviador_de_email import Enviador, EmailInvalido
+
+
+def test_criar_enviador_de_email():
+    enviador= Enviador()
+    assert enviador is not None
+
+@pytest.mark.parametrize(
+    'destinatario',
+    ['[email protected]','[email protected]']
+)
+def test_remetente(destinatario):
+    enviador = Enviador()
+    destinatario
+    resultado = enviador.enviar(
+        destinatario,
+        '[email protected]',
+        'Cursos Python Pro',
+        'Primeira Turma Guido Von Rossum aberta.')
+    assert destinatario in resultado
+
+
+@pytest.mark.parametrize(
+    'remetente',
+    ['','renzo']
+)
+def test_remetente_invalido(remetente):
+    enviador = Enviador()
+    with pytest.raises(EmailInvalido):
+        enviador.enviar(
+            remetente,
+            '[email protected]',
+            'Cursos Python Pro',
+            'Primeira Turma Guido Von Rossum aberta.'
+        )
+