Implementada versão Inicial de Conexao, Sessao e Usuario

renzon · renzon · commit f7c587bc48d1 · 2018-05-16T17:11:10.000-03:00
diff --git a/libpythonpro/spam/db.py b/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
diff --git a/libpythonpro/spam/modelos.py b/libpythonpro/spam/modelos.py
@@ -0,0 +1,4 @@
+class Usuario:
+    def __init__(self, nome):
+        self.nome = nome
+        self.id = None
diff --git a/libpythonpro/tests/test_spam/test_usuarios.py b/libpythonpro/tests/test_spam/test_usuarios.py
@@ -1,3 +1,7 @@
+from libpythonpro.spam.db import Conexao
+from libpythonpro.spam.modelos import Usuario
+
+
 def test_salvar_usuario():
     conexao = Conexao()
     sessao = conexao.gerar_sessao()
@@ -15,7 +19,7 @@ def test_listar_usuarios():
     usuarios = [Usuario(nome='Renzo'), Usuario(nome='Luciano')]
     for usuario in usuarios:
         sessao.salvar(usuario)
-    assert usuario == sessao.listar()
+    assert usuarios == sessao.listar()
     sessao.roll_back()
     sessao.fechar()
     conexao.fechar()
