1
+ from datetime import time
1
2
import unittest
2
3
import os
3
4
import requests_mock
4
5
import tableauserverclient as TSC
5
- from datetime import time
6
+ from tableauserverclient . datetime_helpers import format_datetime
6
7
7
8
TEST_ASSET_DIR = os .path .join (os .path .dirname (__file__ ), "assets" )
8
9
@@ -37,19 +38,19 @@ def test_get(self):
37
38
self .assertEqual ("Weekday early mornings" , all_schedules [0 ].name )
38
39
self .assertEqual ("Active" , all_schedules [0 ].state )
39
40
self .assertEqual (50 , all_schedules [0 ].priority )
40
- self .assertEqual ("2016-07-06T20:19:00Z" , all_schedules [0 ].created_at )
41
- self .assertEqual ("2016-09-13T11:00:32Z" , all_schedules [0 ].updated_at )
41
+ self .assertEqual ("2016-07-06T20:19:00Z" , format_datetime ( all_schedules [0 ].created_at ) )
42
+ self .assertEqual ("2016-09-13T11:00:32Z" , format_datetime ( all_schedules [0 ].updated_at ) )
42
43
self .assertEqual ("Extract" , all_schedules [0 ].schedule_type )
43
- self .assertEqual ("2016-09-14T11:00:00Z" , all_schedules [0 ].next_run_at )
44
+ self .assertEqual ("2016-09-14T11:00:00Z" , format_datetime ( all_schedules [0 ].next_run_at ) )
44
45
45
46
self .assertEqual ("bcb79d07-6e47-472f-8a65-d7f51f40c36c" , all_schedules [1 ].id )
46
47
self .assertEqual ("Saturday night" , all_schedules [1 ].name )
47
48
self .assertEqual ("Active" , all_schedules [1 ].state )
48
49
self .assertEqual (80 , all_schedules [1 ].priority )
49
- self .assertEqual ("2016-07-07T20:19:00Z" , all_schedules [1 ].created_at )
50
- self .assertEqual ("2016-09-12T16:39:38Z" , all_schedules [1 ].updated_at )
50
+ self .assertEqual ("2016-07-07T20:19:00Z" , format_datetime ( all_schedules [1 ].created_at ) )
51
+ self .assertEqual ("2016-09-12T16:39:38Z" , format_datetime ( all_schedules [1 ].updated_at ) )
51
52
self .assertEqual ("Subscription" , all_schedules [1 ].schedule_type )
52
- self .assertEqual ("2016-09-18T06:00:00Z" , all_schedules [1 ].next_run_at )
53
+ self .assertEqual ("2016-09-18T06:00:00Z" , format_datetime ( all_schedules [1 ].next_run_at ) )
53
54
54
55
def test_get_empty (self ):
55
56
with open (GET_EMPTY_XML , "rb" ) as f :
@@ -82,10 +83,10 @@ def test_create_hourly(self):
82
83
self .assertEqual ("hourly-schedule-1" , new_schedule .name )
83
84
self .assertEqual ("Active" , new_schedule .state )
84
85
self .assertEqual (50 , new_schedule .priority )
85
- self .assertEqual ("2016-09-15T20:47:33Z" , new_schedule .created_at )
86
- self .assertEqual ("2016-09-15T20:47:33Z" , new_schedule .updated_at )
86
+ self .assertEqual ("2016-09-15T20:47:33Z" , format_datetime ( new_schedule .created_at ) )
87
+ self .assertEqual ("2016-09-15T20:47:33Z" , format_datetime ( new_schedule .updated_at ) )
87
88
self .assertEqual (TSC .ScheduleItem .Type .Extract , new_schedule .schedule_type )
88
- self .assertEqual ("2016-09-16T01:30:00Z" , new_schedule .next_run_at )
89
+ self .assertEqual ("2016-09-16T01:30:00Z" , format_datetime ( new_schedule .next_run_at ) )
89
90
self .assertEqual (TSC .ScheduleItem .ExecutionOrder .Parallel , new_schedule .execution_order )
90
91
self .assertEqual (time (2 , 30 ), new_schedule .interval_item .start_time )
91
92
self .assertEqual (time (23 ), new_schedule .interval_item .end_time )
@@ -105,10 +106,10 @@ def test_create_daily(self):
105
106
self .assertEqual ("daily-schedule-1" , new_schedule .name )
106
107
self .assertEqual ("Active" , new_schedule .state )
107
108
self .assertEqual (90 , new_schedule .priority )
108
- self .assertEqual ("2016-09-15T21:01:09Z" , new_schedule .created_at )
109
- self .assertEqual ("2016-09-15T21:01:09Z" , new_schedule .updated_at )
109
+ self .assertEqual ("2016-09-15T21:01:09Z" , format_datetime ( new_schedule .created_at ) )
110
+ self .assertEqual ("2016-09-15T21:01:09Z" , format_datetime ( new_schedule .updated_at ) )
110
111
self .assertEqual (TSC .ScheduleItem .Type .Subscription , new_schedule .schedule_type )
111
- self .assertEqual ("2016-09-16T11:45:00Z" , new_schedule .next_run_at )
112
+ self .assertEqual ("2016-09-16T11:45:00Z" , format_datetime ( new_schedule .next_run_at ) )
112
113
self .assertEqual (TSC .ScheduleItem .ExecutionOrder .Serial , new_schedule .execution_order )
113
114
self .assertEqual (time (4 , 45 ), new_schedule .interval_item .start_time )
114
115
@@ -128,10 +129,10 @@ def test_create_weekly(self):
128
129
self .assertEqual ("weekly-schedule-1" , new_schedule .name )
129
130
self .assertEqual ("Active" , new_schedule .state )
130
131
self .assertEqual (80 , new_schedule .priority )
131
- self .assertEqual ("2016-09-15T21:12:50Z" , new_schedule .created_at )
132
- self .assertEqual ("2016-09-15T21:12:50Z" , new_schedule .updated_at )
132
+ self .assertEqual ("2016-09-15T21:12:50Z" , format_datetime ( new_schedule .created_at ) )
133
+ self .assertEqual ("2016-09-15T21:12:50Z" , format_datetime ( new_schedule .updated_at ) )
133
134
self .assertEqual (TSC .ScheduleItem .Type .Extract , new_schedule .schedule_type )
134
- self .assertEqual ("2016-09-16T16:15:00Z" , new_schedule .next_run_at )
135
+ self .assertEqual ("2016-09-16T16:15:00Z" , format_datetime ( new_schedule .next_run_at ) )
135
136
self .assertEqual (TSC .ScheduleItem .ExecutionOrder .Parallel , new_schedule .execution_order )
136
137
self .assertEqual (time (9 , 15 ), new_schedule .interval_item .start_time )
137
138
self .assertEqual (("Monday" , "Wednesday" , "Friday" ),
@@ -151,10 +152,10 @@ def test_create_monthly(self):
151
152
self .assertEqual ("monthly-schedule-1" , new_schedule .name )
152
153
self .assertEqual ("Active" , new_schedule .state )
153
154
self .assertEqual (20 , new_schedule .priority )
154
- self .assertEqual ("2016-09-15T21:16:56Z" , new_schedule .created_at )
155
- self .assertEqual ("2016-09-15T21:16:56Z" , new_schedule .updated_at )
155
+ self .assertEqual ("2016-09-15T21:16:56Z" , format_datetime ( new_schedule .created_at ) )
156
+ self .assertEqual ("2016-09-15T21:16:56Z" , format_datetime ( new_schedule .updated_at ) )
156
157
self .assertEqual (TSC .ScheduleItem .Type .Extract , new_schedule .schedule_type )
157
- self .assertEqual ("2016-10-12T14:00:00Z" , new_schedule .next_run_at )
158
+ self .assertEqual ("2016-10-12T14:00:00Z" , format_datetime ( new_schedule .next_run_at ) )
158
159
self .assertEqual (TSC .ScheduleItem .ExecutionOrder .Serial , new_schedule .execution_order )
159
160
self .assertEqual (time (7 ), new_schedule .interval_item .start_time )
160
161
self .assertEqual ("12" , new_schedule .interval_item .interval )
@@ -174,9 +175,9 @@ def test_update(self):
174
175
self .assertEqual ("7bea1766-1543-4052-9753-9d224bc069b5" , single_schedule .id )
175
176
self .assertEqual ("weekly-schedule-1" , single_schedule .name )
176
177
self .assertEqual (90 , single_schedule .priority )
177
- self .assertEqual ("2016-09-15T23:50:02Z" , single_schedule .updated_at )
178
+ self .assertEqual ("2016-09-15T23:50:02Z" , format_datetime ( single_schedule .updated_at ) )
178
179
self .assertEqual (TSC .ScheduleItem .Type .Extract , single_schedule .schedule_type )
179
- self .assertEqual ("2016-09-16T14:00:00Z" , single_schedule .next_run_at )
180
+ self .assertEqual ("2016-09-16T14:00:00Z" , format_datetime ( single_schedule .next_run_at ) )
180
181
self .assertEqual (TSC .ScheduleItem .ExecutionOrder .Parallel , single_schedule .execution_order )
181
182
self .assertEqual (time (7 ), single_schedule .interval_item .start_time )
182
183
self .assertEqual (("Monday" , "Friday" ),
0 commit comments