
    gkX                     :   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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jF                        Z$ G d dejF                        Z% G d dejF                        Z& G d dejF                        Z' G d dejF                        Z( G d dejF                        Z) G d dejF                        Z* G d dejF                        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)	ExpectLog)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       O/var/www/openai/venv/lib/python3.12/site-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_amp3   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_amp7   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 r6   r$   r&   s     r   test_url_concat_none_paramsz)TestUrlConcat.test_url_concat_none_paramsC   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_fragG   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_paramsK   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_paramsO   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_paramsS   s$    14#;?:;r   N)__name__
__module____qualname__r)   r+   r-   r0   r2   r4   r7   r9   r;   r?   rA   rD   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)rK   3)b2)urllibparseparse_qslistr
   assertIn)r'   qsstringqsqsls       r   test_parsingzQsParseTest.test_parsingY   sS     \\""8,9R=!j#&j#&j#&r   N)rE   rF   rG   rX   r   r   r   rI   rI   X   s    'r   rI   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_uploadc   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   ra   rb   rc   rd   re   rg   s        r   test_unquoted_namesz)MultipartFormDataTest.test_unquoted_namesr   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   ra   rc   rd   )loggingdebugrf   r   r   r   r%   )r'   	filenamesra   str_datarh   ri   r`   rj   s           r   test_special_filenamesz,MultipartFormDataTest.test_special_filenames   s    
	 "HMM.9
 fgU	H ((v67D(*KD%%gtT5A>!$DT*-x8T&\62# "r   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   ra   u   áb.txtrc   rd   re   rg   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   ra   rb   rc   rd   re   rg   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               \  }}t        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 headersr_   rf   r   r   r   r   r%   r'   rh   ri   r`   s       r   test_missing_headersz*MultipartFormDataTest.test_missing_headers   sd     G7
	 	 %&ew EF%gtT5A G# GF   AA#c                     dj                  dd      }t               \  }}t        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   s       r    test_invalid_content_dispositionz6MultipartFormDataTest.test_invalid_content_disposition   sb    
 G7
 	 %&ew =>%gtT5A ?# ?>r   c                     dj                  dd      }t               \  }}t        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   s       r   .test_line_does_not_end_with_correct_line_breakzDMultipartFormDataTest.test_line_does_not_end_with_correct_line_break   sb     w7
	 	 %&ew =>%gtT5A ?# ?>r   c                     dj                  dd      }t               \  }}t        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   s       r   6test_content_disposition_header_without_name_parameterzLMultipartFormDataTest.test_content_disposition_header_without_name_parameter   sd    
 G7
 	 %&ew HI%gtT5A J# JIr   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   ra   rb   rc   rd   re   rg   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)rE   rF   rG   rk   rm   rx   rz   r|   r   r   r   r   r   r   r   r   rZ   rZ   b   s4    // 3://$$$$/r   rZ   c                   B    e Zd 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
rr   rs   asdf	qwer zxcvFoozbar baz,even more linesfoobar bazeven more lines))Asdfr   )r   r   )r   r   )rf   r   rQ   r%   get_listsortedrS   get_all)r'   rh   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)assertRaisesr   r   rQ   )r'   rh   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   rQ   r   decoder%   rS   r   	Exceptionr   warning)r'   newlinesnewlineencodingencodedrh   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                     g d}|D ]T  \  }}t        j                  d|z        }|rdg}ndd|z  fg}| j                  |t        |j	                                      V y )N)) T)	T)    F)r   FzTransfer-Encoding: %schunked)Transfer-Encodingchunkedr   z	%schunked)r   rQ   r%   rS   r   )r'   
whitespacecstrippedr   r   s         r   test_unicode_whitespacez'HTTPHeadersTest.test_unicode_whitespaceQ  sg    


 &KAx!''(F(JKG<=0+/BCXtGOO,='>? &r   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   rQ   r%   r   r   r'   r   s     r   test_optional_crz HTTPHeadersTest.test_optional_crd  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))ArL   )r   rO   )Br   r   )
r   addcopydeepcopyr%   rS   r   r   assertIsNotr   )	r'   	all_pairsh1kvh2h3h4r   s	            r   	test_copyzHTTPHeadersTest.test_copyn  s    8	]DAqFF1aL WWYYYr]]]2BB'G T&):";<iH ( B|GWb)W--c2BKK4DE $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_roundtrip  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_setdefault  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   rL   rO   rM   )r   r   rQ   strr%   )r'   r   headers2s      r   test_stringzHTTPHeadersTest.test_string  sW    -E3E3E3$$S\2(+r   N)rE   rF   rG   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s5    
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utcrf   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  s5    !JJx((99$..IJ "!!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)rE   rF   rG   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   assertIsInstancerc   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)rE   rF   rG   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 )Nr   )	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)rE   rF   rG   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\012rq   )keeblerLfudge zkeebler=E=mc2r5  zE=mc2zkey:term=value:termzkey:termz
value:termza=b; c=[; d=r; f=hrN   [rh)rK   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.comrN   zexample.com)rK   Domainza=b; h=i; a=cr   i)rK   r;  r>  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=drN   rq   r<  )rK   r8  r   za b c=d e = f; gh=izd e = frB  )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  =  ;  r8  )r8  r   N)rT   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"rK  )z
a="b=\134"rL  )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"rM  )za="b=\134\""rM  )za="b=\134\042"rM  rK   )subTestr	   r%   )r'   casesr   decodedr   s        r   test_unquotezParseCookieTest.test_unquote4  sS    
" !&GWg& )  31 '& !&&&s   !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 )rp   z\134za="b=rq   rK      rL  z\\\      zTest took too long)rN  r   r	   r%   r   
assertLess)r'   nr   startrh   r   ends          r   test_unquote_largez"ParseCookieTest.test_unquote_largeM  s     %Gg&		1,s2$T*3/iik  rF3  rsX6  UQU3 eQ0DE '& &&&s   B CC	N)rE   rF   rG   r?  rC  rI  rQ  r[  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.testingr   tornado.test.utilr   r   r   rt   r   r   urllib.parserP   unittesttypingr   r   r   r   r  r   TestCaser   rI   rZ   r   r   r  r!  r/  r   r   r   <module>re     s     ,  % 0        $ $d3U#34d3X;N6OOP /<H%% /<d'(## 'b/H-- b/JS,h'' S,l,
(++ ,
b7H-- 7$
B 1 1 
BsFh'' sFr   