17
17
build_batch_delete , DomainManager
18
18
from qiniu import urlsafe_base64_encode , urlsafe_base64_decode , canonical_mime_header_key
19
19
20
- from qiniu .compat import is_py2 , is_py3 , b
20
+ from qiniu .compat import is_py2 , is_py3 , b , json
21
21
22
22
from qiniu .services .storage .uploader import _form_put
23
23
@@ -414,7 +414,7 @@ def test_list(self):
414
414
assert len (ret .get ('items' )) == 4
415
415
ret , eof , info = self .bucket .list (bucket_name , limit = 1000 )
416
416
print (ret , eof , info )
417
- assert eof is False
417
+ assert info . status_code == 200
418
418
419
419
def test_buckets (self ):
420
420
ret , info = self .bucket .buckets ()
@@ -699,14 +699,26 @@ def test_putWithoutKey(self):
699
699
assert info .status_code == 403 # key not match
700
700
701
701
def test_withoutRead_withoutSeek_retry (self ):
702
- key = 'retry'
703
- data = 'hello retry!'
704
- set_default (default_zone = Zone ('http://a' , 'https://upload.qiniup.com' ))
705
- token = self .q .upload_token (bucket_name )
706
- ret , info = put_data (token , key , data )
707
- print (info )
708
- assert ret ['key' ] == key
709
- assert ret ['hash' ] == 'FlYu0iBR1WpvYi4whKXiBuQpyLLk'
702
+ try :
703
+ key = 'retry'
704
+ data = 'hello retry!'
705
+ zone = Zone ()
706
+ try :
707
+ hosts = json .loads (
708
+ zone .bucket_hosts (access_key , bucket_name )
709
+ ).get ('hosts' )
710
+ up_host_backup = 'https://' + hosts [0 ].get ('up' , {}).get ('domains' )[0 ]
711
+ except IndexError :
712
+ up_host_backup = 'https://upload.qiniup.com'
713
+ set_default (default_zone = Zone ('http://a' , up_host_backup ))
714
+ token = self .q .upload_token (bucket_name )
715
+ ret , info = put_data (token , key , data )
716
+ print (info )
717
+ assert ret ['key' ] == key
718
+ assert ret ['hash' ] == 'FlYu0iBR1WpvYi4whKXiBuQpyLLk'
719
+ finally :
720
+ set_default (default_zone = Zone ())
721
+ qiniu .config ._is_customized_default ['default_zone' ] = False
710
722
711
723
def test_putData_without_fname (self ):
712
724
if is_travis ():
@@ -780,7 +792,6 @@ def test_put_stream(self):
780
792
localfile = __file__
781
793
key = 'test_file_r'
782
794
size = os .stat (localfile ).st_size
783
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
784
795
with open (localfile , 'rb' ) as input_stream :
785
796
token = self .q .upload_token (bucket_name , key )
786
797
ret , info = put_stream (token , key , input_stream , os .path .basename (__file__ ), size , hostscache_dir ,
@@ -792,7 +803,6 @@ def test_put_stream_v2_without_bucket_name(self):
792
803
localfile = __file__
793
804
key = 'test_file_r'
794
805
size = os .stat (localfile ).st_size
795
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
796
806
with open (localfile , 'rb' ) as input_stream :
797
807
token = self .q .upload_token (bucket_name , key )
798
808
ret , info = put_stream (token , key , input_stream , os .path .basename (__file__ ), size , hostscache_dir ,
@@ -804,7 +814,6 @@ def test_put_2m_stream_v2(self):
804
814
localfile = create_temp_file (2 * 1024 * 1024 + 1 )
805
815
key = 'test_file_r'
806
816
size = os .stat (localfile ).st_size
807
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
808
817
with open (localfile , 'rb' ) as input_stream :
809
818
token = self .q .upload_token (bucket_name , key )
810
819
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -817,7 +826,6 @@ def test_put_4m_stream_v2(self):
817
826
localfile = create_temp_file (4 * 1024 * 1024 )
818
827
key = 'test_file_r'
819
828
size = os .stat (localfile ).st_size
820
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
821
829
with open (localfile , 'rb' ) as input_stream :
822
830
token = self .q .upload_token (bucket_name , key )
823
831
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -830,7 +838,6 @@ def test_put_10m_stream_v2(self):
830
838
localfile = create_temp_file (10 * 1024 * 1024 + 1 )
831
839
key = 'test_file_r'
832
840
size = os .stat (localfile ).st_size
833
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
834
841
with open (localfile , 'rb' ) as input_stream :
835
842
token = self .q .upload_token (bucket_name , key )
836
843
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -844,7 +851,6 @@ def test_put_stream_v2_without_key(self):
844
851
localfile = create_temp_file (part_size + 1 )
845
852
key = None
846
853
size = os .stat (localfile ).st_size
847
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
848
854
with open (localfile , 'rb' ) as input_stream :
849
855
token = self .q .upload_token (bucket_name , key )
850
856
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -858,7 +864,6 @@ def test_put_stream_v2_with_empty_return_body(self):
858
864
localfile = create_temp_file (part_size + 1 )
859
865
key = 'test_file_empty_return_body'
860
866
size = os .stat (localfile ).st_size
861
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
862
867
with open (localfile , 'rb' ) as input_stream :
863
868
token = self .q .upload_token (bucket_name , key , policy = {'returnBody' : ' ' })
864
869
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -873,7 +878,6 @@ def test_big_file(self):
873
878
token = self .q .upload_token (bucket_name , key )
874
879
localfile = create_temp_file (4 * 1024 * 1024 + 1 )
875
880
progress_handler = lambda progress , total : progress
876
- qiniu .set_default (default_zone = Zone ('http://a' , 'https://upload.qiniup.com' ))
877
881
ret , info = put_file (token , key , localfile , self .params , self .mime_type , progress_handler = progress_handler )
878
882
print (info )
879
883
assert ret ['key' ] == key
@@ -882,7 +886,6 @@ def test_big_file(self):
882
886
def test_retry (self ):
883
887
localfile = __file__
884
888
key = 'test_file_r_retry'
885
- qiniu .set_default (default_zone = Zone ('http://a' , 'https://upload.qiniup.com' ))
886
889
token = self .q .upload_token (bucket_name , key )
887
890
ret , info = put_file (token , key , localfile , self .params , self .mime_type )
888
891
print (info )
@@ -893,7 +896,6 @@ def test_put_stream_with_key_limits(self):
893
896
localfile = __file__
894
897
key = 'test_file_r'
895
898
size = os .stat (localfile ).st_size
896
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
897
899
with open (localfile , 'rb' ) as input_stream :
898
900
token = self .q .upload_token (bucket_name , key , policy = {'keylimit' : ['test_file_d' ]})
899
901
ret , info = put_stream (token , key , input_stream , os .path .basename (__file__ ), size , hostscache_dir ,
@@ -910,7 +912,6 @@ def test_put_stream_with_metadata(self):
910
912
localfile = __file__
911
913
key = 'test_put_stream_with_metadata'
912
914
size = os .stat (localfile ).st_size
913
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
914
915
with open (localfile , 'rb' ) as input_stream :
915
916
token = self .q .upload_token (bucket_name , key )
916
917
ret , info = put_stream (token , key , input_stream , os .path .basename (__file__ ), size , hostscache_dir ,
@@ -927,7 +928,6 @@ def test_put_stream_v2_with_metadata(self):
927
928
localfile = create_temp_file (part_size + 1 )
928
929
key = 'test_put_stream_v2_with_metadata'
929
930
size = os .stat (localfile ).st_size
930
- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
931
931
with open (localfile , 'rb' ) as input_stream :
932
932
token = self .q .upload_token (bucket_name , key )
933
933
ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
0 commit comments