B
�/g^N: c @ 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�codecsZlookup�encoding�encode�decodeZ streamreader�readerZ 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 )Z.0�liner r r z
<listcomp>* s z-TestBase.test_chunkcoding.<locals>.<listcomp>r �utf-8) �tstring�splitZpop�appendZzipr � assertEqual�
roundtriptestr )r Z
tstring_linesZbZlinesZlastZnative�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 �%a.decode(%r, %r)=%a != %a�%a.encode(%r, %r)=%a != %a) �
codectests�
isinstance�bytesr r �
assertTrue�type�strr r
� assertRaises� UnicodeError�r ZsourceZschemeZexpected�funcZresultr r r �test_errorhandle2 s$
zTestBase.test_errorhandlec C s0 | j r| �d� d}| �| �|d�d d� d S )N�$encoding contains full ISO 10646 mapu ଓଣୠ nd eggsZxmlcharrefreplacer 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 ) Nr. r ��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�objectZstart�end�ordr �join)�excZlZcr3 r r �xmlcharnamereplaceY s
z>TestBase.test_customreplace_encode.<locals>.xmlcharnamereplaceztest.xmlcharnamereplaceu «ℜ» = 〈ሴ〉s* «ℜ» = ⟨ሴ⟩) r/ r0 Z
html.entitiesr4 r �register_error�xmlcharnametestr r )r r< ZsinZsoutr r3 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 �
d| j fS )Ns 1234�r8 �r; r r r � myreplaceo � z7TestBase.test_callback_returns_bytes.<locals>.myreplace� test.cjktestZabcZdefr s
abc1234def)r r= r �unmappedunicoder )r rC Zencr 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 r rA rB �Zretr r rC v rD z7TestBase.test_callback_wrong_objects.<locals>.myreplacerE r � � )r r= r7 r) r6 r rF �r rC r rH 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)�intr8 rB r r r rC rD z4TestBase.test_callback_long_index.<locals>.myreplacerE �abcd�efgh)s abcdxefgh� c S s dt jd fS )NrM r )�sysZmaxsizerB r r r rC � rD )r r= r r rF r) �
IndexErrorrK r r r �test_callback_long_index~ s
z!TestBase.test_callback_long_indexc C �, dd� }t �d|� | �t| j| jd� d S )Nc S � dS )N)rM Nr rB r r r rC � rD z4TestBase.test_callback_None_index.<locals>.myreplacerE )r r= r) r6 r rF rK r r r �test_callback_None_index� � 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)�limitr8 rB �rC r r rC � s
z8TestBase.test_callback_backward_index.<locals>.myreplacerJ rE rO rP )s4 abcdREPLACEDabcdREPLACEDabcdREPLACEDabcdTERMINALefghrQ )rZ r r= r r rF r r r[ 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 )NrY rI rA rB r r r rC � rD z7TestBase.test_callback_forward_index.<locals>.myreplacerE rO rP )s abcdREPLACEDghrQ )r r= r r rF rK r r r �test_callback_forward_index� s z$TestBase.test_callback_forward_indexc C rU )Nc S rV )N)ZTERMid r rB r r r rC � rD z:TestBase.test_callback_index_outofbound.<locals>.myreplacerE )r r= r) rS r rF rK r r r � test_callback_index_outofbound� rX 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 )
Nr r �! �@ � � � � r ) r � getreader�list�ranger r r �readr �writer �getvalue)r �
UTF8Reader�sizehint�istream�ostreamZencoder�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 ) Nr �����r r_ r` ra rb rc rd r ) r � getwriterrf rg r r r rh r ri r rj )r �
UTF8Writerrl rm rn Zdecoderro 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 )NTZignore� c S r@ )NZcalledrA rB r r r � tempreplace� rD zDTestBase.test_incrementalencoder_error_callback.<locals>.tempreplacez test.incremental_error_callbacks called)
rF r r) r5 r �errorsr �resetr r= )r Zinvrp rw 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 )
Nr �rh Zreadline� readlinesrr r r_ r` ra rb rc rd r r| ) r rs rf rg r
r r �getattr�
writelinesri r rj )r rt �namerl rm rn r, ro 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 ) Nr{ r r r_ r` ra rb rc rd r| r ) r re rf rg r r r r} r~ ri r rj ) r Z readfuncsrk r rl rm rn r, ro r r r �test_streamwriter� s$
zTestBase.test_streamwriterc C s t � }| �|�}|�� d S r )r r ry )r Zstreamr 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 r )r r) ZAttributeErrorrx )r rp r r r �$test_incrementalencoder_del_segfault s z-TestBase.test_incrementalencoder_del_segfault) �__name__�
__module__� __qualname__r
r r r# r r/ r> rF r r r- r2 r? rG rL rT rW r\ r] r^ rq ru rz 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_fileZcloseZOSErrorr r0 �
mapfileurlr r r r r $ s zTestBase_Mapping.setUpc C s t �| j�S r )r Zopen_urlresourcer� r r r r r� * rD z"TestBase_Mapping.open_mapping_filec C s" | j �d�r| �� n| �� d S )Nz.xml)r� Zendswith�_test_mapping_file_ucm�_test_mapping_file_plainr 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 z+)r: Zmap�chr�evalr )r1 r r r Z<lambda>4 rv z;TestBase_Mapping._test_mapping_file_plain.<locals>.<lambda>z#r rI i i� i i � i i rb r i�� )r� r ZstripZlenr� r% r9 �
_testpoint) r ZunichrsZurt_wa�fr ro 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� rh �reZfindallr� rN r% Zfromhexr� )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 r )�supmapsr� )r Zmappingr r r �test_mapping_supplemental] s z*TestBase_Mapping.test_mapping_supplementalc C sH ||f| j kr"| �|�| j�|� ||f| jkrD| �t|| j�|� d S r )� 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 )Nr! r" ) r# r$ r% r r r
r& r' r( r r) r* r+ r r r r- g s&
z!TestBase_Mapping.test_errorhandleN)r� r� r� r� r� r� r# r r� r� r� r� r� r� r- 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.txtZrbz -utf8.txt)�osZpathr: ZdirnameZ__file__Zopenrh )r Zdirr� Zencodedr r r r �load_teststring s r� )r r� r� rR ZunittestZ http.clientr Ztestr Zior r ZTestCaser� r� r r r r Z<module> s a