Raise exception on user_quota, tier_* on SiteItem

Author: Jordan Woods

tableauserverclient/server/endpoint/sites_endpoint.py

@@ -67,6 +67,13 @@ def update(self, site_item):
                 error = 'You cannot set admin_mode to ContentOnly and also set a user quota'
                 raise ValueError(error)
 
+        if site_item.user_quota:
+            if any([site_item.tier_creator_capacity,
+                    site_item.tier_explorer_capacity,
+                    site_item.tier_viewer_capacity, ]):
+                error = 'You cannot set tiered capacity and a user quota.'
+                raise ValueError(error)
+
         url = "{0}/{1}".format(self.baseurl, site_item.id)
         update_req = RequestFactory.Site.update_req(site_item)
         server_response = self.put_request(url, update_req)

test/test_site.py

@@ -156,3 +156,18 @@ def test_decrypt(self):
         with requests_mock.mock() as m:
             m.post(self.baseurl + '/0626857c-1def-4503-a7d8-7907c3ff9d9f/decrypt-extracts', status_code=200)
             self.server.sites.decrypt_extracts('0626857c-1def-4503-a7d8-7907c3ff9d9f')
+
+    def test_quota_and_tiers_error(self):
+        with open(UPDATE_XML, 'rb') as f:
+            response_xml = f.read().decode('utf-8')
+        with requests_mock.mock() as m:
+            m.put(self.baseurl + '/6b7179ba-b82b-4f0f-91ed-812074ac5da6', text=response_xml)
+            single_site = TSC.SiteItem(name='Tableau', content_url='tableau',
+                                       admin_mode=TSC.SiteItem.AdminMode.ContentAndUsers,
+                                       user_quota=15, storage_quota=1000,
+                                       disable_subscriptions=True, revision_history_enabled=False,
+                                       data_acceleration_mode='disable')
+            single_site.tier_creator_capacity = 11
+            single_site._id = '6b7179ba-b82b-4f0f-91ed-812074ac5da6'
+            with self.assertRaises(ValueError):
+                single_site = self.server.sites.update(single_site)