B
�/g^N: � @ 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 d*d+� Z dS ),�TestBase� N� r u ���c C sX | j d krt�| 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 �3/usr/lib64/python3.7/test/multibytecodec_support.py�setUp s
zTestBase.setUpc C s� g }x8| j D ].}|�d�}|�� }dd� |D �}|�|� q W xPt|� D ]D\}}| �|�d }| �||�d�� | jrH| �|| �|�d � qHW 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�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~| �t|�tkt|�� | �||d|| j |||f � q�| �t|�tkt|�� | �||d|| j |||f � q| �
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| �d� d}| �| �|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| �d� ddlm� � fdd�}t�d|� | jrF| j\}}nd}d}| �| �|d�d |� d S ) Nz$encoding contains full ISO 10646 mapr )�codepoint2namec s| t | t�std| ��g }xP| j| j| j� D ]:}t|�� krV|�d� t|� � q.|�dt|� � q.W d�|�| 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* «ℜ» = ⟨ሴ⟩) r7 r8 Z
html.entitiesr; r �register_error�xmlcharnametestr! r )r rF ZsinZsoutr )r; r �test_customreplace_encodeS s
z"TestBase.test_customreplace_encodec C s> dd� }t �d|� | �d| j d d�d }| �|d� d S )Nc S s
d| j fS )Ns 1234)r@ )rC r r r � myreplaceo s z7TestBase.test_callback_returns_bytes.<locals>.myreplacez test.cjktest�abc�defr s
abc1234def)r rG r �unmappedunicoder! )r rJ �encr r r �test_callback_returns_bytesn s z$TestBase.test_callback_returns_bytesc sN � fdd�}t �d|� x0dddgg d t� fD ]� | �t| j| jd� q.W d S )Nc s
� | j fS )N)r@ )rC )�retr r rJ v s z7TestBase.test_callback_wrong_objects.<locals>.myreplacez test.cjktestr � � )r rG r> r. r= r rM )r rJ r )rP r �test_callback_wrong_objectsu s
z$TestBase.test_callback_wrong_objectsc C s^ dd� }t �d|� | �| �d| j d d�d� dd� }t �d|� | �t| j| jd� d S )Nc S s dt | j�fS )N�x)�intr@ )rC r r r rJ s z4TestBase.test_callback_long_index.<locals>.myreplacez test.cjktest�abcd�efgh)s abcdxefgh� c S s dt jd fS )NrT r )�sys�maxsize)rC r r r rJ � s )r rG r! r rM r. �
IndexError)r rJ r r r �test_callback_long_index~ s
z!TestBase.test_callback_long_indexc C s, dd� }t �d|� | �t| j| jd� d S )Nc S s dS )N)rT Nr )rC r r r rJ � s z4TestBase.test_callback_None_index.<locals>.myreplacez test.cjktest)r rG r. r= r rM )r rJ r r r �test_callback_None_index� s z!TestBase.test_callback_None_indexc s@ � fdd�� d� _ t�d� � | �| �d| j d d�d� d S )Nc s* � j dkr � j d8 _ dS d| jfS d S )Nr r )�REPLACEDr ZTERMINAL)�limitr@ )rC )rJ r r rJ � s
z8TestBase.test_callback_backward_index.<locals>.myreplacerR z test.cjktestrV rW )s4 abcdREPLACEDabcdREPLACEDabcdREPLACEDabcdTERMINALefghrX )r_ r rG r! r rM )r r )rJ r � test_callback_backward_index� s z%TestBase.test_callback_backward_indexc C s6 dd� }t �d|� | �| �d| j d d�d� d S )Nc S s d| j d fS )Nr^ rQ )r@ )rC r r r rJ � s z7TestBase.test_callback_forward_index.<locals>.myreplacez test.cjktestrV rW )s abcdREPLACEDghrX )r rG r! r rM )r rJ r r r �test_callback_forward_index� s z$TestBase.test_callback_forward_indexc C s, dd� }t �d|� | �t| j| jd� d S )Nc S s dS )N)ZTERM�d r )rC r r r rJ � s z:TestBase.test_callback_index_outofbound.<locals>.myreplacez test.cjktest)r rG r. r[ r rM )r rJ r r r � test_callback_index_outofbound� s z'TestBase.test_callback_index_outofboundc C s� t �d�}x�d gttdd�� dddddg D ]v}|t| jd ��}t� }| �� }x:|d k rh|�|�}n|�� }|svP |�|�}|� |� qTW | �
|� � | 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 �d�}x�d dgttdd�� ddddd g D ]f}t| jd
�}|t� �}| �� }x*|�|�}|shP qV|�|�}|� |� qVW | �
|� � | jd � q0W d S ) Nzutf-8�����r rd re rf rg i i r ) r � getwriterri rj r r r rk r
rl r! rm )r �
UTF8Writerro rp rq �decoderrs r& r r r �test_incrementaldecoder� s
z TestBase.test_incrementaldecoderc C s� | j }| �� }| �t|j|d� d|_| �|�|d�d� |�� dd� }t� d|� d|_| �|�|d�d� d|_| �|�|d�d� d S )NT�ignore� c S s
d| j fS )NZcalled)r@ )rC r r r � tempreplace� s zDTestBase.test_incrementalencoder_error_callback.<locals>.tempreplacez test.incremental_error_callbacks called)
rM r r. r< r �errorsr! �resetr rG )r �invrt r} r r r �&test_incrementalencoder_error_callback� s z/TestBase.test_incrementalencoder_error_callbackc C s� t �d�}x�dD ]�}x�d dgttdd�� dddd d
g D ]t}| �t| jd ��}|t� �}t||�}x0||�}|svP |d kr�|�|� qh|� |� qhW | �
|� � | jd � q:W qW d S )
Nzutf-8)rk �readline� readlinesrv r rd re rf rg i i r r� ) r rw ri rj r r r �getattr�
writelinesrl r! rm )r rx �namero rp rq r3 rs r r r �test_streamreader� s
zTestBase.test_streamreaderc C s� d}t �d�}x�|D ]�}x�d gttdd�� ddddd g D ]�}|t| jd ��}| �t� �}t||�}x@|d k r|||�}n|� }|s�P |d
kr�|�|� qj|� |� qjW | �
|� � | jd � q<W qW d S ) N)rk r� r� zutf-8r rd re rf rg i i r� r ) r rh ri rj r r r r� r� rl r! rm ) r Z readfuncsrn r� ro rp rq r3 rs r r r �test_streamwriter� s$
zTestBase.test_streamwriterc C s t � }| �|�}|�� d S )N)r r r )r �streamr r r r �"test_streamwriter_reset_no_pending s
z+TestBase.test_streamwriter_reset_no_pendingc C s& | � � }| �t��
|`W d Q R X d S )N)r r. �AttributeErrorr~ )r rt r r r �$test_incrementalencoder_del_segfault s z-TestBase.test_incrementalencoder_del_segfault) �__name__�
__module__� __qualname__r r r r( r" r7 rH rM r r' r5 r: rI rO rS r\ r] r` ra rc ru rz r� r� r� r� r� r r r r r s6
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| � � �� W n( ttfk
r8 | �d| j � Y nX d S )NzCould not retrieve )�open_mapping_file�close�OSErrorr r8 �
mapfileurl)r r r r r $ s zTestBase_Mapping.setUpc C s t �| j�S )N)r Zopen_urlresourcer� )r r r r r� * s z"TestBase_Mapping.open_mapping_filec C s" | j �d�r| �� n| �� 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 s8 dd� }i }| � � ��}�x|D �]}|s,P |�d�d �� �� }t|�dkrPq t|d �}|dkrtt|d@ g�}nz|dkr�t|d ? |d
? d@ |d ? d@ |d@ g�}nH|d kr�t|d
? |d ? d@ |d@ g�}n |d
kr t|d ? |d@ g�}nq ||d �}t|�dks ||k�rq |||< | �||� q W W d Q R X d S )Nc S s d� tttt| �d����S )Nr �+)rB �map�chr�evalr )r9 r r r �<lambda>4 r| z;TestBase_Mapping._test_mapping_file_plain.<locals>.<lambda>�#r rQ � �� i � � � i rg r i�� )r� r �strip�lenr� r* rA �
_testpoint) r ZunichrsZurt_wa�fr rs Zcsetval�csetch�unichr r r r� 3 s4 z)TestBase_Mapping._test_mapping_file_plainc C sb | � � �}|�� }W d Q R X t�d|�}x4|D ],\}}tt|d��}t�|�}| �||� q.W d S )Nz'<a u="([A-F0-9]{4})" b="([0-9A-F ]+)"/>r� ) r� rk �re�findallr� rU r* �fromhexr� )r r� ZucmdataZucZuniZcodedr� Zcodechr r r r� T 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_supplemental] s z*TestBase_Mapping.test_mapping_supplementalc C sH ||f| j kr"| �|�| j�|� ||f| jkrD| �t|| j�|� d S )N)� pass_enctestr! r r � pass_dectestr- )r r� r� r r r r� a s zTestBase_Mapping._testpointc C s� x�| j D ]�\}}}t|t�r$|j}n|j}|r�t|t�r||| j|�}| �t|�tkt|�� | � ||d|| j|||f � q�|| j|�}| �t|�tkt|�� | � ||d|| j|||f � q| �
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! r. r/ )r r0 r1 r2 r3 r4 r r r r5 g s&
z!TestBase_Mapping.test_errorhandleN)r� r� r� r� r� r� r( r r� r� r� r� r� r� r5 r r r r r� s ! r� c C sv t j�t j�t�d�}tt j�|| d �d��}|�� }W d Q R X tt j�|| d �d��}|�� }W d Q R X ||fS )NZ cjkencodingsz.txt�rbz -utf8.txt)�os�pathrB �dirname�__file__�openrk )r� �dirr� Zencodedr% r r r �load_teststring s r� )r r� r� rY ZunittestZ http.clientr Ztestr �ior r ZTestCaser� r� r r r r �<module> s a