test: setting user idp id

Author: jorwoods

test/test_site.py

@@ -276,14 +276,13 @@ def test_list_auth_configurations(self) -> None:
 
         assert len(configs) == 2, "Expected 2 auth configurations"
 
-        assert configs[0].auth_setting =="OIDC"
+        assert configs[0].auth_setting == "OIDC"
         assert configs[0].enabled
         assert configs[0].idp_configuration_id == "00000000-0000-0000-0000-000000000000"
         assert configs[0].idp_configuration_name == "Initial Salesforce"
         assert configs[0].known_provider_alias == "Salesforce"
-        assert configs[1].auth_setting =="SAML"
+        assert configs[1].auth_setting == "SAML"
         assert configs[1].enabled
         assert configs[1].idp_configuration_id == "11111111-1111-1111-1111-111111111111"
         assert configs[1].idp_configuration_name == "Initial SAML"
         assert configs[1].known_provider_alias is None
-

test/test_user.py

@@ -1,6 +1,7 @@
 import os
 import unittest
 
+from defusedxml import ElementTree as ET
 import requests_mock
 
 import tableauserverclient as TSC
@@ -249,3 +250,20 @@ def test_get_users_from_file(self):
             users, failures = self.server.users.create_from_file(USERS)
         assert users[0].name == "Cassie", users
         assert failures == []
+
+    def test_add_user_idp_configuration(self):
+        with open(ADD_XML) as f:
+            response_xml = f.read()
+        user = TSC.UserItem(name="Cassie", site_role="Viewer", auth_setting="ServerDefault")
+        user.idp_configuration_id = "012345"
+
+        with requests_mock.mock() as m:
+            m.post(self.server.users.baseurl, text=response_xml)
+            user = self.server.users.add(user)
+
+            history = m.request_history[0]
+
+        tree = ET.fromstring(history.text)
+        user_elem = tree.find(".//user")
+        assert user_elem is not None
+        assert user_elem.attrib["idpConfigurationId"] == "012345"