Isolated tests implemented with Mocks and Fixtures.

libpythonpro_doug/github_api.py

@@ -10,7 +10,3 @@ def search_avatar(user):
     url = f'https://api.github.com/users/{user}'
     resp = requests.get(url)
     return resp.json()['avatar_url']
-
-
-if __name__ == '__main__':
-    print(search_avatar('dougfraga'))

libpythonpro_doug/tests/test_spam/test_github_api.py

@@ -0,0 +1,29 @@
+from unittest.mock import Mock
+
+import pytest
+
+from libpythonpro_doug import github_api
+
+
+@pytest.fixture
+def avatar_url():
+    resp_mock = Mock()
+    url = 'https://avatars3.githubusercontent.com/u/61299254?v=4'
+    resp_mock.json.return_value = {
+        'login': 'dougfraga', 'id': 61299254,
+        'avatar_url': url
+    }
+    get_original = github_api.requests.get
+    github_api.requests.get = Mock(return_value=resp_mock)
+    yield url
+    github_api.requests.get = get_original
+
+
+def test_search_avatar(avatar_url):
+    url = github_api.search_avatar('dougfraga')
+    assert avatar_url == url
+
+
+def test_search_avatar_integration():
+    url = github_api.search_avatar('dougfraga')
+    assert 'https://avatars3.githubusercontent.com/u/61299254?v=4' == url

libpythonpro_doug/tests/test_spam/test_send_to_user_base.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from libpythonpro_doug.spam.email_sender import Sender
+# from libpythonpro_doug.spam.email_sender import Sender
 from libpythonpro_doug.spam.main import SpamSender
 from libpythonpro_doug.spam.models import User