Alterado código de produção

renzon · renzon · commit 8fe60eb100fb · 2018-05-19T20:15:23.000-03:00
diff --git a/libpythonpro/spam/enviador_de_email.py b/libpythonpro/spam/enviador_de_email.py
@@ -1,7 +1,11 @@
 class Enviador:
+    def __init__(self):
+        self.qtd_email_enviados = 0
+
     def enviar(self, remetente, destinatario, assunto, corpo):
         if '@' not in remetente:
             raise EmailInvalido(f'Email de remetente inválido: {remetente}')
+        self.qtd_email_enviados += 1
         return remetente
 
 
diff --git a/libpythonpro/spam/main.py b/libpythonpro/spam/main.py
@@ -4,4 +4,10 @@ def __init__(self, sessao, enviador):
         self.enviador = enviador
 
     def enviar_emails(self, remetente, assunto, corpo):
-        pass
+        for usuario in self.sessao.listar():
+            self.enviador.enviar(
+                remetente,
+                usuario.email,
+                assunto,
+                corpo
+            )
diff --git a/libpythonpro/spam/modelos.py b/libpythonpro/spam/modelos.py
@@ -1,4 +1,5 @@
 class Usuario:
-    def __init__(self, nome):
+    def __init__(self, nome, email):
+        self.email = email
         self.nome = nome
         self.id = None
diff --git a/libpythonpro/tests/test_spam/test_envio_para_base_de_usuarios.py b/libpythonpro/tests/test_spam/test_envio_para_base_de_usuarios.py
@@ -1,11 +1,30 @@
+import pytest
+
 from libpythonpro.spam.enviador_de_email import Enviador
 from libpythonpro.spam.main import EnviadorDeSpam
+from libpythonpro.spam.modelos import Usuario
 
 
-def test_envio_de_spam(sessao):
-    enviador_de_spam = EnviadorDeSpam(sessao, Enviador())
+@pytest.mark.parametrize(
+    'usuarios',
+    [
+        [
+            Usuario(nome='Renzo', email='renzo@python.pro.br'),
+            Usuario(nome='Luciano', email='renzo@python.pro.br')
+        ],
+        [
+            Usuario(nome='Renzo', email='renzo@python.pro.br')
+        ]
+    ]
+)
+def test_qde_de_spam(sessao, usuarios):
+    for usuario in usuarios:
+        sessao.salvar(usuario)
+    enviador = Enviador()
+    enviador_de_spam = EnviadorDeSpam(sessao, enviador)
     enviador_de_spam.enviar_emails(
         'renzo@python.pro.br',
         'Curso Python Pro',
         'Confira os módulos fantásticos'
     )
+    assert len(usuarios) == enviador.qtd_email_enviados
diff --git a/libpythonpro/tests/test_spam/test_usuarios.py b/libpythonpro/tests/test_spam/test_usuarios.py
@@ -2,13 +2,16 @@
 
 
 def test_salvar_usuario(sessao):
-    usuario = Usuario(nome='Renzo')
+    usuario = Usuario(nome='Renzo', email='renzo@python.pro.br')
     sessao.salvar(usuario)
     assert isinstance(usuario.id, int)
 
 
 def test_listar_usuarios(sessao):
-    usuarios = [Usuario(nome='Renzo'), Usuario(nome='Luciano')]
+    usuarios = [
+        Usuario(nome='Renzo', email='renzo@python.pro.br'),
+        Usuario(nome='Luciano', email='renzo@python.pro.br')
+    ]
     for usuario in usuarios:
         sessao.salvar(usuario)
     assert usuarios == sessao.listar()
