@@ -14,13 +14,21 @@ def setUp(self):
14
14
def test_basic_unload_case (self ):
15
15
''' Tests that the simplest type of UnloadFromSelect works
16
16
'''
17
- expected_result = "unload ('SELECT count(t1.id) AS count_1 \n FROM t1') to 'cookies' credentials 'aws_access_key_id=cookies;aws_secret_access_key=cookies' delimiter ',' addquotes allowoverwrite parallel on"
18
- insert = UnloadFromSelect (select ([func .count (self .t1 .c .id )]), 'cookies' , 'cookies' , 'cookies' )
19
- self .assertEqual (expected_result , str (insert ))
17
+ expected_result = "UNLOAD ('SELECT count(t1.id) AS count_1 \n FROM t1') TO 's3://bucket/key'\n " \
18
+ "CREDENTIALS 'aws_access_key_id=cookies;aws_secret_access_key=cookies;token=cookies'\n " \
19
+ " DELIMITER ','\n ADDQUOTES\n ALLOWOVERWRITE\n " \
20
+ " PARALLEL ON;"
21
+ insert = UnloadFromSelect (select ([func .count (self .t1 .c .id )]), 's3://bucket/key' , 'cookies' , 'cookies' ,
22
+ 'cookies' )
23
+ self .assertEqual (expected_result , str (insert ).strip ())
20
24
21
25
def test_parallel_off_unload_case (self ):
22
26
''' Tests that UnloadFromSelect handles parallel being set to off
23
27
'''
24
- expected_result = "unload ('SELECT count(t1.id) AS count_1 \n FROM t1') to 'cookies' credentials 'aws_access_key_id=cookies;aws_secret_access_key=cookies' delimiter ',' addquotes allowoverwrite parallel off"
25
- insert = UnloadFromSelect (select ([func .count (self .t1 .c .id )]), 'cookies' , 'cookies' , 'cookies' , 'off' )
26
- self .assertEqual (expected_result , str (insert ))
28
+ expected_result = "UNLOAD ('SELECT count(t1.id) AS count_1 \n FROM t1') TO 's3://bucket/key'\n " \
29
+ "CREDENTIALS 'aws_access_key_id=cookies;aws_secret_access_key=cookies;token=cookies'\n " \
30
+ " DELIMITER ','\n ADDQUOTES\n ALLOWOVERWRITE\n " \
31
+ " PARALLEL OFF;"
32
+ insert = UnloadFromSelect (select ([func .count (self .t1 .c .id )]), 's3://bucket/key' , 'cookies' , 'cookies' ,
33
+ 'cookies' , {'parallel' : 'OFF' })
34
+ self .assertEqual (expected_result , str (insert ).strip ())
0 commit comments