
    =*hbU                     .   d dl mZmZmZmZmZmZmZmZm	Z	m
Z
 d dlmZmZ d dlmZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZmZmZ deeeee   f   eeee
   f   f   fdZ  G d	 d
ejB                        Z" G d dejB                        Z# G d dejB                        Z$ G d dejB                        Z% G d dejB                        Z& G d dejB                        Z' G d dejB                        Z( G d dejB                        Z)y)    )

url_concatparse_multipart_form_dataHTTPHeadersformat_timestampHTTPServerRequestparse_request_start_lineparse_cookie	qs_to_qslHTTPInputErrorHTTPFile)utf8
native_str)gen_log)ignore_deprecationN)TupleDictListreturnc                  
    i i fS )zReturn two empty dicts suitable for use with parse_multipart_form_data.

    mypy insists on type annotations for dict literals, so this lets us avoid
    the verbose types throughout this test.
     r       </usr/lib/python3/dist-packages/tornado/test/httputil_test.pyform_data_argsr      s     r6Mr   c                   T    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zy)TestUrlConcatc                 D    t        dddg      }| j                  |d       y )Nhttps://localhost/pathyr   zr!   https://localhost/path?y=y&z=zr   assertEqualselfurls     r   test_url_concat_no_query_paramsz-TestUrlConcat.test_url_concat_no_query_params&   s%    1J
3KL>?r   c                 D    t        dddg      }| j                  |d       y )Nr   )r   z/yr    z!https://localhost/path?y=%2Fy&z=zr#   r%   s     r   test_url_concat_encode_argsz)TestUrlConcat.test_url_concat_encode_args*   s%    1K3LMABr   c                 D    t        dddg      }| j                  |d       y )Nzhttps://localhost/path?r   r    r"   r#   r%   s     r   test_url_concat_trailing_qz(TestUrlConcat.test_url_concat_trailing_q.   s%    2Z4LM>?r   c                 D    t        dddg      }| j                  |d       y )Nzhttps://localhost/path?xr   r    !https://localhost/path?x=&y=y&z=zr#   r%   s     r   &test_url_concat_q_with_no_trailing_ampz4TestUrlConcat.test_url_concat_q_with_no_trailing_amp2   s%    3j*5MNABr   c                 D    t        dddg      }| j                  |d       y )Nzhttps://localhost/path?x&r   r    r.   r#   r%   s     r   test_url_concat_trailing_ampz*TestUrlConcat.test_url_concat_trailing_amp6   s%    4z:6NOABr   c                 D    t        dddg      }| j                  |d       y )Nzhttps://localhost/path?a=1&b=2r   r    z&https://localhost/path?a=1&b=2&y=y&z=zr#   r%   s     r   test_url_concat_mult_paramsz)TestUrlConcat.test_url_concat_mult_params:   s%    9J
;STFGr   c                 @    t        dg       }| j                  |d       y Nzhttps://localhost/path?r=1&t=2r#   r%   s     r   test_url_concat_no_paramsz'TestUrlConcat.test_url_concat_no_params>   s     92>>?r   c                 @    t        dd       }| j                  |d       y r5   r#   r%   s     r   test_url_concat_none_paramsz)TestUrlConcat.test_url_concat_none_paramsB   s     94@>?r   c                 B    t        ddg      }| j                  |d       y )Nzhttps://localhost/path#tabr   zhttps://localhost/path?y=y#tabr#   r%   s     r   test_url_concat_with_fragz'TestUrlConcat.test_url_concat_with_fragF   s"    5
|D>?r   c                 D    t        dddg      }| j                  |d       y )Nr   )r   y1)r   y2z https://localhost/path?y=y1&y=y2r#   r%   s     r   !test_url_concat_multi_same_paramsz/TestUrlConcat.test_url_concat_multi_same_paramsJ   s%    1K3MN@Ar   c                 B    t        ddg      }| j                  |d       y )Nzhttps://localhost/path?r=1&r=2r   z"https://localhost/path?r=1&r=2&y=yr#   r%   s     r   'test_url_concat_multi_same_query_paramsz5TestUrlConcat.test_url_concat_multi_same_query_paramsN   s"    9J<HBCr   c                 T    t        dt        d            }| j                  |d       y )Nr   r   )r   zhttps://localhost/path?y=y)r   dictr$   r%   s     r   test_url_concat_dict_paramsz)TestUrlConcat.test_url_concat_dict_paramsR   s$    14#;?:;r   N)__name__
__module____qualname__r(   r*   r,   r/   r1   r3   r6   r8   r:   r>   r@   rC   r   r   r   r   r   %   sI    @C@CCH@@@BD<r   r   c                       e Zd Zd Zy)QsParseTestc                     d}t         j                  j                  |      }t        t	        |            }| j                  d|       | j                  d|       | j                  d|       y )Nza=1&b=2&a=3)a1)rJ   3)b2)urllibparseparse_qslistr
   assertIn)r&   qsstringqsqsls       r   test_parsingzQsParseTest.test_parsingX   sS     \\""8,9R=!j#&j#&j#&r   N)rD   rE   rF   rW   r   r   r   rH   rH   W   s    'r   rH   c                   H    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zy)MultipartFormDataTestc                     dj                  dd      }t               \  }}t        d|||       |d   d   }| j                  |d   d       | j                  |d	   d
       y )NT   --1234
Content-Disposition: form-data; name="files"; filename="ab.txt"

Foo
--1234--   
   
   1234filesr   filenameab.txtbody   Fooreplacer   r   r$   r&   dataargsr_   files        r   test_file_uploadz&MultipartFormDataTest.test_file_uploadb   sp    
 G7
 	 %&e!'4u=W~a j)84fv.r   c                     dj                  dd      }t               \  }}t        d|||       |d   d   }| j                  |d   d       | j                  |d	   d
       y )NsP   --1234
Content-Disposition: form-data; name=files; filename=ab.txt

Foo
--1234--r\   r]   r^   r_   r   r`   ra   rb   rc   rd   rf   s        r   test_unquoted_namesz)MultipartFormDataTest.test_unquoted_namesq   sp    
 G7
 	 %&e!'4u=W~a j)84fv.r   c                 `   g d}|D ]  }t        j                  d|       d|j                  dd      j                  dd      z  }t        |j                  dd	            }t	               \  }}t        d
|||       |d   d   }| j                  |d   |       | j                  |d   d        y )N)za;b.txtza"b.txtza";b.txtza;"b.txtz	a";";.txtza\"b.txtza\b.txtztrying filename %rzP--1234
Content-Disposition: form-data; name="files"; filename="%s"

Foo
--1234--\\\"z\"

r^   r_   r   r`   rb   rc   )loggingdebugre   r   r   r   r$   )r&   	filenamesr`   str_datarg   rh   r_   ri   s           r   test_special_filenamesz,MultipartFormDataTest.test_special_filenames   s    
	 " 	3HMM.9
 fgU	H ((v67D(*KD%%gtT5A>!$DT*-x8T&\62#	3r   c                     dj                  dd      }t               \  }}t        d|||       |d   d   }| j                  |d   d       | j                  |d	   d
       y )Nsr   --1234
Content-Disposition: form-data; name="files"; filename="ab.txt"; filename*=UTF-8''%C3%A1b.txt

Foo
--1234--r\   r]   r^   r_   r   r`   u   áb.txtrb   rc   rd   rf   s        r   test_non_ascii_filenamez-MultipartFormDataTest.test_non_ascii_filename   sp    
 G7
 	 %&e!'4u=W~a j)95fv.r   c                     dj                  dd      }t               \  }}t        d|||       |d   d   }| j                  |d   d       | j                  |d	   d
       y )Nr[   r\   r]   s   "1234"r_   r   r`   ra   rb   rc   rd   rf   s        r   )test_boundary_starts_and_ends_with_quotesz?MultipartFormDataTest.test_boundary_starts_and_ends_with_quotes   sp    
 G7
 	 %&e!)T4?W~a j)84fv.r   c                     dj                  dd      }t               \  }}| j                  t        d      5  t	        d|||       d d d        | j                  |i        y # 1 sw Y   xY w)Ns   --1234

Foo
--1234--r\   r]   z#multipart/form-data missing headersmsgr^   re   r   assertRaisesr   r   r$   r&   rg   rh   r_   s       r   test_missing_headersz*MultipartFormDataTest.test_missing_headers   s     G7
	 	 %&e E  
 	B &gtT5A	B 	#		B 	B   A!!A*c                     dj                  dd      }t               \  }}| j                  t        d      5  t	        d|||       d d d        | j                  |i        y # 1 sw Y   xY w)NsR   --1234
Content-Disposition: invalid; name="files"; filename="ab.txt"

Foo
--1234--r\   r]   Invalid multipart/form-datar}   r^   r   r   s       r    test_invalid_content_dispositionz6MultipartFormDataTest.test_invalid_content_disposition   sv    
 G7
 	 %&e~3PQ 	B%gtT5A	B#	B 	Br   c                     dj                  dd      }t               \  }}| j                  t        d      5  t	        d|||       d d d        | j                  |i        y # 1 sw Y   xY w)NsS   --1234
Content-Disposition: form-data; name="files"; filename="ab.txt"

Foo--1234--r\   r]   r   r}   r^   r   r   s       r   .test_line_does_not_end_with_correct_line_breakzDMultipartFormDataTest.test_line_does_not_end_with_correct_line_break   sv     w7
	 	 %&e~3PQ 	B%gtT5A	B#	B 	Br   c                     dj                  dd      }t               \  }}| j                  t        d      5  t	        d|||       d d d        | j                  |i        y # 1 sw Y   xY w)NsF   --1234
Content-Disposition: form-data; filename="ab.txt"

Foo
--1234--r\   r]   z&multipart/form-data value missing namer}   r^   r   r   s       r   6test_content_disposition_header_without_name_parameterzLMultipartFormDataTest.test_content_disposition_header_without_name_parameter   s    
 G7
 	 %&e H  
 	B &gtT5A	B 	#		B 	Br   c                     dj                  dd      }t               \  }}t        d|||       |d   d   }| j                  |d   d       | j                  |d	   d
       y )NsU   --1234
Content-Disposition: form-data; name="files"; filename="ab.txt"

Foo
--1234--
r\   r]   r^   r_   r   r`   ra   rb   rc   rd   rf   s        r   test_data_after_final_boundaryz4MultipartFormDataTest.test_data_after_final_boundary   sp     G7
 	 %&e!'4u=W~a j)84fv.r   N)rD   rE   rF   rj   rl   rw   ry   r{   r   r   r   r   r   r   r   r   rY   rY   a   s4    // 3://$$$$ /r   rY   c                   <    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
y	)
HTTPHeadersTestc                    dj                  dd      }t        j                  |      }| j                  |d   d       | j                  |j	                  d      dg       | j                  |d   d       | j                  |j	                  d      d	d
g       | j                  t        t        |j                                     g d       y )Nz>Foo: bar
 baz
Asdf: qwer
	zxcv
Foo: even
     more
     lines
rq   rr   asdf	qwer zxcvFoozbar baz,even more linesfoobar bazeven more lines))Asdfr   )r   r   )r   r   )re   r   rP   r$   get_listsortedrR   get_all)r&   rg   headerss      r   test_multi_linezHTTPHeadersTest.test_multi_line  s    
 G&
 	 ##D)+6))&1K=A)BC))%09>O2PQ4)*+S	
r   c                 R    d}| j                  t        t        j                  |       y )Nz	 Foo: bar)r   r   r   rP   )r&   rg   s     r   test_malformed_continuationz+HTTPHeadersTest.test_malformed_continuation%  s!     .+*;*;TBr   c           	         g d}|D ]  }dD ]  }	 	 |j                  |      }d|z   dz   }t        j                  t	        |j                  d                  }ddt	        |j                  d            z   dz   fg}| j                  |t        |j                                        y # t        $ r Y w xY w# t        $ r t        j                  d	||        w xY w)
N)   u    u    )r   latin1s   Cookie: foo=s   barr   Cookiezfoo=barzfailed while trying %r in %s)encodeUnicodeEncodeErrorr   rP   r   decoder$   rR   r   	Exceptionr   warning)r&   newlinesnewlineencodingencodedrg   r   expecteds           r   test_unicode_newlinesz%HTTPHeadersTest.test_unicode_newlines,  s    
   	G. !")..": +W4v=D *//
4;;x;P0QRG %"Zx0H%IIEQ H $$XtGOO4E/FG#	 . ! ! ! OO$BGXVs)   B(BB7(	B41B73B44B77"Cc                     t        j                  d      }| j                  t        |j	                               g d       y )Nz&CRLF: crlf
LF: lf
CR: crMore: more
))CrzcrMore: more)Crlfcrlf)Lflf)r   rP   r$   r   r   r&   r   s     r   test_optional_crz HTTPHeadersTest.test_optional_crT  s5     ##$RS7??$%F	
r   c           	         g d}t               }|D ]  \  }}|j                  ||        |j                         }t        j                  |      }t        j                  |      }||||fD ]4  }| j	                  t        t        |j                                     |       6 |||fD ]D  }| j                  ||       | j                  |j                  d      |j                  d             F y )N))ArK   )r   rN   )Bcr   )
r   addcopydeepcopyr$   rR   r   r   assertIsNotr   )	r&   	all_pairsh1kvh2h3h4r   s	            r   	test_copyzHTTPHeadersTest.test_copy^  s    8	] 	DAqFF1aL	WWYYYr]]]2BB' 	IG T&):";<iH	I B| 	FGWb)W--c2BKK4DE	Fr   c                    t               }|j                  dd       |j                  dd       |j                  dd       t        j                  |      }t        j                  |      }| j                  t        |j                               t        |j                                      | j                  t        |j                               t        |j                                      y )Nz
Set-Cookieza=bzc=dzContent-Typez	text/html)	r   r   pickledumpsloadsr$   r   r   items)r&   r   pickled	unpickleds       r   test_pickle_roundtripz%HTTPHeadersTest.test_pickle_roundtripo  s    -L%(L%(NK0,,w'LL)	 12F9;L;L;N4OP0&9J2KLr   c                 T   t               }d|d<   | j                  |j                  dd      d       | j                  |d   d       | j                  |j                  dd      d       | j                  |d   d       | j                  t        |j	                               ddg       y )Nr   r   bazquuxxyzzy)r   r   )Quuxr   )r   r$   
setdefaultr   r   r   s     r   test_setdefaultzHTTPHeadersTest.test_setdefaulty  s    -++E595A/++FG<gF'2 12^EV4WXr   c                     t               }|j                  dd       |j                  dd       |j                  dd       t        j                  t        |            }| j	                  ||       y )Nr   rK   rN   rL   )r   r   rP   strr$   )r&   r   headers2s      r   test_stringzHTTPHeadersTest.test_string  sW    -E3E3E3$$S\2(+r   N)rD   rE   rF   r   r   r   r   r   r   r   r   r   r   r   r   r   
  s/    
4C&P
F"M	Y,r   r   c                   J    e Zd ZdZdZd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zy)FormatTimestampTestg*; R]AAzSun, 27 Jan 2013 18:43:20 GMTc                 N    | j                  t        |      | j                         y N)r$   r   EXPECTED)r&   values     r   checkzFormatTimestampTest.check  s    )%0$--@r   c                 :    | j                  | j                         y r   )r   	TIMESTAMPr&   s    r   test_unix_time_floatz(FormatTimestampTest.test_unix_time_float  s    

4>>"r   c                 L    | j                  t        | j                               y r   )r   intr   r   s    r   test_unix_time_intz&FormatTimestampTest.test_unix_time_int  s    

3t~~&'r   c                 `    | j                  t        j                  | j                               y r   )r   timegmtimer   r   s    r   test_struct_timez$FormatTimestampTest.test_struct_time  s    

4;;t~~./r   c                     t        t        j                  | j                              }| j	                  dt        |             | j                  |       y )N	   )tupler   r   r   r$   lenr   )r&   tups     r   test_time_tuplez#FormatTimestampTest.test_time_tuple  s8    DKK/0CH%

3r   c                     | j                  t        j                  j                  | j                  t        j                  j
                        j                  d              y )N)tzinfo)r   datetimefromtimestampr   timezoneutcre   r   s    r   test_utc_naive_datetimez+FormatTimestampTest.test_utc_naive_datetime  sC    

++ 1 1 5 5gTg"	
r   c                     t               5  | j                  t        j                  j                  | j                               d d d        y # 1 sw Y   y xY wr   )r   r   r   utcfromtimestampr   r   s    r   "test_utc_naive_datetime_deprecatedz6FormatTimestampTest.test_utc_naive_datetime_deprecated  sB    ! 	KJJx((99$..IJ	K 	K 	Ks   9AAc                     | j                  t        j                  j                  | j                  t        j                  j
                               y r   )r   r   r   r   r   r   r   s    r   test_utc_aware_datetimez+FormatTimestampTest.test_utc_aware_datetime  s4    

++DNNH<M<M<Q<QR	
r   c           
          | j                  t        j                  j                  | j                  t        j                  t        j
                  d                         y )N)hours)r   r   r   r   r   	timedeltar   s    r   test_other_aware_datetimez-FormatTimestampTest.test_other_aware_datetime  sA    

++ 1 1(2D2D22N O	
r   N)rD   rE   rF   r   r   r   r   r   r   r   r   r   r   r  r   r   r   r   r     s;    !I.HA#(0

K


r   r   c                       e Zd Zd Zd Zd Zy)HTTPServerRequestTestc                     t        d       y N/)uri)r   r   s    r   test_default_constructorz.HTTPServerRequestTest.test_default_constructor  s     	c"r   c                 \    t        d      }| j                  |j                  t               y r	  )r   assertIsInstancerb   bytes)r&   requetss     r   test_body_is_a_byte_stringz0HTTPServerRequestTest.test_body_is_a_byte_string  s     #,gllE2r   c                 n    t        dt        ddgi            }| j                  dt        |      v       y )Nr
  Canaryz	Coal Mine)r  r   )r   r   
assertTruerepr)r&   requests     r   "test_repr_does_not_contain_headersz8HTTPServerRequestTest.test_repr_does_not_contain_headers  s4    #[([M)BC
 	W56r   N)rD   rE   rF   r  r  r  r   r   r   r  r    s    #37r   r  c                       e Zd ZdZdZdZd Zy)ParseRequestStartLineTestGETz/foozHTTP/1.1c                 b   dj                  | j                  | j                  | j                  g      }t	        |      }| j                  |j                  | j                         | j                  |j                  | j                         | j                  |j                  | j                         y )N )	joinMETHODPATHVERSIONr   r$   methodpathversion)r&   
start_lineparsed_start_lines      r   test_parse_request_start_linez7ParseRequestStartLineTest.test_parse_request_start_line  s    XXt{{DIIt||DE
4Z@*114;;?*//;*22DLLAr   N)rD   rE   rF   r  r  r   r&  r   r   r   r  r    s    FDGBr   r  c                   *    e Zd Zd Zd Zd Zd Zd Zy)ParseCookieTestc                 4   | j                  t        d      ddd       | j                  t        d      dddd	d
       | j                  t        d      ddi       | j                  t        d      ddi       | j                  t        d      ddddd       y)zO
        Test cases copied from Python's Lib/test/test_http_cookies.py
        zchips=ahoy; vienna=fingerahoyfinger)chipsviennaz)keebler="E=mc2; L=\"Loves\"; fudge=\012;"z"E=mc2z	\"Loves\"z\012rp   )keeblerLfudge zkeebler=E=mc2r.  zE=mc2zkey:term=value:termzkey:termz
value:termza=b; c=[; d=r; f=hrM   [rh)rJ   r   dfNr$   r	   r   s    r   test_python_cookiesz#ParseCookieTest.test_python_cookies  s     	451	
 	GH }wCP	

 	o6G8LM./*l1K	
 	-.c#TW0X	
r   c                 |    | j                  t        d      ddd       | j                  t        d      ddd       y )	Nza=b; Domain=example.comrM   zexample.com)rJ   Domainza=b; h=i; a=cr   i)rJ   r4  r7  r   s    r   test_cookie_edgecasesz%ParseCookieTest.test_cookie_edgecases  s<    233-5X	

 	o6c8LMr   c                    | j                  dt        d      j                                | j                  t        d      dddd       | j                  t        d      d	d
d       | j                  t        d      ddi       | j                  t        d      dt	        d      i       | j                  t        d      ddd       y)zs
        Cookie strings that go against RFC6265 but browsers will send if set
        via document.cookie.
        django_languagez$abc=def; unnamed; django_language=enza=b; "; c=drM   rp   r5  )rJ   r1  r   za b c=d e = f; gh=izd e = fr;  )za b cghza   b,c<>@:/[]?{}=d  "  =e,f gza   b,c<>@:/[]?{}zd  "  =e,f gu   saint=André Bessettesaintu   André Bessettez  =  b  ;  ;  =  ;   c  =  ;  r1  )r1  r   N)rS   r	   keysr$   r   r   s    r   test_invalid_cookiesz$ParseCookieTest.test_invalid_cookies  s     	?@EEG	

 	m4CSs6ST./9C1P	
 	9: .1	

 	01j!234	
 	9:2<N	
r   c                     g d}|D ]?  \  }}| j                  |      5  t        |      }| j                  |d   |       d d d        A y # 1 sw Y   LxY w)N))za="b=\""b=")za="b=\\"b=\)za="b=\="zb==)za="b=\n"zb=n)z
a="b=\042"rD  )z
a="b=\134"rE  )z
a="b=\377"u   b=ÿ)z
a="b=\400"zb=400)z	a="b=\42"zb=42)za="b=\\042"zb=\042)za="b=\\134"zb=\134)z
a="b=\\\""b=\")za="b=\\\042"rF  )za="b=\134\""rF  )za="b=\134\042"rF  rJ   )subTestr	   r$   )r&   casesr   decodedr   s        r   test_unquotezParseCookieTest.test_unquote$  sc    
" !& 	2GWg& 2 )  312 2	22 2s   !A

A	c                    d}dD ]  }| j                  |      5  t        j                         }d||z  z   dz   }t        |      d   }t        j                         }| j                  |d d d       | j                  |dd  d	       | j                  t	        |      |d
z          | j                  ||z
  dd       d d d         y # 1 sw Y   xY w)Ni )ro   z\134za="b=rp   rJ      rE  z\\\      zTest took too long)rG  r   r	   r$   r   
assertLess)r&   nr   startrg   r   ends          r   test_unquote_largez"ParseCookieTest.test_unquote_large=  s     % 	FGg& 
F		1,s2$T*3/iik  rF3  rsX6  UQU3 eQ0DE
F 
F	F
F 
Fs   B CC	N)rD   rE   rF   r8  r<  rB  rJ  rT  r   r   r   r(  r(    s    
2N 
D22Fr   r(  )*tornado.httputilr   r   r   r   r   r   r	   r
   r   r   tornado.escaper   r   tornado.logr   tornado.test.utilr   r   r   rs   r   r   urllib.parserO   unittesttypingr   r   r   r   r  r   TestCaser   rH   rY   r   r   r  r  r(  r   r   r   <module>r]     s
     ,  0        $ $d3U#34d3X;N6OOP /<H%% /<d'(## 'f/H-- f/R@,h'' @,F,
(++ ,
b7H-- 7$
B 1 1 
BsFh'' sFr   