Atualizando o código fonte das aulas 4-12 com a fixture de pausar o t…

…empo

related #251

codigo_das_aulas/04/tests/conftest.py

@@ -1,6 +1,9 @@
+from contextlib import contextmanager
+from datetime import datetime
+
 import pytest
 from fastapi.testclient import TestClient
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, event
 from sqlalchemy.orm import Session
 
 from fast_zero.app import app
@@ -21,3 +24,24 @@ def session():
         yield session
 
     table_registry.metadata.drop_all(engine)
+
+
+@contextmanager
+def _mock_db_time(*, model, time=datetime(2024, 1, 1)):
+
+    def fake_time_handler(mapper, connection, target):
+        if hasattr(target, 'created_at'):
+            target.created_at = time
+        if hasattr(target, 'updated_at'):
+            target.updated_at = time
+
+    event.listen(model, 'before_insert', fake_time_handler)
+
+    yield time
+
+    event.remove(model, 'before_insert', fake_time_handler)
+
+
+@pytest.fixture
+def mock_db_time():
+    return _mock_db_time

codigo_das_aulas/04/tests/test_db.py

@@ -1,43 +1,25 @@
-from contextlib import contextmanager
 from dataclasses import asdict
-from datetime import datetime
 
-from sqlalchemy import event, select
+from sqlalchemy import select
 
 from fast_zero.models import User
 
 
-@contextmanager
-def mock_db_time(*, model, time=datetime(2024, 1, 1)):
-
-    def fake_time_handler(mapper, connection, target):
-        if hasattr(target, 'created_at'):
-            target.created_at = time
-        if hasattr(target, 'updated_at'):
-            target.updated_at = time
-
-    event.listen(model, 'before_insert', fake_time_handler)
-
-    yield time
-
-    event.remove(model, 'before_insert', fake_time_handler)
-
-
-def test_create_user(session):
+def test_create_user(session, mock_db_time):
     with mock_db_time(model=User) as time:
         new_user = User(
             username='alice', password='secret', email='teste@test'
         )
         session.add(new_user)
         session.commit()
 
-        user = session.scalar(select(User).where(User.username == 'alice'))
+    user = session.scalar(select(User).where(User.username == 'alice'))
 
     assert asdict(user) == {
         'id': 1,
         'username': 'alice',
         'password': 'secret',
         'email': 'teste@test',
         'created_at': time,
-        'updated_at': time,
+        'updated_at': time,  # Exercício
     }

codigo_das_aulas/05/tests/conftest.py

@@ -1,6 +1,9 @@
+from contextlib import contextmanager
+from datetime import datetime
+
 import pytest
 from fastapi.testclient import TestClient
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, event
 from sqlalchemy.orm import Session
 from sqlalchemy.pool import StaticPool
 
@@ -36,6 +39,27 @@ def session():
     table_registry.metadata.drop_all(engine)
 
 
+@contextmanager
+def _mock_db_time(*, model, time=datetime(2024, 1, 1)):
+
+    def fake_time_handler(mapper, connection, target):
+        if hasattr(target, 'created_at'):
+            target.created_at = time
+        if hasattr(target, 'updated_at'):
+            target.updated_at = time
+
+    event.listen(model, 'before_insert', fake_time_handler)
+
+    yield time
+
+    event.remove(model, 'before_insert', fake_time_handler)
+
+
+@pytest.fixture
+def mock_db_time():
+    return _mock_db_time
+
+
 @pytest.fixture
 def user(session):
     user = User(username='Teste', email='[email protected]', password='testtest')

codigo_das_aulas/05/tests/test_db.py

@@ -1,43 +1,25 @@
-from contextlib import contextmanager
 from dataclasses import asdict
-from datetime import datetime
 
-from sqlalchemy import event, select
+from sqlalchemy import select
 
 from fast_zero.models import User
 
 
-@contextmanager
-def mock_db_time(*, model, time=datetime(2024, 1, 1)):
-
-    def fake_time_handler(mapper, connection, target):
-        if hasattr(target, 'created_at'):
-            target.created_at = time
-        if hasattr(target, 'updated_at'):
-            target.updated_at = time
-
-    event.listen(model, 'before_insert', fake_time_handler)
-
-    yield time
-
-    event.remove(model, 'before_insert', fake_time_handler)
-
-
-def test_create_user(session):
+def test_create_user(session, mock_db_time):
     with mock_db_time(model=User) as time:
         new_user = User(
             username='alice', password='secret', email='teste@test'
         )
         session.add(new_user)
         session.commit()
 
-        user = session.scalar(select(User).where(User.username == 'alice'))
+    user = session.scalar(select(User).where(User.username == 'alice'))
 
     assert asdict(user) == {
         'id': 1,
         'username': 'alice',
         'password': 'secret',
         'email': 'teste@test',
         'created_at': time,
-        'updated_at': time,
+        'updated_at': time,  # Exercício
     }

codigo_das_aulas/06/tests/conftest.py

@@ -1,6 +1,9 @@
+from contextlib import contextmanager
+from datetime import datetime
+
 import pytest
 from fastapi.testclient import TestClient
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, event
 from sqlalchemy.orm import Session
 from sqlalchemy.pool import StaticPool
 
@@ -37,6 +40,27 @@ def session():
     table_registry.metadata.drop_all(engine)
 
 
+@contextmanager
+def _mock_db_time(*, model, time=datetime(2024, 1, 1)):
+
+    def fake_time_handler(mapper, connection, target):
+        if hasattr(target, 'created_at'):
+            target.created_at = time
+        if hasattr(target, 'updated_at'):
+            target.updated_at = time
+
+    event.listen(model, 'before_insert', fake_time_handler)
+
+    yield time
+
+    event.remove(model, 'before_insert', fake_time_handler)
+
+
+@pytest.fixture
+def mock_db_time():
+    return _mock_db_time
+
+
 @pytest.fixture
 def user(session):
     user = User(

codigo_das_aulas/06/tests/test_db.py

@@ -1,43 +1,25 @@
-from contextlib import contextmanager
 from dataclasses import asdict
-from datetime import datetime
 
-from sqlalchemy import event, select
+from sqlalchemy import select
 
 from fast_zero.models import User
 
 
-@contextmanager
-def mock_db_time(*, model, time=datetime(2024, 1, 1)):
-
-    def fake_time_handler(mapper, connection, target):
-        if hasattr(target, 'created_at'):
-            target.created_at = time
-        if hasattr(target, 'updated_at'):
-            target.updated_at = time
-
-    event.listen(model, 'before_insert', fake_time_handler)
-
-    yield time
-
-    event.remove(model, 'before_insert', fake_time_handler)
-
-
-def test_create_user(session):
+def test_create_user(session, mock_db_time):
     with mock_db_time(model=User) as time:
         new_user = User(
             username='alice', password='secret', email='teste@test'
         )
         session.add(new_user)
         session.commit()
 
-        user = session.scalar(select(User).where(User.username == 'alice'))
+    user = session.scalar(select(User).where(User.username == 'alice'))
 
     assert asdict(user) == {
         'id': 1,
         'username': 'alice',
         'password': 'secret',
         'email': 'teste@test',
         'created_at': time,
-        'updated_at': time,
+        'updated_at': time,  # Exercício
     }

codigo_das_aulas/07/tests/conftest.py

@@ -1,6 +1,9 @@
+from contextlib import contextmanager
+from datetime import datetime
+
 import pytest
 from fastapi.testclient import TestClient
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, event
 from sqlalchemy.orm import Session
 from sqlalchemy.pool import StaticPool
 
@@ -37,6 +40,27 @@ def session():
     table_registry.metadata.drop_all(engine)
 
 
+@contextmanager
+def _mock_db_time(*, model, time=datetime(2024, 1, 1)):
+
+    def fake_time_handler(mapper, connection, target):
+        if hasattr(target, 'created_at'):
+            target.created_at = time
+        if hasattr(target, 'updated_at'):
+            target.updated_at = time
+
+    event.listen(model, 'before_insert', fake_time_handler)
+
+    yield time
+
+    event.remove(model, 'before_insert', fake_time_handler)
+
+
+@pytest.fixture
+def mock_db_time():
+    return _mock_db_time
+
+
 @pytest.fixture
 def user(session):
     user = User(

codigo_das_aulas/07/tests/test_db.py

@@ -1,43 +1,25 @@
-from contextlib import contextmanager
 from dataclasses import asdict
-from datetime import datetime
 
-from sqlalchemy import event, select
+from sqlalchemy import select
 
 from fast_zero.models import User
 
 
-@contextmanager
-def mock_db_time(*, model, time=datetime(2024, 1, 1)):
-
-    def fake_time_handler(mapper, connection, target):
-        if hasattr(target, 'created_at'):
-            target.created_at = time
-        if hasattr(target, 'updated_at'):
-            target.updated_at = time
-
-    event.listen(model, 'before_insert', fake_time_handler)
-
-    yield time
-
-    event.remove(model, 'before_insert', fake_time_handler)
-
-
-def test_create_user(session):
+def test_create_user(session, mock_db_time):
     with mock_db_time(model=User) as time:
         new_user = User(
             username='alice', password='secret', email='teste@test'
         )
         session.add(new_user)
         session.commit()
 
-        user = session.scalar(select(User).where(User.username == 'alice'))
+    user = session.scalar(select(User).where(User.username == 'alice'))
 
     assert asdict(user) == {
         'id': 1,
         'username': 'alice',
         'password': 'secret',
         'email': 'teste@test',
         'created_at': time,
-        'updated_at': time,
+        'updated_at': time,  # Exercício
     }

codigo_das_aulas/08/tests/conftest.py

@@ -1,7 +1,10 @@
+from contextlib import contextmanager
+from datetime import datetime
+
 import factory
 import pytest
 from fastapi.testclient import TestClient
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, event
 from sqlalchemy.orm import Session
 from sqlalchemy.pool import StaticPool
 
@@ -52,6 +55,27 @@ def user(session):
     return user
 
 
+@contextmanager
+def _mock_db_time(*, model, time=datetime(2024, 1, 1)):
+
+    def fake_time_handler(mapper, connection, target):
+        if hasattr(target, 'created_at'):
+            target.created_at = time
+        if hasattr(target, 'updated_at'):
+            target.updated_at = time
+
+    event.listen(model, 'before_insert', fake_time_handler)
+
+    yield time
+
+    event.remove(model, 'before_insert', fake_time_handler)
+
+
+@pytest.fixture
+def mock_db_time():
+    return _mock_db_time
+
+
 @pytest.fixture
 def other_user(session):
     password = 'testtest'