Módulo e Import como Objeto

renzon · renzon · commit f5561cdd6520 · 2018-05-20T08:01:14.000-03:00
diff --git a/libpythonpro/github_api.py b/libpythonpro/github_api.py
@@ -11,7 +11,3 @@ def buscar_avatar(usuario):
     url = f'https://api.github.com/users/{usuario}'
     resp = requests.get(url)
     return resp.json()['avatar_url']
-
-
-if __name__ == '__main__':
-    print(buscar_avatar('renzon'))
diff --git a/libpythonpro/tests/test_spam/test_github_api.py b/libpythonpro/tests/test_spam/test_github_api.py
@@ -0,0 +1,14 @@
+from unittest.mock import Mock
+
+from libpythonpro import github_api
+
+
+def test_buscar_avatar():
+    resp_mock = Mock()
+    resp_mock.json.return_value = {
+        'login': 'renzo', 'id': 402714,
+        'avatar_url': 'https://avatars3.githubusercontent.com/u/402714?v=4',
+    }
+    github_api.requests.get = Mock(return_value=resp_mock)
+    url = github_api.buscar_avatar('renzo')
+    assert 'https://avatars3.githubusercontent.com/u/402714?v=4' == url
