1717 build_batch_delete , DomainManager
1818from qiniu import urlsafe_base64_encode , urlsafe_base64_decode , canonical_mime_header_key
1919
20- from qiniu .compat import is_py2 , is_py3 , b
20+ from qiniu .compat import is_py2 , is_py3 , b , json
2121
2222from qiniu .services .storage .uploader import _form_put
2323
@@ -414,7 +414,7 @@ def test_list(self):
414414 assert len (ret .get ('items' )) == 4
415415 ret , eof , info = self .bucket .list (bucket_name , limit = 1000 )
416416 print (ret , eof , info )
417- assert eof is False
417+ assert info . status_code == 200
418418
419419 def test_buckets (self ):
420420 ret , info = self .bucket .buckets ()
@@ -699,14 +699,26 @@ def test_putWithoutKey(self):
699699 assert info .status_code == 403 # key not match
700700
701701 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
710722
711723 def test_putData_without_fname (self ):
712724 if is_travis ():
@@ -780,7 +792,6 @@ def test_put_stream(self):
780792 localfile = __file__
781793 key = 'test_file_r'
782794 size = os .stat (localfile ).st_size
783- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
784795 with open (localfile , 'rb' ) as input_stream :
785796 token = self .q .upload_token (bucket_name , key )
786797 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):
792803 localfile = __file__
793804 key = 'test_file_r'
794805 size = os .stat (localfile ).st_size
795- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
796806 with open (localfile , 'rb' ) as input_stream :
797807 token = self .q .upload_token (bucket_name , key )
798808 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):
804814 localfile = create_temp_file (2 * 1024 * 1024 + 1 )
805815 key = 'test_file_r'
806816 size = os .stat (localfile ).st_size
807- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
808817 with open (localfile , 'rb' ) as input_stream :
809818 token = self .q .upload_token (bucket_name , key )
810819 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):
817826 localfile = create_temp_file (4 * 1024 * 1024 )
818827 key = 'test_file_r'
819828 size = os .stat (localfile ).st_size
820- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
821829 with open (localfile , 'rb' ) as input_stream :
822830 token = self .q .upload_token (bucket_name , key )
823831 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):
830838 localfile = create_temp_file (10 * 1024 * 1024 + 1 )
831839 key = 'test_file_r'
832840 size = os .stat (localfile ).st_size
833- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
834841 with open (localfile , 'rb' ) as input_stream :
835842 token = self .q .upload_token (bucket_name , key )
836843 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):
844851 localfile = create_temp_file (part_size + 1 )
845852 key = None
846853 size = os .stat (localfile ).st_size
847- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
848854 with open (localfile , 'rb' ) as input_stream :
849855 token = self .q .upload_token (bucket_name , key )
850856 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):
858864 localfile = create_temp_file (part_size + 1 )
859865 key = 'test_file_empty_return_body'
860866 size = os .stat (localfile ).st_size
861- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
862867 with open (localfile , 'rb' ) as input_stream :
863868 token = self .q .upload_token (bucket_name , key , policy = {'returnBody' : ' ' })
864869 ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
@@ -873,7 +878,6 @@ def test_big_file(self):
873878 token = self .q .upload_token (bucket_name , key )
874879 localfile = create_temp_file (4 * 1024 * 1024 + 1 )
875880 progress_handler = lambda progress , total : progress
876- qiniu .set_default (default_zone = Zone ('http://a' , 'https://upload.qiniup.com' ))
877881 ret , info = put_file (token , key , localfile , self .params , self .mime_type , progress_handler = progress_handler )
878882 print (info )
879883 assert ret ['key' ] == key
@@ -882,7 +886,6 @@ def test_big_file(self):
882886 def test_retry (self ):
883887 localfile = __file__
884888 key = 'test_file_r_retry'
885- qiniu .set_default (default_zone = Zone ('http://a' , 'https://upload.qiniup.com' ))
886889 token = self .q .upload_token (bucket_name , key )
887890 ret , info = put_file (token , key , localfile , self .params , self .mime_type )
888891 print (info )
@@ -893,7 +896,6 @@ def test_put_stream_with_key_limits(self):
893896 localfile = __file__
894897 key = 'test_file_r'
895898 size = os .stat (localfile ).st_size
896- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
897899 with open (localfile , 'rb' ) as input_stream :
898900 token = self .q .upload_token (bucket_name , key , policy = {'keylimit' : ['test_file_d' ]})
899901 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):
910912 localfile = __file__
911913 key = 'test_put_stream_with_metadata'
912914 size = os .stat (localfile ).st_size
913- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
914915 with open (localfile , 'rb' ) as input_stream :
915916 token = self .q .upload_token (bucket_name , key )
916917 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):
927928 localfile = create_temp_file (part_size + 1 )
928929 key = 'test_put_stream_v2_with_metadata'
929930 size = os .stat (localfile ).st_size
930- set_default (default_zone = Zone ('https://upload.qiniup.com' ))
931931 with open (localfile , 'rb' ) as input_stream :
932932 token = self .q .upload_token (bucket_name , key )
933933 ret , info = put_stream (token , key , input_stream , os .path .basename (localfile ), size , hostscache_dir ,
0 commit comments