Your IP : 216.73.216.85
3
� f 8 � @ sx d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
G dd� d�Z G dd� dej �Z
d d
� ZdS ) � N)�
HTTPException)�support)�BytesIOc @ s� e Zd ZdZdZdZdZdZdZdZ dZ
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z dS )*�TestBase� N� r u �c C sX | j d krtj| j�| _ | j j| _| j j| _| j j| _| j j| _ | j j
| _
| j j | _ d S )N) �codec�codecs�lookup�encoding�encode�decode� streamreader�reader� streamwriter�writer�incrementalencoder�incrementaldecoder)�self� r �@/opt/alt/python36/lib64/python3.6/test/multibytecodec_support.py�setUp s
zTestBase.setUpc C s� g }xD| j D ]:}|jd�}|j� }|dks.t�dd� |D �}|j|� q W xPt|� D ]D\}}| j|�d }| j||jd�� | jrT| j|| j |�d � qTW d S )N�
� c S s g | ] }|d �qS )r r )�.0�liner r r �
<listcomp>* s z-TestBase.test_chunkcoding.<locals>.<listcomp>r zutf-8)
�tstring�split�pop�AssertionError�append�zipr
� assertEqual�
roundtriptestr )r Z
tstring_lines�b�linesZlastZnative�utf8�ur r r �test_chunkcoding$ s
zTestBase.test_chunkcodingc
C s� x�| j D ]�\}}}t|t�r$| j}n| j}|r�|||�d }|| jkr~| jt|�tkt|�� | j||d|| j |||f � q�| jt|�tkt|�� | j||d|| j |||f � q| j
t |||� qW d S )Nr z%a.decode(%r, %r)=%a != %az%a.encode(%r, %r)=%a != %a) �
codectests�
isinstance�bytesr
r �
assertTrue�type�strr# r � assertRaises� UnicodeError)r �source�scheme�expected�func�resultr r r �test_errorhandle2 s$
zTestBase.test_errorhandlec C s0 | j r| jd� d}| j| j|d�d d� d S )Nz$encoding contains full ISO 10646 mapu ଓଣୠ nd eggs�xmlcharrefreplacer s ଓଣୠ nd eggs)� has_iso10646�skipTestr# r )r �sr r r �test_xmlcharrefreplaceI s
z TestBase.test_xmlcharrefreplacec sj | j r| jd� ddlm� � fdd�}tjd|� | jrF| j\}}nd}d}| j| j|d�d |� d S ) Nz$encoding contains full ISO 10646 mapr )�codepoint2namec s| t | t�std| ��g }xP| j| j| j� D ]:}t|�� krV|jd� t|� � q.|jdt|� � q.W dj|�| jfS )Nzdon't know how to handle %rz&%s;z&#%d;r ) r+ �UnicodeEncodeError� TypeError�object�start�end�ordr! �join)�exc�l�c)r= r r �xmlcharnamereplaceY s
z>TestBase.test_customreplace_encode.<locals>.xmlcharnamereplaceztest.xmlcharnamereplaceu «ℜ» = 〈ሴ〉s* «ℜ» = ⟨ሴ⟩) r9 r: Z
html.entitiesr= r �register_error�xmlcharnametestr# r )r rH ZsinZsoutr )r= r �test_customreplace_encodeS s
z"TestBase.test_customreplace_encodec C s> dd� }t jd|� | jd| j d d�d }| j|d� d S )Nc S s
d| j fS )Ns 1234)rB )rE r r r � myreplaceo s z7TestBase.test_callback_returns_bytes.<locals>.myreplacez test.cjktest�abc�defr s
abc1234def)r rI r �unmappedunicoder# )r rL �encr r r �test_callback_returns_bytesn s z$TestBase.test_callback_returns_bytesc sN � fdd�}t jd|� x0dddgg d t� fD ]� | jt| j| jd� q.W d S )Nc s
� | j fS )N)rB )rE )�retr r rL v s z7TestBase.test_callback_wrong_objects.<locals>.myreplacez test.cjktestr � � )r rI r@ r0 r? r rO )r rL r )rR r �test_callback_wrong_objectsu s
z$TestBase.test_callback_wrong_objectsc C s^ dd� }t jd|� | j| jd| j d d�d � dd� }t jd|� | jt| j| jd� d S )
Nc S s dt | j�fS )N�x)�intrB )rE r r r rL s z4TestBase.test_callback_long_index.<locals>.myreplacez test.cjktest�abcd�efgh� abcdxefgh� c S s dt jd fS )NrV r )�sys�maxsize)rE r r r rL � s )rZ r[ )r rI r# r rO r0 �
IndexError)r rL r r r �test_callback_long_index~ s
z!TestBase.test_callback_long_indexc C s, dd� }t jd|� | jt| j| jd� d S )Nc S s dS )NrV )rV Nr )rE r r r rL � s z4TestBase.test_callback_None_index.<locals>.myreplacez test.cjktest)r rI r0 r? r rO )r rL r r r �test_callback_None_index� s z!TestBase.test_callback_None_indexc s@ � fdd�� d� _ tjd� � | j| jd| j d d�d � d S )
Nc s* � j dkr � j d8 _ dS d| jfS d S )Nr r �REPLACEDZTERMINAL)ra r )�limitrB )rE )rL r r rL � s
z8TestBase.test_callback_backward_index.<locals>.myreplacerT z test.cjktestrX rY �4 abcdREPLACEDabcdREPLACEDabcdREPLACEDabcdTERMINALefghr[ )rc r[ )rb r rI r# r rO )r r )rL r � test_callback_backward_index� s z%TestBase.test_callback_backward_indexc C s6 dd� }t jd|� | j| jd| j d d�d� d S ) Nc S s d| j d fS )Nra rS )rB )rE r r r rL � s z7TestBase.test_callback_forward_index.<locals>.myreplacez test.cjktestrX rY � abcdREPLACEDghr[ )re r[ )r rI r# r rO )r rL r r r �test_callback_forward_index� s z$TestBase.test_callback_forward_indexc C s, dd� }t jd|� | jt| j| jd� d S )Nc S s dS )N�TERM�d )rg rh r )rE r r r rL � s z:TestBase.test_callback_index_outofbound.<locals>.myreplacez test.cjktest)r rI r0 r^ r rO )r rL r r r � test_callback_index_outofbound� s z'TestBase.test_callback_index_outofboundc C s� t jd�}x�d gttdd�� dddddg D ]v}|t| jd ��}t� }| j� }x:|d k rh|j|�}n|j� }|svP |j|�}|j |� qTW | j
|j � | jd � q.W d S )
Nzutf-8r �! �@ � � i i r ) r � getreader�list�ranger r r �readr �writer# �getvalue)r �
UTF8Reader�sizehint�istream�ostream�encoder�data�er r r �test_incrementalencoder� s
z TestBase.test_incrementalencoderc C s� t jd�}x�d d
gttdd�� dddddg D ]f}t| jd �}|t� �}| j� }x*|j|�}|shP qV|j|�}|j |� qVW | j
|j � | jd � q0W d S ) Nzutf-8r rj rk rl rm i i r ���) r � getwriterro rp r r r rq r
rr r# rs )r �
UTF8Writerru rv rw �decoderry r( r r r �test_incrementaldecoder� s
z TestBase.test_incrementaldecoderc C s� | j }| j� }| jt|j|d� d|_| j|j|d�d� |j� dd� }tj d|� d|_| j|j|d�d� d|_| j|j|d�d� d S )NT�ignorer c S s
d| j fS )NZcalled)rB )rE r r r � tempreplace� s zDTestBase.test_incrementalencoder_error_callback.<locals>.tempreplacez test.incremental_error_callbacks called)
rO r r0 r> r �errorsr# �resetr rI )r �invrz r� r r r �&test_incrementalencoder_error_callback� s z/TestBase.test_incrementalencoder_error_callbackc C s� t jd�}x�d
D ]�}x�d dgttdd�� ddd d
d g D ]t}| jt| jd ��}|t� �}t||�}x0||�}|svP |dkr�|j|� qh|j |� qhW | j
|j � | jd � q:W qW d S )Nzutf-8rq �readline� readlinesr rj rk rl rm i i r )rq r� r� r| ) r r} ro rp r r r �getattr�
writelinesrr r# rs )r r~ �nameru rv rw r5 ry r r r �test_streamreader� s
zTestBase.test_streamreaderc C s� d
}t jd�}x�|D ]�}x�d gttdd�� ddd d
d g D ]�}|t| jd ��}| jt� �}t||�}x@|d k r|||�}n|� }|s�P |dkr�|j|� qj|j |� qjW | j
|j � | jd � q<W qW d S )Nrq r� r� zutf-8r rj rk rl rm i i r )rq r� r� ) r rn ro rp r r r r� r� rr r# rs ) r Z readfuncsrt r� ru rv rw r5 ry r r r �test_streamwriter� s$
zTestBase.test_streamwriterc C s t � }| j|�}|j� d S )N)r r r� )r �streamr r r r �"test_streamwriter_reset_no_pending s
z+TestBase.test_streamwriter_reset_no_pending) �__name__�
__module__� __qualname__r r r r* r$ r9 rJ rO r r) r7 r< rK rQ rU r_ r` rd rf ri r{ r� r� r� r� r� r r r r r s4
r c @ s\ e Zd Zg Zg Zg Zg Zdd� Zdd� Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� ZdS )�TestBase_Mappingc C s> y| j � j� W n( ttfk
r8 | jd| j � Y nX d S )NzCould not retrieve )�open_mapping_file�close�OSErrorr r: �
mapfileurl)r r r r r s zTestBase_Mapping.setUpc C s t j| j�S )N)r Zopen_urlresourcer� )r r r r r� % s z"TestBase_Mapping.open_mapping_filec C s" | j jd�r| j� n| j� d S )Nz.xml)r� �endswith�_test_mapping_file_ucm�_test_mapping_file_plain)r r r r �test_mapping_file( s
z"TestBase_Mapping.test_mapping_filec C s� dd� }i }| j � ��}x�|D ]�}|s&P |jd�d j� }t|�dkrFq |d d d� dkrj| jd|��� tj|d dd � �}t|�dkr�d |d kr�q ||d �}t|�d
ks ||kr�q |||< | j||� q W W d Q R X d S ) Nc S s dj dd� | jd�D ��S )Nr c s s | ]}t t|d ��V qdS )� N)�chrrW )r rV r r r � <genexpr>0 s zMTestBase_Mapping._test_mapping_file_plain.<locals>.unichrs.<locals>.<genexpr>�+)rD r )r; r r r �unichrs/ s z:TestBase_Mapping._test_mapping_file_plain.<locals>.unichrs�#r rS Z0xzInvalid line: r rl i�� )r� r �lenZfailr, �fromhexrC �
_testpoint)r r� Zurt_wa�fr ry �csetch�unichr r r r� . s&
z)TestBase_Mapping._test_mapping_file_plainc C sn | j � �}|j� }W d Q R X tjd|�}x@|D ]8\}}tt|d��}tdd� |j� D ��}| j||� q.W d S )Nz'<a u="([A-F0-9]{4})" b="([0-9A-F ]+)"/>r� c s s | ]}t |d �V qdS )r� N)rW )r rG r r r r� O s z:TestBase_Mapping._test_mapping_file_ucm.<locals>.<genexpr>) r� rq �re�findallr� rW r, r r� )r r� ZucmdataZucZuniZcodedr� Zcodechr r r r� I s
z'TestBase_Mapping._test_mapping_file_ucmc C s x| j D ]}| j|� qW d S )N)�supmapsr� )r �mappingr r r �test_mapping_supplementalR s z*TestBase_Mapping.test_mapping_supplementalc C sH ||f| j kr"| j|j| j�|� ||f| jkrD| jt|| j�|� d S )N)� pass_enctestr# r r � pass_dectestr/ )r r� r� r r r r� V s zTestBase_Mapping._testpointc
C s� x�| j D ]�\}}}t|t�r$|j}n|j}|r�t|t�r||| j|�}| jt|�tkt|�� | j ||d|| j|||f � q�|| j|�}| jt|�tkt|�� | j ||d|| j|||f � q| j
t || j|� qW d S )Nz%a.decode(%r, %r)=%a != %az%a.encode(%r, %r)=%a != %a) r* r+ r, r
r r r- r. r/ r# r0 r1 )r r2 r3 r4 r5 r6 r r r r7 \ s&
z!TestBase_Mapping.test_errorhandleN)r� r� r� r� r� r� r* r r� r� r� r� r� r� r7 r r r r r� s r� c C sv t jjt jjt�d�}tt jj|| d �d��}|j� }W d Q R X tt jj|| d �d��}|j� }W d Q R X ||fS )NZ cjkencodingsz.txt�rbz -utf8.txt)�os�pathrD �dirname�__file__�openrq )r� �dirr� Zencodedr' r r r �load_teststringt s r� )r r� r� r\ ZunittestZ http.clientr Ztestr �ior r ZTestCaser� r� r r r r �<module> s [