File: //usr/local/lib/python3.10/test/__pycache__/multibytecodec_support.cpython-310.pyc
o
�i�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 d*d+� Z dS ),�TestBase� N� r u �c C sX | j d u r t�| 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 �8/usr/local/lib/python3.10/test/multibytecodec_support.py�setUp s
zTestBase.setUpc C s� g }| j D ] }|�d�}|�� }|dksJ �dd� |D �}|�|� qt|� D ]"\}}| �|�d }| �||�d�� | jrI| �|| �|�d � q'd S )N�
� c S s g | ]}|d �qS )r r )�.0�liner r r �
<listcomp>* s z-TestBase.test_chunkcoding.<locals>.<listcomp>r �utf-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� | j D ]_\}}}t|t�r| j}n| j}|rZ|||�d }|| ju r>| �t|�tu t|�� | �||d|| j |||f � q| �t|�tu t|�� | �||d|| j |||f � q| �
t |||� qd 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 �source�schemeZexpected�func�resultr r r �test_errorhandle2 s2
���
����zTestBase.test_errorhandlec C s0 | j r| �d� d}| �| �|d�d d� d S )N�$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|� | jr#| j\}}nd}d}| �| �|d�d |� d S ) Nr< r ��codepoint2namec sx t | t�s td| ��g }| j| j| j� D ] }t|�� v r*|�d� t|� � q|�dt|� � qd�|�| jfS )Nzdon't know how to handle %rz&%s;z&#%d;r ) r/ �UnicodeEncodeError� TypeError�object�start�end�ordr# �join)�exc�l�crB r r �xmlcharnamereplaceY s
z>TestBase.test_customreplace_encode.<locals>.xmlcharnamereplaceztest.xmlcharnamereplaceu «ℜ» = 〈ሴ〉s* «ℜ» = ⟨ሴ⟩) r> r? Z
html.entitiesrC r
�register_error�xmlcharnametestr% r
)r rN �sinZsoutr rB 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�rH �rK r r r � myreplaceo �
z7TestBase.test_callback_returns_bytes.<locals>.myreplace� test.cjktest�abc�defr s
abc1234def)r
rO r
�unmappedunicoder% )r rV �encr r r �test_callback_returns_bytesn s z$TestBase.test_callback_returns_bytesc sH � fdd�}t �d|� g d�g d t� fD ] � | �t| j| jd� qd S )Nc s
� | j fS r rT rU ��retr r rV v rW z7TestBase.test_callback_wrong_objects.<locals>.myreplacerX )r � � )r
rO rF r4 rE r
r[ �r rV r r^ 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)�intrH rU r r r rV � z4TestBase.test_callback_long_index.<locals>.myreplacerX �abcd�efgh)s abcdxefgh� c S s dt jd fS )Nrd r )�sys�maxsizerU r r r rV � rf )r
rO r% r
r[ r4 �
IndexErrorrb 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)rd Nr rU r r r rV � � z4TestBase.test_callback_None_index.<locals>.myreplacerX )r
rO r4 rE r
r[ rb 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 )Nr r )�REPLACEDr ZTERMINAL)�limitrH rU �rV r r rV � s
z8TestBase.test_callback_backward_index.<locals>.myreplacera rX rg rh )s4 abcdREPLACEDabcdREPLACEDabcdREPLACEDabcdTERMINALefghri )rt r
rO r% r
r[ r r ru 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 )Nrs r` rT rU r r r rV � rf z7TestBase.test_callback_forward_index.<locals>.myreplacerX rg rh )s abcdREPLACEDghri )r
rO r% r
r[ rb r r r �test_callback_forward_index� s ��z$TestBase.test_callback_forward_indexc C rn )Nc S ro )N)ZTERM�d r rU r r r rV � rp z:TestBase.test_callback_index_outofbound.<locals>.myreplacerX )r
rO r4 rl r
r[ rb r r r � test_callback_index_outofbound� rr z'TestBase.test_callback_index_outofboundc C s� t �d�}d gttdd�� g d� D ]:}|t| jd ��}t� }| �� } |d ur0|�|�}n|�� }|s7n |�|�}|� |� q&| �
|� � | jd � qd S )Nr r �! ��@ � � 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�}d dgttdd�� g d� D ]1}t| jd �}|t� �}| �� } |�|�}|s/n |�|�}|� |� q'| �
|� � | jd � qd S )Nr ���r rz r{ r ) r
� getwriterr� r� r r r r� r r� r% r� )r �
UTF8Writerr� r� r� �decoderr� 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�ignorer c S rS )NZcalledrT rU r r r � tempreplace� rW zDTestBase.test_incrementalencoder_error_callback.<locals>.tempreplacez test.incremental_error_callbacks called)
r[ r r4 rD r
�errorsr% �resetr
rO )r �invr� r� r r r �&test_incrementalencoder_error_callback� s z/TestBase.test_incrementalencoder_error_callbackc C s� t �d�}dD ]K}d dgttdd�� g d� D ]9}| �t| jd ��}|t� �}t||�} ||�}|s6n|dkr@|�|� n|� |� q/| �
|� � | jd � qqd S ) Nr �r� �readline� readlinesr� r rz r{ r r� ) r
r� r� r� r r r �getattr�
writelinesr� r% r� )r r� �namer� r� r� r9 r� r r r �test_streamreader� s&
�
� ��zTestBase.test_streamreaderc C s� d}t �d�}|D ]R}d gttdd�� g d� D ]A}|t| jd ��}| �t� �}t||�} |d ur9||�}n|� }|s?n|dkrI|�|� n|� |� q0| �
|� � | jd � qq d S )Nr� r r rz r{ r� r ) r
r r� r� r r r r� r� r� r% r� ) r Z readfuncsr� r� r� r� r� r9 r� r r r �test_streamwriter� s,
�
�
��zTestBase.test_streamwriterc C s t � }| �|�}|�� d S r )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 � d S 1 sw Y d S r )r r4 �AttributeErrorr� )r r� r r r �$test_incrementalencoder_del_segfault s "�z-TestBase.test_incrementalencoder_del_segfault) �__name__�
__module__� __qualname__r r r r. r& r> rP r[ r r+ r; rA rR r] rc rm rq rv rw ry r� r� r� r� r� r� r� r r r r r s8
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< z | � � �� W d S ttfy | �d| j � Y d S w )NzCould not retrieve )�open_mapping_file�close�OSErrorr r? �
mapfileurlr r r r r $ s
�zTestBase_Mapping.setUpc C s t j| jdd�S )Nr )r )r Zopen_urlresourcer� r r r r r� * s z"TestBase_Mapping.open_mapping_filec C s$ | j �d�r | �� d S | �� d S )Nz.xml)r� �endswith�_test_mapping_file_ucm�_test_mapping_file_plainr r r r �test_mapping_file- s z"TestBase_Mapping.test_mapping_filec C s dd� }i }| � � �o}|D ][}|s n^|�d�d �� }t|�dkr#q
|d d d� dkr5| �d|��� t�|d dd � �}t|�dkrMd |d krMq
||d �}t|�d
ks]||v r^q
|||< | �||� q
W d � d S W d � d S 1 s|w Y d S ) Nc S s d� dd� | �d�D ��S )Nr c s s � | ]
}t t|d ��V qdS )� N)�chrre )r rd r r r � <genexpr>5 s � zMTestBase_Mapping._test_mapping_file_plain.<locals>.unichrs.<locals>.<genexpr>�+)rJ r! )r@ r r r �unichrs4 s z:TestBase_Mapping._test_mapping_file_plain.<locals>.unichrs�#r r` �0xzInvalid line: r r} i�� )r� r! �lenZfailr0 �fromhexrI �
_testpoint)r r� Zurt_wa�fr r� �csetch�unichr r r r� 3 s.
��"�z)TestBase_Mapping._test_mapping_file_plainc C sr | � � � }|�� }W d � n1 sw Y t�d|�}|D ]\}}tt|d��}t�|�}| �||� q d S )Nz'<a u="([A-F0-9]{4})" b="([0-9A-F ]+)"/>r� ) r� r� �re�findallr� re r0 r� r� )r r� ZucmdataZucZuniZcodedr� Zcodechr r r r� N s
�
�z'TestBase_Mapping._test_mapping_file_ucmc C s | j D ]}| j|� qd S r )�supmapsr� )r �mappingr r r �test_mapping_supplementalW s
�z*TestBase_Mapping.test_mapping_supplementalc C sL ||f| j vr| �|�| j�|� ||f| jvr$| �t|| j�|� d S d S r )� pass_enctestr% r
r � pass_dectestr3 )r r� r� r r r r� [ s
�zTestBase_Mapping._testpointc C s� | j D ]e\}}}t|t�r|j}n|j}|r_t|t�r=|| j|�}| �t|�tu t|�� | � ||d|| j|||f � q|| j|�}| �t|�tu t|�� | � ||d|| j|||f � q| �
t || j|� qd S )Nr, r- ) r. r/ r0 r r
r r1 r2 r3 r% r4 r5 r6 r r r r; a s4
���
����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 s� t j�t j�t�d�}tt j�|| d �d�� }|�� }W d � n1 s&w Y tt j�|| d �d��}|�� }W d � ||fS 1 sIw Y ||fS )NZ cjkencodingsz.txt�rbz -utf8.txt)�os�pathrJ �dirname�__file__�openr� )r� �dirr� �encodedr) r r r �load_teststringy s
�
��r� )r
r� r� rj ZunittestZ http.clientr �testr �ior r ZTestCaser� r� r r r r �<module> s [