Handle pickle in BaseUrlSession for base_url

HISTORY.rst

@@ -1,6 +1,14 @@
 History
 =======
 
+1.1.0
+-------------------
+
+Fixed Bugs
+~~~~~~~~~~
+
+- Save and recover ``base_url`` in ``BaseUrlSession`` during pickle and unpickle
+
 1.0.0 -- 2023-05-01
 -------------------
 

requests_toolbelt/sessions.py

@@ -87,3 +87,15 @@ def prepare_request(self, request, *args, **kwargs):
     def create_url(self, url):
         """Create the URL based off this partial path."""
         return urljoin(self.base_url, url)
+
+    def __getstate__(self):
+        """Save base URL as well during the pickle"""
+        states = super(BaseUrlSession, self).__getstate__()
+        states.update({"base_url": self.base_url})
+        return states
+
+    def __setstate__(self, state):
+        """Load base URL as well during the unpickle"""
+        super(BaseUrlSession, self).__setstate__(state)
+        if "base_url" in state:
+            self.base_url = state["base_url"]

tests/test_sessions.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import unittest
+import pickle
 import pytest
 
 from requests_toolbelt import sessions
@@ -53,3 +54,9 @@ def test_prepared_request_override_base(self):
             response = session.send(prepared_request)
         response.raise_for_status()
         assert response.json()['headers']['Host'] == 'httpbin.org'
+
+    def test_pickle_unpickle_session(self):
+        session = sessions.BaseUrlSession('https://www.google.com')
+        pickled_session = pickle.dumps(session)
+        unpickled_session = pickle.loads(pickled_session)
+        assert session.base_url == unpickled_session.base_url