time to mock the dependencies

Author: mikeckennedy

apps/py/ch09_testing/test_fixtures.py

@@ -0,0 +1,17 @@
+from typing import List
+import pytest
+import lib
+
+
+@pytest.fixture
+def guitar_data() -> List[lib.Guitar]:
+    # This guitar data simulates our mock data to avoid DB calls and dependencies.
+    guitars = [
+        lib.Guitar('Jet Black Electric', 599, '/static/img/guitars/jet-black-electric.jpg', style='electric'),
+        lib.Guitar('Weezer Classic', 1499, '/static/img/guitars/weezer-classic.jpg', style='electric'),
+        lib.Guitar('Acoustic Black', 1299, '/static/img/guitars/black-acoustic.jpg', style='acoustic'),
+        lib.Guitar('Brush Riffs', 599, '/static/img/guitars/brushed-black-electric.jpg', style='electric'),
+        lib.Guitar('Electric Wood Grain', 399, '/static/img/guitars/woodgrain-electric.jpg', style='electric'),
+    ]
+    guitars.sort(key=lambda g: g.price, reverse=True)
+    return guitars

apps/py/ch09_testing/test_lib.py

@@ -1,4 +1,20 @@
+import lib
+from test_fixtures import guitar_data
 
 
 def test_something():
     assert 7 == 7
+
+
+# def test_electric_guitars_wrong():
+#     style = 'electric'
+#     guitars = lib.all_guitars(style)
+#     # sweet little generator expression
+#     assert all(g.style == style for g in guitars)
+
+
+def test_electric_guitars(guitar_data):
+    style = 'electric'
+    guitars = lib.all_guitars(style)
+    # sweet little generator expression
+    assert all(g.style == style for g in guitars)