This repository was archived by the owner on Nov 2, 2020. It is now read-only.
forked from tableau/server-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_subscription.py
More file actions
74 lines (60 loc) · 3.68 KB
/
test_subscription.py
File metadata and controls
74 lines (60 loc) · 3.68 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import unittest
import os
import requests_mock
import tableauserverclient as TSC
TEST_ASSET_DIR = os.path.join(os.path.dirname(__file__), "assets")
CREATE_XML = os.path.join(TEST_ASSET_DIR, "subscription_create.xml")
GET_XML = os.path.join(TEST_ASSET_DIR, "subscription_get.xml")
GET_XML_BY_ID = os.path.join(TEST_ASSET_DIR, "subscription_get_by_id.xml")
class SubscriptionTests(unittest.TestCase):
def setUp(self):
self.server = TSC.Server("http://test")
self.server.version = '2.6'
# Fake Signin
self.server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
self.server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
self.baseurl = self.server.subscriptions.baseurl
def test_get_subscriptions(self):
with open(GET_XML, "rb") as f:
response_xml = f.read().decode("utf-8")
with requests_mock.mock() as m:
m.get(self.baseurl, text=response_xml)
all_subscriptions, pagination_item = self.server.subscriptions.get()
subscription = all_subscriptions[0]
self.assertEqual('382e9a6e-0c08-4a95-b6c1-c14df7bac3e4', subscription.id)
self.assertEqual('View', subscription.target.type)
self.assertEqual('cdd716ca-5818-470e-8bec-086885dbadee', subscription.target.id)
self.assertEqual('c0d5fc44-ad8c-4957-bec0-b70ed0f8df1e', subscription.user_id)
self.assertEqual('Not Found Alert', subscription.subject)
self.assertEqual('7617c389-cdca-4940-a66e-69956fcebf3e', subscription.schedule_id)
def test_get_subscription_by_id(self):
with open(GET_XML_BY_ID, "rb") as f:
response_xml = f.read().decode("utf-8")
with requests_mock.mock() as m:
m.get(self.baseurl + '/382e9a6e-0c08-4a95-b6c1-c14df7bac3e4', text=response_xml)
subscription = self.server.subscriptions.get_by_id('382e9a6e-0c08-4a95-b6c1-c14df7bac3e4')
self.assertEqual('382e9a6e-0c08-4a95-b6c1-c14df7bac3e4', subscription.id)
self.assertEqual('View', subscription.target.type)
self.assertEqual('cdd716ca-5818-470e-8bec-086885dbadee', subscription.target.id)
self.assertEqual('c0d5fc44-ad8c-4957-bec0-b70ed0f8df1e', subscription.user_id)
self.assertEqual('Not Found Alert', subscription.subject)
self.assertEqual('7617c389-cdca-4940-a66e-69956fcebf3e', subscription.schedule_id)
def test_create_subscription(self):
with open(CREATE_XML, 'rb') as f:
response_xml = f.read().decode('utf-8')
with requests_mock.mock() as m:
m.post(self.baseurl, text=response_xml)
target_item = TSC.Target("960e61f2-1838-40b2-bba2-340c9492f943", "workbook")
new_subscription = TSC.SubscriptionItem("subject", "4906c453-d5ec-4972-9ff4-789b629bdfa2",
"8d30c8de-0a5f-4bee-b266-c621b4f3eed0", target_item)
new_subscription = self.server.subscriptions.create(new_subscription)
self.assertEqual("78e9318d-2d29-4d67-b60f-3f2f5fd89ecc", new_subscription.id)
self.assertEqual("sub_name", new_subscription.subject)
self.assertEqual("960e61f2-1838-40b2-bba2-340c9492f943", new_subscription.target.id)
self.assertEqual("Workbook", new_subscription.target.type)
self.assertEqual("4906c453-d5ec-4972-9ff4-789b629bdfa2", new_subscription.schedule_id)
self.assertEqual("8d30c8de-0a5f-4bee-b266-c621b4f3eed0", new_subscription.user_id)
def test_delete_subscription(self):
with requests_mock.mock() as m:
m.delete(self.baseurl + '/78e9318d-2d29-4d67-b60f-3f2f5fd89ecc', status_code=204)
self.server.subscriptions.delete('78e9318d-2d29-4d67-b60f-3f2f5fd89ecc')