add test cheking GitHubSession pickling

zubchick · zubchick · commit 5da99705a8b0 · 2014-12-05T19:55:26.000+03:00
diff --git a/tests/unit/test_github_session.py b/tests/unit/test_github_session.py
@@ -1,3 +1,4 @@
+import cPickle as pickle
 import pytest
 
 import requests
@@ -231,3 +232,14 @@ def test_retrieve_client_credentials_returns_none(self):
         """
         s = self.build_session()
         assert s.retrieve_client_credentials() == (None, None)
+
+    def test_pickling(self):
+        s = self.build_session('https://api.github.com')
+        dumped = pickle.dumps(s, pickle.HIGHEST_PROTOCOL)
+        loaded = pickle.loads(dumped)
+
+        assert hasattr(loaded, 'base_url')
+        assert hasattr(loaded, 'two_factor_auth_cb')
+
+        assert loaded.base_url == s.base_url
+        assert loaded.two_factor_auth_cb == s.two_factor_auth_cb
