fix regression for as_user() and as_admin()

Author: rob

dropbox/dropbox.py

@@ -596,4 +596,12 @@ def _get_dropbox_client_with_select_header(self, select_header_name, team_member
 
         new_headers = self._headers.copy() if self._headers else {}
         new_headers[select_header_name] = team_member_id
-        return self.clone(headers=new_headers)
+        return Dropbox(
+            self._oauth2_access_token,
+            max_retries_on_error=self._max_retries_on_error,
+            max_retries_on_rate_limit=self._max_retries_on_rate_limit,
+            timeout=self._timeout,
+            user_agent=self._raw_user_agent,
+            session=self._session,
+            headers=new_headers,
+        )

test/test_dropbox.py

@@ -144,6 +144,28 @@ def test_team(self, dbxt):
             team_member_id = r.members[0].profile.team_member_id
             dbxt.as_user(team_member_id).files_list_folder('')
 
+    @dbx_team_from_env
+    def test_as_user(self, dbxt):
+        dbx_as_user = dbxt.as_user('1')
+        self.assertIsInstance(dbx_as_user, Dropbox)
+
+    @dbx_team_from_env
+    def test_as_admin(self, dbxt):
+        dbx_as_admin = dbxt.as_admin('1')
+        self.assertIsInstance(dbx_as_admin, Dropbox)
+
+    @dbx_from_env
+    def test_clone_when_user_linked(self, dbx):
+        new_dbx = dbx.clone()
+        self.assertIsNot(dbx, new_dbx)
+        self.assertIsInstance(new_dbx, dbx.__class__)
+
+    @dbx_team_from_env
+    def test_clone_when_team_linked(self, dbxt):
+        new_dbxt = dbxt.clone()
+        self.assertIsNot(dbxt, new_dbxt)
+        self.assertIsInstance(new_dbxt, dbxt.__class__)
+
     @dbx_from_env
     def test_with_path_root_constructor(self, dbx):
         # Verify valid mode types