File: //usr/local/lib/python3.7/test/__pycache__/test_http_cookiejar.cpython-37.pyc
B
��g�. � @ sB d Z ddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! G dd� dej"�Z#G dd� dej"�Z$G dd � d �Z%d
d � Z&d d
� Z'dd� Z(G dd� dej"�Z)G dd� dej"�Z*G dd� dej"�Z+ddd�Z,e-dk�r>e,dd� dS ) z Tests for http/cookiejar.py.� N)� time2isoz� http2time�iso2time�
time2netscape�parse_ns_headers�join_header_words�split_header_words�Cookie� CookieJar�DefaultCookiePolicy� LWPCookieJar�MozillaCookieJar� LoadError�lwp_cookie_str�DEFAULT_HTTP_PORT� escape_path�reach�is_HDN� domain_match�user_domain_match� request_path� request_port� request_hostc @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�
DateTimeTestsc C s� d}d}| � t|�d� | � t|| �d� | � t|d| �d� | � t|d| �d� t� }td �}x&||fD ]}| �|d
d ||f � qtW d S ) Nix+�<i�Q z2002-04-19 14:36:40Zz2002-04-20 14:36:40Z� z2002-04-21 14:36:40Z� z2002-04-22 14:36:40Zi � z!^\d{4}-\d\d-\d\d \d\d:\d\d:\d\dZ$zbad time2isoz format: %s %s)� assertEqualr � assertRegex)�self�base�day�az�bz�text� r$ �4/usr/local/lib/python3.7/test/test_http_cookiejar.py�test_time2isoz s z DateTimeTests.test_time2isozc C s� d}d}| � t|�d� | � t|| �d� | � t|d| �d� | � t|d| �d� t� }td �}x&||fD ]}| �|d
d ||f � qtW d S ) Nix+�<i�Q z Fri, 19-Apr-2002 14:36:40 GMTz Sat, 20-Apr-2002 14:36:40 GMTr z Sun, 21-Apr-2002 14:36:40 GMTr z Mon, 22-Apr-2002 14:36:40 GMTi � z;[a-zA-Z]{3}, \d{2}-[a-zA-Z]{3}-\d{4} \d{2}:\d{2}:\d{2} GMT$z bad time2netscape format: %s %s)r r r )r r r r! r" r# r$ r$ r% �test_time2netscape# s z DateTimeTests.test_time2netscapec C s< dd� }| � |d�d� | � |d�d� | � |d�d� d S ) Nc S s t �t| ��d d� S )N� )�time�gmtimer )r# r$ r$ r% �
parse_date: s z0DateTimeTests.test_http2time.<locals>.parse_datez 01 Jan 2001)i� � r, r r g z 03-Feb-20)i� r r r r g z 03-Feb-98)i� r r r r g )r )r r+ r$ r$ r% �test_http2time9 s z DateTimeTests.test_http2timec C s� ddddddddd d
d d d
dg}d}t |�}d}| �||d|||f � xR|D ]J}| �t|�||� | �t|�� �||�� � | �t|�� �||�� � qNW d S )Nz Thu, 03 Feb 1994 00:00:00 GMTz Thursday, 03-Feb-94 00:00:00 GMTz"Thursday, 03-Feb-1994 00:00:00 GMTz03 Feb 1994 00:00:00 GMTz03-Feb-94 00:00:00 GMTz03-Feb-1994 00:00:00 GMTz03-Feb-1994 00:00 GMTz03-Feb-1994 00:00z02-Feb-1994 24:00z 03-Feb-94z 03-Feb-1994z 03 Feb 1994z 03 Feb 1994 0:00 z 03-Feb-1994 i�>P-z1994-02-03 00:00:00Zz%s => '%s' (%s))r r r �lower�upper)r �tests�test_t�result�expected�sr$ r$ r% �test_http2time_formatsE s.
z$DateTimeTests.test_http2time_formatsc C s. x(dD ] }| � t|�d|t|�f � qW d S )N)� �GarbagezMandag 16. September 1996z
01-00-1980z
01-13-1980z
00-01-1980z
32-01-1980z01-01-1980 25:00:00z01-01-1980 00:61:00z01-01-1980 00:00:62z08-Oct-3697739z
08-01-3697739z 09 Feb 19942632 22:23:32 GMTz Wed, 09 Feb 1994834 22:23:32 GMTz,http2time(%s) is not None
http2time(test) %s)� assertIsNoner )r �testr$ r$ r% �test_http2time_garbagei s
z$DateTimeTests.test_http2time_garbagec C s( t d�dd �� t d�dd �� d S )Nz01 Jan 1970{}00:00:00 GMT!� i�� z01 Jan 1970 00:00:00{}GMT!)r �format)r r$ r$ r% �2test_http2time_redos_regression_actually_completes~ s z@DateTimeTests.test_http2time_redos_regression_actually_completesc C s< dd� }| � |d�d� | � |d�d� | � |d�d� d S )Nc S s t �t| ��d d� S )Nr( )r) r* r )r# r$ r$ r% r+ � s z/DateTimeTests.test_iso2time.<locals>.parse_dateZ19940203T141529Z)i� r r � � � z1994-02-03 07:15:29 -0700z1994-02-03 19:45:29 +0530)r )r r+ r$ r$ r% �
test_iso2time� s
zDateTimeTests.test_iso2timec
C st ddddddddd d
g
}d }xR|D ]J}| � t|�||� | � t|�� �||�� � | � t|�� �||�� � q"W d S ) Nz1994-02-03 00:00:00 -0000z1994-02-03 00:00:00 +0000z1994-02-03 00:00:00z
1994-02-03z1994-02-03T00:00:00Z19940203z1994-02-02 24:00:00Z19940203T000000Zz
1994-02-03 z 1994-02-03T00:00:00 i�>P-)r r r. r/ )r r0 r1 r4 r$ r$ r% �test_iso2time_formats� s
z#DateTimeTests.test_iso2time_formatsc C s& x dD ]}| � t|�d| � qW d S )N) r6 r7 z Thursday, 03-Feb-94 00:00:00 GMTz
1980-00-01z
1980-13-01z
1980-01-00z
1980-01-32z1980-01-01 25:00:00z1980-01-01 00:61:00z01-01-1980 00:00:62z01-01-1980T00:00:62Z19800101T250000Zz iso2time(%r))r8 r )r r9 r$ r$ r% �test_iso2time_garbage� s
z#DateTimeTests.test_iso2time_garbagec C s( t d�dd �� t d�dd �� d S )Nz1994-02-03{}14:15:29 -0100!r; i@B z1994-02-03 14:15:29{}-0100!)r r< )r r$ r$ r% �$test_iso2time_performance_regression� s z2DateTimeTests.test_iso2time_performance_regressionN)
�__name__�
__module__� __qualname__r& r' r- r5 r: r= rA rB rC rD r$ r$ r$ r% r s $r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� HeaderTestsc C s0 dddgg}x dD ]}| � t|g�|� qW d S )N)�foo�bar)�expiresl d9W )�version�0)z)foo=bar; expires=01 Jan 2040 22:23:32 GMTz+foo=bar; expires="01 Jan 2040 22:23:32 GMT")r r )r r3 �hdrr$ r$ r% �test_parse_ns_headers� s
z!HeaderTests.test_parse_ns_headersc C s. ddgg}x dD ]}| � t|g�|� qW d S )N)rI rJ )rL �1)zfoo=bar; version="1"zfoo=bar; Version="1")r r )r r3 rN r$ r$ r% � test_parse_ns_headers_version� s
z)HeaderTests.test_parse_ns_headers_versionc C s$ d}ddgg}| � t|g�|� d S )Nz expires=01 Jan 2040 22:23:32 GMT)rK z01 Jan 2040 22:23:32 GMT)rL rM )r r )r rN r3 r$ r$ r% �#test_parse_ns_headers_special_names� s
z/HeaderTests.test_parse_ns_headers_special_namesc C s0 t ddgg�}| �|d� | �t g g�d� d S )N)rI N)rJ �bazz foo; bar=bazr6 )r r )r Zjoinedr$ r$ r% �test_join_header_words� s z"HeaderTests.test_join_header_wordsc C s ddggfddggfddggfddggfddggfd dggfd
dd ggfd dd ggfd
dggfddd ggfddddggfddgdgdggfddd gdgdgdggfg
}xt|D ]l\}}yt |g�}W n< dd l}dd l}|�� }|�d |� d |�� }Y nX | �||d |||f � q�W d S ) NrI )rI Nzfoo=bar)rI rJ z foo z
foo= )rI r6 z foo=z foo= ; z foo= ; bar= baz )rJ rS zfoo=bar bar=bazz foo= bar=baz)rI zbar=bazzfoo=bar;bar=bazz foo bar baz)rJ N)rS Nza, b, c)�aN)�bN)�cNz(foo; bar=baz, spam=, foo="\,\;\"", bar= )�spamr6 )rI z,;")rJ r6 r z (error -- traceback follows)
%sz:
When parsing: '%s'
Expected: '%s'
Got: '%s'
)r � traceback�io�StringIO� print_exc�getvaluer )r r0 �arg�expectr2 rY rZ �fr$ r$ r% �test_split_header_words� s2
z#HeaderTests.test_split_header_wordsc
C s` ddddddddd d
d d d
g
}x<|D ]4\}}t |g�}t|�}| �||d||||f � q$W d S )N)rI rI )zfoo=barzfoo=bar)z foo rI )zfoo=zfoo="")zfoo=bar bar=bazzfoo=bar; bar=baz)zfoo=bar;bar=bazzfoo=bar; bar=baz)z foo bar bazz
foo; bar; baz)zfoo="\"" bar="\\"zfoo="\""; bar="\\")z foo,,,barzfoo, bar)zfoo=bar,bar=bazzfoo=bar, bar=baz)z text/html; charset=iso-8859-1z text/html; charset="iso-8859-1")z)foo="bar"; port="80,81"; discard, bar=bazz'foo=bar; port="80,81"; discard, bar=baz)z Basic realm="\"foo\\\\bar\""z Basic; realm="\"foo\\\\bar\""zM
When parsing: '%s'
Expected: '%s'
Got: '%s'
Input was: '%s'
)r r r )r r0 r^ r_ �input�resr$ r$ r% �test_roundtrip
s$
zHeaderTests.test_roundtripN) rE rF rG rO rQ rR rT ra rd r$ r$ r$ r% rH � s
"rH c @ s" e Zd Zg dfdd�Zdd� ZdS )� FakeResponseNc C s$ ddl }|�d�|��| _|| _dS )zD
headers: list of RFC822-style 'Key: value' strings
r N�
)�emailZmessage_from_string�join�_headersZ_url)r �headers�urlrg r$ r$ r% �__init__- s zFakeResponse.__init__c C s | j S )N)ri )r r$ r$ r% �info4 � zFakeResponse.info)rE rF rG rl rm r$ r$ r$ r% re , s re c G s t | ||d�S )Nz Set-Cookie2)� _interact)� cookiejarrk �set_cookie_hdrsr$ r$ r% �
interact_29656 s rr c G s t | ||d�S )Nz
Set-Cookie)ro )rp rk rq r$ r$ r% �interact_netscape9 s rs c C s` t j�|�}| �|� |�dd�}g }x |D ]}|�d||f � q,W t||�}| �||� |S )zDPerform a single request / response cycle, returning Cookie: header.r r6 z%s: %s)�urllib�request�Request�add_cookie_header�
get_header�appendre �extract_cookies) rp rk rq Zhdr_name�reqZ
cookie_hdrrj rN rc r$ r$ r% ro < s
ro c @ s e Zd Zdd� Zdd� ZdS )�FileCookieJarTestsc C s� t jj}t� }t|dd� | �|jd d d jd � z&|j|dd� t� }|j |dd� W d yt
� |� W n t k
r� Y nX X | �|jd d d jd � d S )Nzhttp://www.acme.com/Zbooz www.acme.com�/T)�ignore_discard)
r9 �support�TESTFNr rs r �_cookies�value�save�load�os�unlink�OSError)r �filenamerW r$ r$ r% �test_lwp_valueless_cookieJ s z,FileCookieJarTests.test_lwp_valueless_cookiec C s� t jj}xbttfD ]V}|� }y|jdd� W n0 tk
r\ } z| �|jt � W d d }~X Y qX | �
d� qW zLt |d��8}|� d� x&ttfD ]}|� }| �
t |j|� q�W W d Q R X W d yt�|� W n tk
r� Y nX X d S )NzAfor this test to work, a file with this filename should not exist)r� z%expected OSError for invalid filename�wzoops
)r9 r r� r r
r� r� Z assertIsNot� __class__r Zfail�open�writeZ assertRaisesr� r� )r r� Zcookiejar_classrW �excr` r$ r$ r% �test_bad_magicY s$
" z!FileCookieJarTests.test_bad_magicN)rE rF rG r� r� r$ r$ r$ r% r| I s r| c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAS )B� CookieTestsc C sP t � }xDdD ]<\}}}tj�|�}|�||�}|r>| �|� q | �|� q W d S )N))zhttp://foo.bar.com/zblah.comF)zhttp://foo.bar.com/zrhubarb.blah.comF)zhttp://foo.bar.com/zrhubarb.foo.bar.comF)zhttp://foo.bar.com/z .foo.bar.comT)zhttp://foo.bar.com/z foo.bar.comT)zhttp://foo.bar.com/z.bar.comT)zhttp://foo.bar.com/zbar.comT)zhttp://foo.bar.com/�comT)zhttp://foo.com/zrhubarb.foo.comF)zhttp://foo.com/z.foo.comT)zhttp://foo.com/zfoo.comT)zhttp://foo.com/r� T)z http://foo/z rhubarb.fooF)z http://foo/z.fooT)z http://foo/rI T)z http://foo/z foo.localT)z http://foo/z.localT)zhttp://barfoo.comz.foo.comF)zhttp://barfoo.comzfoo.comF)r rt ru rv Zdomain_return_ok�
assertTrue� assertFalse)r �polrk �domain�okru �rr$ r$ r% �test_domain_return_ok� s z!CookieTests.test_domain_return_okc C s t jj}t|�}t|dd� t|dd� |jd d d }| �|j� | �|j d� |jd d d }| �|j� | �|j d� | �t
|�d� t |�}|j d d d
� zt|�}|j
d d d
� W d t�|j� X | �t |�t�d d d d |�� | �t|d
�d� d S )Nzhttp://www.acme.com/�eggsz"spam"; path=/foo/z www.acme.comr} z/foo/z"spam"zJ"spam"; path="/foo/"; domain="www.acme.com"; path_spec; discard; version=0T)Zignore_expiresr~ zpath_specified=%sFzhttp://www.acme.com/foo/z "spam"; eggs)r9 r r� r
rs r� r8 r� r �namer �reprr� Zrevertr� r� r� �re�sub)r r� rW �cookieZold_strr$ r$ r% �test_missing_value� s0
z CookieTests.test_missing_valuec C s� x�dD ]�\}}}t ||d�}t|�}t|dd� y|jd d d }W n tk
rd | �|� Y qX | �|j|� t|dd� |r|jd d d }| �|jd
� qW d S ) N))NFr )NTr, )FFN)FTr, )TFr )TTr )�rfc2109_as_netscape�rfc2965zhttp://www.example.com/zni=ni; Version=1zwww.example.comr} �nizfoo=bar; Version=1rI r, ) r r
rs r� �KeyErrorr8 r rL rr )r r� r� rL �policyrW r� Z
cookie2965r$ r$ r% �test_rfc2109_handling� s
z!CookieTests.test_rfc2109_handlingc C s� t � }t|dd� t|dd� t|dd� t|dd� t|dd� t|dd� t|dd � t|dd
� |jd d d
}| �|jd � | �|j� | �|jt� | � |j
� | �|� d�� | � |� d�� |jd d d }| �|jd� | � |j� | �|jd� | �|j
� |jd d d }| � |j� | � |j
� |jd d d }|jd d d }| � |j
� | � |j
� |jd d d }| �|j� | �|jd� |jd d d
}| � |j� d S )Nzhttp://www.acme.com/z0spam=eggs; DoMain=.acme.com; port; blArgh="feep"zni=ni; port=80,8080zhttp://www.acme.com:80/znini=nizfoo=bar; expires=z-spam=eggs; expires="Foo Bar 25 33:22:11 3022"z fortytwo=z=unladenswallowZholyhandgrenadez .acme.comr} rX ZblArghZblarghz www.acme.comr� z80,8080ZninirI Zfortytwor6 )r
rs r� r r� r� Zdomain_specifiedZportr r� Zport_specifiedZhas_nonstandard_attrr8 rK ZassertIsNotNoner� )r rW r� rI rX r$ r$ r% �test_ns_parser sD zCookieTests.test_ns_parserc C sH t � }t|dd� t|dd� |jd d }| �d|� | �d|� d S )Nzhttp://www.acme.com/z expires=eggszversion=eggs; spam=eggsz www.acme.comr} rK rL )r
rs r� �assertIn)r rW �cookiesr$ r$ r% � test_ns_parser_special_names2 s z(CookieTests.test_ns_parser_special_namesc C s� t � }tt�� d �}tjdd��j}d|� �g}tj�d�}t|d�}|� ||�}| �
t |�d� | �
t|d j �|� | �
t |�d� W d Q R X t
|dd | � | �
t |�d� tt�� d �}t
|dd
| � t
|d�} | �
t |�d� | �d | � | �d | � t
|dd
| � t
|dd| � | �
t |�d� t
|dd| � t
|dd| � t
|d�} | �
t |�d� t
|dd� | �
t |�d� |�� | �
t |�d� | �d | � tdddd dddddddddd d i �}
| �
|
j d� d S ) Ni T)�recordz%Set-Cookie: FOO=BAR; path=/; expires=zhttp://www.coyote.com/r, r zhttp://www.acme.com/zspam="bar"; expires=%szfoo="eggs"; expires=%sz
spam="bar"rI zeggs="bar"; expires=%szbar="bar"; expires=%sr z!eggs="bar"; expires=%s; max-age=0z bar="bar"; max-age=0; expires=%szhttp://www.rhubarb.net/z whum="fizz"r r� r� Fzwww.python.orgr} z1444312383.018307i?uV)r
r r) �warnings�catch_warningsrt ru rv re � make_cookiesr �lenrK rs r� � assertNotIn�clear_session_cookiesr ) r rW �futureZwarnsrj r{ rc r� Znow�hr� r$ r$ r% � test_expires= sV
zCookieTests.test_expiresc C s6 t dd�}t|�}t|dd� | �d|jd � t|�}t|dd� | �d|jd � t|�}t|d d� | �d
|jd � t|�}t|d d� | �d |jd � t� }t|dd
� | �d|jd � t� }t|dd� | �d|jd � t� }t|d d� | �d|jd � t� }t|d d� | �d|jd � d S )NT)r� zhttp://www.acme.com/zspam="bar"; Version="1"r} z www.acme.comzhttp://www.acme.com/blahzeggs="bar"; Version="1"z http://www.acme.com/blah/rhubarbz/blah/z!http://www.acme.com/blah/rhubarb/z/blah/rhubarb/z
spam="bar"z
eggs="bar"z/blahz
/blah/rhubarb)r r
rr r� r� rs )r r� rW r$ r$ r% �test_default_pathw s6
z CookieTests.test_default_pathc C sB t � }d}d}t|||� | �d|jd � | �t||�|� d S )Nz http://example.com/?spam/eggsz
eggs="bar"r} z example.com)r
rs r� r� r )r �cjZurir� r$ r$ r% � test_default_path_with_query� s z(CookieTests.test_default_path_with_queryc C sB ddddddddd d
d d g }x |D ]\}}| � t|�|� q"W d S )
N)z /foo%2f/barz /foo%2F/bar)z /foo%2F/barz /foo%2F/bar)z
/foo%%/barz
/foo%%/bar)z /fo%19o/barz /fo%19o/bar)z /fo%7do/barz /fo%7Do/bar)z /foo/bar&z /foo/bar&)z /foo//barz /foo//bar)z ~/foo/barz ~/foo/bar)z /foo/barz /foo%19/bar)z /}foo/barz /%7Dfoo/bar)u
/foo/barüz/foo/bar%C3%BC)u /foo/barꯍz/foo/bar%EA%AF%8D)r r )r Zcasesr^ r2 r$ r$ r% �test_escape_path� s z CookieTests.test_escape_pathc C sX t j�d�}| �t|�d� t j�d�}| �t|�d� t j�d�}| �t|�d� d S )NzThttp://www.example.com/rheum/rhaponticum;foo=bar;sing=song?apples=pears&spam=eggs#niz$/rheum/rhaponticum;foo=bar;sing=songzBhttp://www.example.com/rheum/rhaponticum?apples=pears&spam=eggs#niz/rheum/rhaponticumzhttp://www.example.comr} )rt ru rv r r )r r{ r$ r$ r% �test_request_path� s
z CookieTests.test_request_pathc C s� t � }t dd�}t|�}d}t||d� |jd d d }x�dD ]|\}}|� |� �}tj�|�} t||�}
|r�| �d |
d
|� �� | �|� || �� q@| �
d |
d |� �� | � |� || �� q@W d S ) NT)Zstrict_ns_set_pathzhttp://bar.comzspam=eggs; Path=/foozbar.comz/foorX ))z/fooT)z/foo/T)z/foo/barT)r} F)z /foobad/fooFz spam=eggszcookie not set for zcookie set for ) r r
rs r� rt ru rv r� r� Z set_ok_pathr� r� ) r r� Zstrict_ns_path_polrW Zbase_urlr� �pathr� rk r{ r� r$ r$ r% �test_path_prefix_match� s
z"CookieTests.test_path_prefix_matchc C sL t jjdddid�}| �t|�d� t jjdddid�}| �t|�t� d S )Nzhttp://www.acme.com:1234/�Hostzwww.acme.com:4321)rj Z1234zhttp://www.acme.com/)rt ru rv r r r )r r{ r$ r$ r% �test_request_port� s z CookieTests.test_request_portc C sp t jjdddid�}| �t|�d� t jjdddid�}| �t|�d� t jjd dd
id�}| �t|�d� d S ) Nzhttp://1.1.1.1/r� zwww.acme.com:80)rj z1.1.1.1zhttp://www.acme.com/zirrelevant.comz www.acme.comz&http://www.acme.com:2345/resource.htmlzwww.acme.com:5432)rt ru rv r r )r r{ r$ r$ r% �test_request_host� s z CookieTests.test_request_hostc C st | � td�� | � td�� | �td�� | �td�� | �td�� | �td�� | �td�� | �td�� d S ) Nz foo.bar.comz1foo2.3bar4.5comz 192.168.1.1r6 �.z .foo.bar.comz..foozfoo.)r� r r� )r r$ r$ r% � test_is_HDN s zCookieTests.test_is_HDNc C s� | � td�d� | � td�d� | � td�d� | � td�d� | � td�d� | � td�d� | � td�d� | � td �d � d S )
Nz www.acme.comz .acme.comzacme.comz
acme.localz.localz.comr� r6 z 192.168.0.1)r r )r r$ r$ r% �
test_reach s zCookieTests.test_reachc C s� | � tdd�� | �tdd�� | � tdd�� | � tdd�� | �tdd�� | � tdd�� | �tdd�� | � td d
�� | �td d �� | �td d
�� | � td d �� | � tdd�� | �tdd�� | � tdd�� | � tdd�� | � tdd�� | � tdd�� | �tdd�� | � tdd�� | �tdd�� | � tdd�� | � tdd�� | �tdd�� | �tdd�� | �tdd�� | �tdd �� | �tdd�� | � tdd�� | �tdd�� | �tdd�� | �tdd �� d S )Nz 192.168.1.1z.168.1.1zx.y.comzx.Y.comz.Y.comzY.comz a.b.c.comz.c.comz
example.localz.localz blah.blahr6 z.rhubarb.rhubarbzacme.comz .acme.comzrhubarb.acme.comzwww.rhubarb.acme.comzy.comz.y.comz.comr� �mz.mr� )r� r r� r )r r$ r$ r% �test_domain_match s> z CookieTests.test_domain_matchc C s& t � }t|dd� | �t|�d� d S )Nzhttp://www.nasty.com/z)foo=bar; domain=friendly.org; Version="1"r )r
rr r r� )r rW r$ r$ r% �test_wrong_domain: s z CookieTests.test_wrong_domainc C sn t dd�}t|d�}t|dd� t|dd� | �t|�d� x,dD ]$}t|d | d
� | �t|�d� qBW d S ) NT)Z
strict_domain)r� zhttp://example.co.uk/z no=problemoz!okey=dokey; Domain=.example.co.ukr )z.co.ukz.org.zaz.tx.usz.name.uszhttp://example.%s/zspam=eggs; Domain=.co.uk)r r
rs r r� )r Zcpr� Z
pseudo_tldr$ r$ r% �test_strict_domainD s
z CookieTests.test_strict_domainc C s8 t � }t|dd� | �t|�d� | �|jd d d jd� | �t|d�d� | �t|d�d� ttjd �}|�|� | �t|d�d
� t|d d � t|d
d� | �t|�d� | �|jd d d jd� | �|jd d d jd� | �t|d
�d� t|dd� | �t|�d� t|dd� | �t|�d� d S )Nzhttp://foo.net/zns=barr, zfoo.netr} �nsrJ zhttp://www.foo.net/)Zstrict_ns_domainr6 zhttp://foo.net/foo/zspam1=eggs; domain=foo.netzhttp://foo.net/foo/bar/zspam2=eggs; domain=.foo.netr z.foo.netz/fooZspam1r� z/foo/barZspam2z spam2=eggs; spam1=eggs; ns=barznini="ni"; domain=.netzhttp://foo.co.ukznasty=trick; domain=.co.uk� ) r
rs r r� r� r� r ZDomainStrictNonDomain�
set_policy)r rW r� r$ r$ r% � test_two_component_domain_nsR s4
z(CookieTests.test_two_component_domain_nsc C s4 t dd�}t|�}t|dd� | �t|�d� | �|jd d d jd � | �t|d�d
� | �t|d �d � t|d
d� | �t|�d� | �t|d
�d
� t|dd� | �|jd d d jd� | �t|�d� | �t|d�d
� | �t|d�d� t|dd� | �t|�d� t|dd� | �t|�d� d S ) NT)r� zhttp://foo.net/zfoo=bar; Version="1"r, zfoo.netr} rI rJ z$Version=1; foo=barzhttp://www.foo.net/r6 zhttp://foo.net/fooz1spam=eggs; domain=foo.net; path=/foo; Version="1"zhttp://www.foo.net/foo/z&spam=eggs; domain=foo.net; Version="1"z.foo.netz/foo/rX r� r zhttp://foo.net/foo/z($Version=1; spam=eggs; $Domain="foo.net"z#ni="ni"; domain=".net"; Version="1"zhttp://foo.co.uk/z'nasty=trick; domain=.co.uk; Version="1"r )r r
rr r r� r� r� )r r� rW r$ r$ r% �!test_two_component_domain_rfc2965� s8
z-CookieTests.test_two_component_domain_rfc2965c C s t tdgdgd�d�}tj�d�}dg}t|d�}|�||� | �t|�d� tj�d�}t|d�}|�||� | �t|�d � tj�d
�}t|d
�}|�||� | �t|�d � tj�d
�}t|d
�}|� ||�}|�
|d � | �t|�d � |� |� | � |�
d �� d S )
Nzacme.comz www.acme.com)�blocked_domainsZallowed_domains)r� zhttp://acme.com/z*Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/r zhttp://www.acme.com/r, zhttp://www.coyote.com/r r )r
r rt ru rv re rz r r� r� �
set_cookierw r� �
has_header)r rW r{ rj rc r� r$ r$ r% �test_domain_allow� s.
z CookieTests.test_domain_allowc C s� t ddgd�}t|d�}dg}tj�d�}t|d�}|�||� | �t|�d� |� dg�}|�||� | �t|�d � |�
� tj�d
�}t|d
�}|�||� | �t|�d � tj�d
�}|� |� | � |�
d �� | � |�
d �� |�
� |� dg� |�||� | �t|�d � tj�d�}t|d�}|�||�}|�|d � | �t|�d
� |� |� | �|�
d �� |�
� |� g � tj�d�}t|d�}|�||�}|�||� | �t|�d � tj�d�}|� |� | � |�
d �� tj�d�}|� |� | �|�|d |�� | �|�
d �� |� dg�}tj�d�}|� |� | �|�
d �� tj�d�}|� |� | �|�
d �� d S )NTz .acme.com)r� r� )r� z*Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/zhttp://www.acme.com/r zacme.comr, zhttp://www.roadrunner.net/r �Cookie2r zhttp://acme.com/zhttp://badacme.com/)r r
rt ru rv re rz r r� Zset_blocked_domains�clearrw r� r� r� r� r� Z return_ok)r r� rW rj r{ rc �pr� r$ r$ r% �test_domain_block� sh
z CookieTests.test_domain_blockc C s� x�dD ]�}x�dD ]�}t � }|r2tdd�}t}d}ntdd�}t}d}|�|� d}|||d ||f � |||d
||f � | �|jd d d
jd� | �|jd d d jd� qW qW d S )N)TF)r; r6 F)r� r6 Tz ; Version=1zhttp://www.acme.com/z foo1=bar%s%szfoo2=bar%s; secure%sz www.acme.comr} Zfoo1z#non-secure cookie registered secureZfoo2z#secure cookie registered non-secure) r
r rs rr r� r� r� Zsecurer� )r r� Z
whitespacerW r� �intZvsrk r$ r$ r% � test_secure s(
zCookieTests.test_securec C s6 t tdd�d�}t|dd� t|d�}| �|d� d S )NT)r� )r� zhttp://www.acme.com/zfoo=\b"a"r; Version=1z$Version=1; foo=\\b\"a\"r)r
r rr r )r rW r� r$ r$ r% �test_quote_cookie_value' s
z#CookieTests.test_quote_cookie_valuec C sX d}t tdd��}t||d� tj�|�}| �t|�d� |�|� | � |�
d�� d S )Nzhttp://www.acme.comT)r� zfoo=bar; Version=1r, r ) r
r rr rt ru rv r r� rw r� r� )r rk rW r{ r$ r$ r% �test_missing_final_slash- s
z$CookieTests.test_missing_final_slashc C s� t dd�}t|�}d}t||d� t||�}| �d|d� t|�}d}t||d� t||�}| �d|d � t|�}d}t||d
� t||�}| �d |d � d S ) NT)r� zhttp://foo.bar.com/zspam=eggs; Version=1ZDomainz*absent domain returned with domain presentz%spam=eggs; Version=1; Domain=.bar.comz$Domain=".bar.com"zdomain not returnedz$spam=eggs; Version=1; Domain=bar.comz$Domain="bar.com")r r
rr r� r� )r r� rW rk r� r$ r$ r% �test_domain_mirror7 s"
z CookieTests.test_domain_mirrorc C sn t dd�}t|�}d}t||d� t||�}| �d|d� t|�}d}t||d� t||�}| �d|d � d S )
NT)r� zhttp://foo.bar.com/zspam=eggs; Version=1ZPathz&absent path returned with path presentz spam=eggs; Version=1; Path=/z $Path="/"zpath not returned)r r
rr r� r� )r r� rW rk r� r$ r$ r% �test_path_mirrorN s
z CookieTests.test_path_mirrorc C s� t dd�}t|�}d}t||d� t||�}| �d|d� t|�}d}t||d� t||�}| �|dd � t|�}d}t||d
� t||�}| �d |d � t|�}d}t||d
� t||�}| �d|d� d S )NT)r� zhttp://foo.bar.com/zspam=eggs; Version=1ZPortz&absent port returned with port presentzspam=eggs; Version=1; Portz\$Port([^=]|$)z-port with no value not returned with no valuez spam=eggs; Version=1; Port="80"z
$Port="80"z5port with single value not returned with single valuez$spam=eggs; Version=1; Port="80,8080"z$Port="80,8080"z;port with multiple values not returned with multiple values)r r
rr r� r r� )r r� rW rk r� r$ r$ r% �test_port_mirror] s0
z CookieTests.test_port_mirrorc C s: t tdd��}d}t||d� t||�}| �d|d� d S )NT)r� zhttp://foo.bar.com/zfspam=eggs; Version=1; Comment="does anybody read these?"; CommentURL="http://foo.bar.net/comment.html"�Commentz:Comment or CommentURL cookie-attributes returned to server)r
r rr r� )r rW rk r� r$ r$ r% �test_no_return_comment| s
z"CookieTests.test_no_return_commentc C s t tdd��}t|dd� t|dd� t|dd� t|dd� t|d d
� d d d d d g}d
ddddg}dddddg}dddddg}x~td�D ]r}d }xh|D ]`}| �|t� | �|j|| � | �|j || � | �|j
|| � | �|j || � |d }q�W q�W d S )NT)r� zhttp://blah.spam.org/zefoo=eggs; Version=1; Comment="does anybody read these?"; CommentURL="http://foo.bar.net/comment.html"zhttp://www.acme.com/blah/zspam=bar; securezfoo=bar; secure; Version=1zfoo=bar; path=/; Version=1zhttp://www.sol.nozsbang=wallop; version=1; domain=".sol.no"; port="90,100, 80,8080"; max-age=100; Comment = "Just kidding! (\"|\\\\) "r, r ZbangrI rX z.sol.noz
blah.spam.orgz www.acme.comr} z/blahz/blah/r� ) r
r rr rs �rangeZassertIsInstancer r rL r� r� r� )r �csZversions�namesZdomains�paths�irW r$ r$ r% �test_Cookie_iterator� s.
z CookieTests.test_Cookie_iteratorc C s� | � tdg�ddddgg� | � tdg�dddgg� | � tdg�d dgg� | � td
g�dddgg� | � td g�dd gg� | � td
g�g � d S )Nzfoo=bar; path=/; domain)rI rJ )r� r} )r� N)rL rM z)foo=bar; expires=Foo Bar 12 33:22:11 2000)rK NrI )rI Nzfoo=bar; expireszfoo=bar; version)rL Nr6 )r r )r r$ r$ r% rO � s" z!CookieTests.test_parse_ns_headersc C s� dd� }t t�� d �}x@dgdgdgdgdgd | ggD ] }||�}| �t|�d
� q:W d g}||�}|jd d
d }| �|j� d S )Nc S s, t � }tj�d�}t| d�}|�||� |S )Nzhttp://www.example.com/)r
rt ru rv re rz )rj rW r{ r� r$ r$ r% � cookiejar_from_cookie_headers� s
zICookieTests.test_bad_cookie_header.<locals>.cookiejar_from_cookie_headersi z Set-Cookie: z
Set-Cookie2: z-Set-Cookie2: a=foo; path=/; Version=1; domainz Set-Cookie: b=foo; max-age=oopsz Set-Cookie: b=foo; version=spamzSet-Cookie:; Expires=%sr z3Set-Cookie: c=foo; expires=Foo Bar 12 33:22:11 2000zwww.example.comr} rW )r r) r r� r� r8 rK )r r� r� rj rW r� r$ r$ r% �test_bad_cookie_header� s z"CookieTests.test_bad_cookie_headerN)#rE rF rG r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rO r� r$ r$ r$ r% r� s s@ /! 0 :(
$
0, C
r� c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�LWPCookieTestsc C s� t �� d d }g }ttdd��}tjjdddid�}|�d | � t|d
�}|� ||� tj�d
�}|�
|� | � |� d �d � | � |� d
�d� |�d� t|d
�}|� ||� tj�d�}|�
|� |� d �}| �
d|� | �
d |� |�d� t|d�}|� ||� tj�d
�}|�
|� |� d �}| �
d|� | �
d |� | �d|� tj�d�}|�
|� |� d �}| �
d|� | �
d |� | �|�d�� d S )Nr r, T)r� zhttp://www.acme.com:80/r� zwww.acme.com:80)rj zVSet-Cookie: CUSTOMER=WILE_E_COYOTE; path=/ ; expires=Wednesday, 09-Nov-%d 23:12:40 GMTzhttp://www.acme.com/r zCUSTOMER=WILE_E_COYOTEr� z $Version="1"z4Set-Cookie: PART_NUMBER=ROCKET_LAUNCHER_0001; path=/zhttp://www.acme.com/foo/barz PART_NUMBER=ROCKET_LAUNCHER_0001z%Set-Cookie: SHIPPING=FEDEX; path=/foozhttp://www.acme.comzSHIPPING=FEDEXzhttp://www.acme.com/foo/zSHIPPING=FEDEX;)r) � localtimer
r rt ru rv ry re rz rw r rx r� r� r� �
startswith)r �
year_plus_onerj rW r{ rc r� r$ r$ r% �test_netscape_example_1� sJ %
z&LWPCookieTests.test_netscape_example_1c C s� t � }g }tj�d�}|�d� t|d�}|�||� tj�d�}|�|� | �|� d�d� |�d� t|d�}|�||� tj�d�}|�|� | �
|� d�d� d S )Nzhttp://www.acme.com/z4Set-Cookie: PART_NUMBER=ROCKET_LAUNCHER_0001; path=/r z PART_NUMBER=ROCKET_LAUNCHER_0001z6Set-Cookie: PART_NUMBER=RIDING_ROCKET_0023; path=/ammozhttp://www.acme.com/ammozBPART_NUMBER=RIDING_ROCKET_0023;\s*PART_NUMBER=ROCKET_LAUNCHER_0001) r
rt ru rv ry re rz rw r rx r )r rW rj r{ rc r$ r$ r% �test_netscape_example_2A s$
z&LWPCookieTests.test_netscape_example_2c C s� t tdd��}t|dd�}| �|� t|dd�}| �|d� t|dd �}| �|d
� | �|d � | �|d � t|d
�}| �|d� | �d|� d S )NT)r� z http://www.acme.com/acme/loginz3Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"z!http://www.acme.com/acme/pickitemz=Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"z=^\$Version="?1"?; Customer="?WILE_E_COYOTE"?; \$Path="/acme"$z!http://www.acme.com/acme/shippingz+Shipping="FedEx"; Version="1"; Path="/acme"z^\$Version="?1"?;z7Part_Number="?Rocket_Launcher_0001"?;\s*\$Path="\/acme"z-Customer="?WILE_E_COYOTE"?;\s*\$Path="\/acme"z http://www.acme.com/acme/processz%Shipping="?FedEx"?;\s*\$Path="\/acme"Z
WILE_E_COYOTE)r
r rr r� r r� )r rW r� r$ r$ r% �test_ietf_example_1t s"
z"LWPCookieTests.test_ietf_example_1c C sX t tdd��}t|ddd� t|d�}| �|d� t|d�}| �d |� | �d
|� d S ) NT)r� z&http://www.acme.com/acme/ammo/specificz=Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"z@Part_Number="Riding_Rocket_0023"; Version="1"; Path="/acme/ammo"z!http://www.acme.com/acme/ammo/...z(Riding_Rocket_0023.*Rocket_Launcher_0001z http://www.acme.com/acme/parts/ZRocket_Launcher_0001ZRiding_Rocket_0023)r
r rr r r� r� )r rW r� r$ r$ r% �test_ietf_example_2� s
z"LWPCookieTests.test_ietf_example_2c C s� t dd�}t|d�}d}t|dd�}| �|� t|dd�}| �t|�d� t|d d
�}| �t|�d� t|d d �}| �t|�d � t|d
d�}| �t|�d � t|dd�}| �t|�d � t|dd�}| �t|�d� t|dd�}| �t|�d� t|dd�}| �t|�d� t|dd�}| �t|�d� t|dd�}| �t|�d� tjj}z2|j |dd � t
|�}t|d�}|j |dd � W d yt �
|� W n tk
�r� Y nX X | �|t
|�� d S ) NT)r� )r� z max-age=3600zhttp://www.acme.comz!foo=bar; domain=".com"; version=1z'ping=pong; domain="acme.com"; version=1r, zhttp://www.a.acme.comz'whiz=bang; domain="acme.com"; version=1z,wow=flutter; domain=".a.acme.com"; version=1r zhttp://125.125.125.125z*zzzz=ping; domain="125.125.125"; version=1zhttp://www.sol.noz6blah=rhubarb; domain=".sol.no"; path="/foo"; version=1zhttp://www.sol.no/foo/barz3bing=bong; domain=".sol.no"; path="/foo"; version=1r z5whiz=ffft; domain=".sol.no"; port="90,100"; version=1zsbang=wallop; version=1; domain=".sol.no"; port="90,100, 80,8080"; max-age=100; Comment = "Just kidding! (\"|\\\\) "r� z9foo9=bar; version=1; domain=".sol.no"; port; max-age=100;� zhttp://www.sol.no/<oo/z"foo8=bar; version=1; path="/%3coo"r( )r~ )r r rr r� r r� r9 r r� r� r� r� r� r� r� )r r� rW Zmax_ager� r� �oldr$ r$ r% �test_rejection s`
z LWPCookieTests.test_rejectionc C sn t tdd��}t|dd� t|dd�}t�dtj�}| �d|� | �||� t|d �}| �|� t|d
�}d S ) NT)r� z7http://www.acme.com/foo%2f%25/%3c%3c%0Anew%C3%A5/%C3%A5z foo = bar; version = 1u/ http://www.acme.com/foo%2f%25/<<%0anewå/æøåz bar=baz; path="/foo/"; version=1z^\$version=\"?1\"?zfoo=baru- http://www.acme.com/foo/%25/<<%0anewå/æøåu http://www.acme.com/ü) r
r rr r� �compile�Ir� r r� )r rW r� Z
version_rer$ r$ r% �test_url_encodingt s
z LWPCookieTests.test_url_encodingc s� t �� d d }tjj� t� tdd�d�}t|dd� t|dd� t|dd � d
|f }t|d d | � t|d d
| � t|d d| � � fdd�}||d�}| � t
|�d� | � dt |�� ||d�}| � t
|�d� | � dt |�� d S )Nr r, T)r� )r� zhttp://www.acme.com/z foo1=bar; max-age=100; Version=1z4foo2=bar; port="80"; max-age=100; Discard; Version=1zfoo3=bar; secure; Version=1z expires=09-Nov-%d 23:12:40 GMTzhttp://www.foo.com/z fooa=bar; %sz foob=bar; Domain=.foo.com; %sz fooc=bar; Domain=www.foo.com; %sc sX z,| j |d� t� tdd��}|j|d� W d yt�� � W n tk
rP Y nX X |S )N)r~ T)r� )r� r
r r� r� r� r� )r� r~ �new_c)r� r$ r% �save_and_restore� s z5LWPCookieTests.test_mozilla.<locals>.save_and_restorer( zname='foo1', value='bar'Fr� )
r) r� r9 r r� r
r rr rs r r� r� r� )r r� rW rK r� r� r$ )r� r% � test_mozilla� s.
zLWPCookieTests.test_mozillac C s� t � }g }tj�d�}|�d� t|d�}|�||� |�d� t|d�}|�||� tj�d�}|�|� | �d|� d�� | �d|� d�� d S )Nzhttp://foo.bar.acme.com/fooz4Set-Cookie: Customer=WILE_E_COYOTE; domain=.acme.comzhttp://www.acme.com/fooz4Set-Cookie: PART_NUMBER=3,4; domain=foo.bar.acme.comzPART_NUMBER=3,4r zCustomer=WILE_E_COYOTE)
r
rt ru rv ry re rz rw r� rx )r rW rj r{ rc r$ r$ r% �test_netscape_misc� s
z!LWPCookieTests.test_netscape_miscc C sh t tdd��}t|dd� t|dd�}| �d|� t|dd� t|d�}| �d|� | �t|�d � d S )
NT)r� zhttp://example/z#foo1=bar; PORT; Discard; Version=1;z$foo2=bar; domain=".local"; Version=1zfoo1=barzfoo3=bar; Version=1zfoo2=barr )r
r rr r� r r� )r rW r� r$ r$ r% �test_intranet_domains_2965� s
z)LWPCookieTests.test_intranet_domains_2965c C sl t tdd��}t|dd� t|dd�}| �t|�d� | �d|� t|d�}| �d|� | �t|�d� d S )NF)r� zhttp://example/zfoo1=barzfoo2=bar; domain=.localr zfoo2=bar)r
r rs r r� r� )r rW r� r$ r$ r% �test_intranet_domains_ns� s
z'LWPCookieTests.test_intranet_domains_nsc C s� t tdd��}g }tj�d�}|�d� t|d�}|�||� tj�d�}|�|� | � |�
d�d� | � |�
d�d� tj�d �}|�|� | � |�
d�d� | � |�
d�d� d S )
NT)r� zhttp://www.ants.com/z,Set-Cookie: JSESSIONID=ABCDERANDOM123; Path=r zJSESSIONID=ABCDERANDOM123r� z $Version="1"zhttp://www.ants.com:8080) r
r rt ru rv ry re rz rw r rx )r rW rj r{ rc r$ r$ r% �test_empty_path� s
z LWPCookieTests.test_empty_pathc C s t �� d d }tj�d�}g }|�d� |�d| � |�d| � |�d� |�d� t|d�}t� }|�||� ddddd �}x&|D ] }d
|j }|| d ||< q�W |�
� x&|D ] }d |j }|| d ||< q�W | � |d |d
� | � |d d� | � |d d� d S )Nr r, z"http://www.perlmeister.com/scriptsz$Set-Cookie: s1=session;Path=/scriptszWSet-Cookie: p1=perm; Domain=.perlmeister.com;Path=/;expires=Fri, 02-Feb-%d 23:24:20 GMTz>Set-Cookie: p2=perm;Path=/;expires=Fri, 02-Feb-%d 23:24:20 GMTz<Set-Cookie: s2=session;Path=/scripts;Domain=.perlmeister.comz2Set-Cookie2: s3=session;Version=1;Discard;Path="/")�
session_after�
perm_after�session_before� perm_beforez %s_beforez%s_afterr� r� r� r� )
r) r� rt ru rv ry re r
rz r� r� r ZassertNotEqual) r r� r{ rj rc rW Zcounterr� �keyr$ r$ r% �test_session_cookies s6
z#LWPCookieTests.test_session_cookiesN)rE rF rG r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% r� � s [3p5[) r� c C s t j�ttttt� d S )N)r9 r Z run_unittestr rH r� r| r� )�verboser$ r$ r% � test_main. s r� �__main__T)r� )N).�__doc__r� r� Z test.supportr9 r) ZunittestZurllib.requestrt r� Zhttp.cookiejarr r r r r r r r r
r r r
r r r r r r r r r r r ZTestCaser rH re rr rs ro r| r� r� r� rE r$ r$ r$ r% �<module> s> d 3g
* v O