File: //usr/local/lib/python3.7/test/__pycache__/test_nntplib.cpython-37.opt-2.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZmZ d dl
Z
d dlmZ y d dlZW n ek
r� dZY nX dZej�ej�e�d�Zedk r�ejZnG dd� de�ZG d d
� d
�Z e � � e fZ edk �re ej f7 Z G d d � d e ej!�Z"e�#ed
�G dd� de"��Z$G dd� de j%�Z&dd� Z'G dd� d�Z(G dd� de(�Z)G dd� d�Z*G dd� de*�Z+G d d � d e+�Z,G d d � d e+�Z-G d d!� d!�Z.G d"d#� d#e.e(ej!�Z/G d$d%� d%e.e(ej!�Z0G d&d'� d'e(ej!�Z1G d(d)� d)e)ej!�Z2G d*d+� d+ej!�Z3G d,d-� d-ej!�Z4G d.d/� d/ej!�Z5G d0d1� d1�Z6e�#ed
�G d2d3� d3e5��Z7G d4d5� d5ej!�Z8e9d6k�r�e�:� dS )7� N)�support)�NNTP� GroupInfo)�patch� z keycert3.pemc @ s e Zd ZdZdS )�SSLErrorzThis will never be raised.N)�__name__�
__module__� __qualname__�reason� r r �-/usr/local/lib/python3.7/test/test_nntplib.pyr 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� Ze�dd�dd � �Zd d � Zd/d d!�Ze�dd"�d#d$� �Zd%d&� Zd'd(� Zd)d*� Zed+d,� �Zd-d.� Zd S )0�NetworkedNNTPTestsMixinc C s | j �� }| �tt|�� d S )N)�serverZ
getwelcome� assertEqual�str�type)�self�welcomer r r
� test_welcome( s
z$NetworkedNNTPTestsMixin.test_welcomec C sB | j �� \}}| �|�d�|� x |D ]}| �tt|�� q&W d S )Nz100 )r �help�
assertTrue�
startswithr r r )r �resp�lines�liner r r
� test_help, s
z!NetworkedNNTPTestsMixin.test_helpc C sH | j �� \}}t|�dkrD| �tt|d �� | �tt|d j�� d S )Nr )r �list�lenr r r r �group)r r �groupsr r r
� test_list2 s z!NetworkedNNTPTestsMixin.test_listc C sL | j �| j�\}}t|�dkrH| �tt|d �� | �tt|d j�� d S )Nr ) r r � GROUP_PATr r r r r r )r r r r r r
�test_list_active8 s z(NetworkedNNTPTestsMixin.test_list_activec C sB | � tj��}| j�d� W d Q R X |jj}| �|�d�|� d S )NZXYZZYz500 ) � assertRaises�nntplib�NNTPPermanentErrorr Z _shortcmd� exception�responser r )r �cmr r r r
�test_unknown_command> s z,NetworkedNNTPTestsMixin.test_unknown_commandc C sX t j�� t jdd� }| j�|�\}}t|�dkrT| �|d t� | �|d j t
� d S )N� )Zdaysr ) �datetime�dateZtodayZ timedeltar Z newgroupsr �assertIsInstancer r r )r �dtr r r r r
�test_newgroupsD s
z&NetworkedNNTPTestsMixin.test_newgroupsc sb � fdd�}� j �� j�}||� � �� j|� � j �� j�}||� � j �d�}� �|d� d S )Nc s � � | t� � �� j| � d S )N)r. r � assertNotIn�
GROUP_NAME)�desc)r r r
� _check_descN s z=NetworkedNNTPTestsMixin.test_description.<locals>._check_descz zk.brrtt.baz� )r � descriptionr2 �assertIn�DESCr" r )r r4 r3 r )r r
�test_descriptionM s z(NetworkedNNTPTestsMixin.test_descriptionc C s^ | j �| j�\}}| �|�d�p(|�d�|� | �|t� || j }| �|| j � | j�� d S )Nz215 z282 )
r � descriptionsr" r r r. �dictr2 r r6 )r r Zdescsr3 r r r
�test_descriptions] s
z)NetworkedNNTPTestsMixin.test_descriptionsc C s� | j �| j�}| �dt|�� |\}}}}}| �|| j� | �|t� | �|t� | �|t� | �||� | �|� d�|� d S )N� z211 )
r r r2 r r r. �int�assertLessEqualr r )r �resultr �count�first�lastr r r r
�
test_groupf s z"NetworkedNNTPTestsMixin.test_groupc C s< | j �� \}}| �|tj� | �|jd� | �|jd� d S )Ni� i� )r r- r. r, �assertGreaterEqualZyearr? )r r r- r r r
� test_dateq s z!NetworkedNNTPTestsMixin.test_datec
C sT | � |t� | �|�� dddddddh� x$|�� D ]}| � |ttd �f� q4W d S )N�subject�fromr- z
message-id�
referencesz:bytesz:lines)r. r; rE �keys�valuesr r )r �art_dict�vr r r
�_check_art_dictx s
z'NetworkedNNTPTestsMixin._check_art_dictc C sz | j �| j�\}}}}}| j �|d |�\}}t|�dkrD| �d� |d \}}| �||d � | �||� | �|� d S )Nr= r zno articles retrieved) r r r2 �xoverr �skipTestrE r? rN ) r r rA rB rC �namer �art_numrL r r r
�
test_xover� s
z"NetworkedNNTPTestsMixin.test_xoverTzHtemporarily skipped until a permanent solution is found for issue #28971c
C s� | j �| j�\}}}}}|d }| j �|d f�\}}|d \}} | �| � | j �||f�\}}|d \}} | �||� | �||� | �| � d S )N�
r ���)r r r2 �overrN rE r? )
r r rA rB rC rQ �startr rR rL r r r
� test_over� s
z!NetworkedNNTPTestsMixin.test_overc C sP | j �| j�\}}}}}| j �d|�\}}x |D ]}| �tt|d �� q0W d S )NrG � )r r r2 Zxhdrr r r )r r rA rB rC rQ r r r r r
� test_xhdr� s
z!NetworkedNNTPTestsMixin.test_xhdrNc C sV | � |tj� |d k r$| �|j|� x|jD ]}| � |t� q,W | �|jd d� d S )NrU )� .s .
s .
)r. r% Z ArticleInfor Znumberr �bytesr1 )r r �articlerR r r r r
�check_article_resp� s z*NetworkedNNTPTestsMixin.check_article_respzFIXME: see bpo-32128c
sR | j �| j�\}}}}}xl|||d fD ]P}y| j �|�\}}W n4 tjk
rt } z|j�d�sb� w(W d d }~X Y nX P q(W | �d� | � |�d�|� | �
|||� | j � |�\}} | � |�d�|� | �
|| |� | j � |�\}}
| � |�d�|� | �
||
|� dd� � � fd d
�|j
D �} � fd d
�|
j
D �} | �| | d g | j
� d S )
NrY z423 z(could not find a suitable article numberz221 z222 z220 c S s
| � d�S )Ns X-Antivirus)r )r r r r
�<lambda>� � z@NetworkedNNTPTestsMixin.test_article_head_body.<locals>.<lambda>c s g | ]}� |�s|�qS r r )�.0r )� blacklistr r
�
<listcomp>� s zBNetworkedNNTPTestsMixin.test_article_head_body.<locals>.<listcomp>c s g | ]}� |�s|�qS r r )ra r )rb r r
rc � s r` )r r r2 �headr% �NNTPTemporaryErrorr( r rP r r^ �bodyr] r r )
r r rA rB rC rQ rR rd �erf r] Zfiltered_head_linesZfiltered_linesr )rb r
�test_article_head_body� s,
z.NetworkedNNTPTestsMixin.test_article_head_bodyc sD � fdd�}� � � jjd� |� j�� � � j�� \}}||� d S )Nc s( | d }� � |ttf� � �d|� d S )N�LISTz OVERVIEW.FMT)r. r �tupler7 )�capsZ caps_list)r r r
� _check_caps� s z>NetworkedNNTPTestsMixin.test_capabilities.<locals>._check_caps� )rE r � nntp_version�getcapabilitiesZ capabilities)r rl r rk r )r r
�test_capabilities� s
z)NetworkedNNTPTestsMixin.test_capabilitiesc C s& d}d}| j tj| jj||dd� d S )NZ notarealuserZnotarealpasswordF)�user�password�usenetrc)r$ r% � NNTPErrorr �login)r ZbaduserZbadpwr r r
� test_zlogin� s z#NetworkedNNTPTestsMixin.test_zloginc C s$ t | �}z| j�� W d d |_X d S )N)r r �quit)r �clsr r r
� test_zzquit� s z#NetworkedNNTPTestsMixin.test_zzquitc C sN dd� }x@t | �D ]4}|�d�s"qt| |�}t|�s6qt| |||�� qW d S )Nc s t �� �� fdd��}|S )Nc s$ t �| j�� � | � W d Q R X d S )N)r �transient_internet� NNTP_HOST)r )�methr r
�wrapped� s zHNetworkedNNTPTestsMixin.wrap_methods.<locals>.wrap_meth.<locals>.wrapped)� functools�wraps)r| r} r )r| r
� wrap_meth� s z7NetworkedNNTPTestsMixin.wrap_methods.<locals>.wrap_methZtest_)�dirr �getattr�callable�setattr)rx r� rQ r| r r r
� wrap_methods� s
z$NetworkedNNTPTestsMixin.wrap_methodsc
s� � fdd�}yz| j | jtdd�� � | �|� � | �� �� � W d Q R X | �|� � | j | jtdd��� � �� W d Q R X | �|� � W nJ tk
r� } z,t� d|j
�r�t � d|� d| j����� W d d }~X Y nX d S )Nc s8 t � d�sdS y � �� W n ttfk
r2 dS X dS )N�fileFT)�hasattrr �OSError�EOFErrorr )r r r
� is_connected s
zANetworkedNNTPTestsMixin.test_with_statement.<locals>.is_connectedF)�timeoutrs z(?i)KEY.TOO.SMALLzGot z connecting to )
�
NNTP_CLASSr{ �TIMEOUTr r � assertFalserw r �re�searchr �unittest�SkipTest)r r� �ssl_errr )r r
�test_with_statement s z+NetworkedNNTPTestsMixin.test_with_statement)N)r r r
r r r! r# r* r0 r9 r< rD rF rN rS r� ZskipIfrX rZ r^ rh rp rv ry � classmethodr� r� r r r r
r & s*
r c @ s8 e Zd ZdZdZdZdZeZe dd� �Z
e dd� �Z d S )
�NetworkedNNTPTestsznews.trigofacile.comzfr.comp.lang.pythonzfr.comp.lang.*�Pythonc C s� t �d� t �| j��� y| j| jtdd�| _W nv tk
r~ } z0t� d|j
�rlt � | � d|� d| j����� W d d }~X Y n* t
k
r� t � | � d| j����Y nX W d Q R X d S )NZnetworkF)r� rs z(?i)KEY.TOO.SMALLz got z connecting to z got EOF error on connecting to )r Zrequiresrz r{ r� r� r r r� r� r r� r� �
EOF_ERRORS)rx r� r r r
�
setUpClass2 s
z NetworkedNNTPTests.setUpClassc C s | j d k r| j �� d S )N)r rw )rx r r r
�
tearDownClassC s
z NetworkedNNTPTests.tearDownClassN) r r r
r{ r2 r" r8 r r� r� r� r� r r r r
r� ) s r� zrequires SSL supportc @ s0 e Zd ZdZdZdZdZeedd�Z dZ
dZ dS )�NetworkedNNTP_SSLTestsz
nntp.aioe.orgz aioe.testzaioe.*�test�NNTP_SSLN) r r r
r{ r2 r" r8 r� r% r� r! �
test_starttlsr r r r
r� H s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
_NNTPServerIOc C s> t j�| � t �� | _t �� | _|| _| j�| jj| j � d S )N)
�io� RawIOBase�__init__�BytesIO�c2s�s2c�handlerrW �readline� push_data)r r� r r r
r� h s
z_NNTPServerIO.__init__c C s dS )NTr )r r r r
�readableq s z_NNTPServerIO.readablec C s dS )NTr )r r r r
�writablet s z_NNTPServerIO.writablec C s4 | j �� }| j �dd� | j �|� | j �|� d S )Nr rm )r� �tell�seek�write)r �data�posr r r
r� w s
z_NNTPServerIO.push_datac C s4 | j �� }| j �|� | j �|� | j�� t|�S )N)r� r� r� r� r� �process_pendingr )r �br� r r r
r� ~ s
z_NNTPServerIO.writec C s2 | j �� | j�t|��}t|�}||d |�< |S )N)r� r� r� �readr )r �bufr� �nr r r
�readinto� s
z_NNTPServerIO.readintoN) r r r
r� r� r� r� r� r� r r r r
r� c s r� c C s t | �}t�||�}||fS )N)r� r� �BufferedRWPair)r� �sior� r r r
�make_mock_file� s r� c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )�MockedNNTPTestsMixinNc s t � �� | �� d S )N)�super�setUp� make_server)r )� __class__r r
r� � s
zMockedNNTPTestsMixin.setUpc s t � �� | `d S )N)r� �tearDownr )r )r� r r
r� � s
z MockedNNTPTestsMixin.tearDownc O s6 | � � | _t| j�\| _}tj|df|�|�| _| jS )Nz test.server)�
handler_classr� r� r� r% Z _NNTPBaser )r �args�kwargsr� r r r
r� � s
z MockedNNTPTestsMixin.make_server)r r r
r� r� r� r� �
__classcell__r r )r� r
r� � s r� c s e Zd Z� fdd�Z� ZS )� MockedNNTPWithReaderModeMixinc s t � �� | jdd� d S )NT)Z
readermode)r� r� r� )r )r� r r
r� � s
z#MockedNNTPWithReaderModeMixin.setUp)r r r
r� r� r r )r� r
r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd7dd �Zd d � Zd d � Zd8d!d"�Zd#d$� Zd%d&� Zdd'�d(d)�Zdd'�d*d+�Zd,Zd-Zed. e Zd9d/d0�Zd:d1d2�Z d;d3d4�Z d5d6� Z dS )<�
NNTPv1Handlerz200 NNTP mock serverc C s0 d| _ d| _|| _|| _d| _d| _| �� d S )NFT)�in_body�
allow_posting� _readline�
_push_data�
_logged_in�
_user_sent�handle_welcome)r r� r� r r r
rW � s zNNTPv1Handler.startc C s t |dd�S )Nzutf-8�surrogateescape)r )r r� r r r
�_decode� s zNNTPv1Handler._decodec
C s4 | j rfx*| �� }|sd S | j�|� |dkrP qW z | j\}}||d| ji� W d d | _d | _d| _ X x�| �| �� �}|s~d S |�d�s�td�|���|d d� }|� � ^}}t
| d|� � d �}|d kr�| � � qhy ||� W n4 t
k
�r } ztd�|��|�W d d }~X Y qhX | j rh||f| _g | _qhW d S ) Ns .
rf Fz
z line doesn't end with \r\n: {!r}���Zhandle_zcommand failed: {!r})r� r� rf �appendZ
body_callbackr� �endswith�
ValueError�format�splitr� �upper�handle_unknown� Exception)r r r| �tokens�cmdrg r r r
r� � s@
"
z NNTPv1Handler.process_pendingc C s
d| _ d S )NT)r� )r r r r
� expect_body� s zNNTPv1Handler.expect_bodyc C s | � |� d S )N)r� )r r� r r r
r� � s zNNTPv1Handler.push_datac C s4 t �|�}d�|�� �d }|�d�}| �|� d S )Nz
zutf-8)�textwrap�dedent�join�
splitlines�encoder� )r Zlitr r r
�push_lit� s
zNNTPv1Handler.push_litc C s | � d� d S )Nz 500 What?)r� )r r r r
r� � s z NNTPv1Handler.handle_unknownc C s | � | j� d S )N)r� r )r r r r
r� � s z NNTPv1Handler.handle_welcomec C s | � d� d S )Nz205 Bye!)r� )r r r r
� handle_QUIT� s zNNTPv1Handler.handle_QUITc C s | � d� d S )Nz111 20100914001155)r� )r r r r
� handle_DATE s zNNTPv1Handler.handle_DATEc C s( |dkr| � d� n| � d�|�� d S )Nzfr.comp.lang.pythonz$211 486 761 1265 fr.comp.lang.pythonz411 No such group {})r� r� )r r r r r
� handle_GROUP s zNNTPv1Handler.handle_GROUPc C s | � d� d S )Nz� 100 Legal commands
authinfo user Name|pass Password|generic <prog> <args>
date
help
Report problems to <root@example.org>
.)r� )r r r r
� handle_HELP s zNNTPv1Handler.handle_HELPNc C sJ |d kr| � d� n2|dkr(| � d� n |dkr<| � d� n
| � d� d S )Nz412 No newsgroup selected�3000234z"223 3000234 <45223423@example.com>z<45223423@example.com>z 223 0 <45223423@example.com>z430 No Such Article Found)r� )r � message_specr r r
� handle_STAT s zNNTPv1Handler.handle_STATc C s | � d� d S )Nz*223 3000237 <668929@example.org> retrieved)r� )r r r r
� handle_NEXT s zNNTPv1Handler.handle_NEXTc C s | � d� d S )Nz,223 3000234 <45223423@example.com> retrieved)r� )r r r r
� handle_LAST s zNNTPv1Handler.handle_LASTc C s� |d kr| � d� n�|dkr<|dkr0| � d� q�| � d� nZ|dkrP| � d� nF|dkr�|d krl| � d
� q�|d kr�| � d � q�| � d
� n
| � d� d S )Na� 215 Newsgroups in form "group high low flags".
comp.lang.python 0000052340 0000002828 y
comp.lang.python.announce 0000001153 0000000993 m
free.it.comp.lang.python 0000000002 0000000002 y
fr.comp.lang.python 0000001254 0000000760 y
free.it.comp.lang.python.learner 0000000000 0000000001 y
tw.bbs.comp.lang.python 0000000304 0000000304 y
.�ACTIVEz *distutils*z� 215 Newsgroups in form "group high low flags"
gmane.comp.python.distutils.devel 0000014104 0000000001 m
gmane.comp.python.distutils.cvs 0000000000 0000000001 m
.zW 215 Newsgroups in form "group high low flags"
.z OVERVIEW.FMTa 215 Order of fields in overview database.
Subject:
From:
Date:
Message-ID:
References:
Bytes:
Lines:
Xref:full
.�
NEWSGROUPSzcomp.lang.pythonz� 215 Descriptions in form "group description".
comp.lang.python The Python computer language.
.zcomp.lang.python*z� 215 Descriptions in form "group description".
comp.lang.python.announce Announcements about the Python language. (Moderated)
comp.lang.python The Python computer language.
.zW 215 Descriptions in form "group description".
.z501 Unknown LIST keyword)r� )r �actionZparamr r r
� handle_LIST" s,
zNNTPv1Handler.handle_LISTc C s� |dkr$|dkr$|dkr$| � d� nZ|dkrH|dkrH|dkrH| � d� n6|dkrt|dkrt|dkrt| � d d
d � n
| � d � d S )
Nzcomp.lang.pythonZ20100913Z082004a 230 list of newsarticles (NNTP v2) created after Mon Sep 13 08:20:04 2010 follows
<a4929a40-6328-491a-aaaf-cb79ed7309a2@q2g2000vbk.googlegroups.com>
<f30c0419-f549-4218-848f-d7d0131da931@y3g2000vbm.googlegroups.com>
.Z100913a 230 list of newsarticles (NNTP v1) created after Mon Sep 13 08:20:04 2010 follows
<a4929a40-6328-491a-aaaf-cb79ed7309a2@q2g2000vbk.googlegroups.com>
<f30c0419-f549-4218-848f-d7d0131da931@y3g2000vbm.googlegroups.com>
.)Z20100101Z100101Z090000z
too long linei� z
.zK 230 An empty list of newsarticles follows
.)r� )r r �date_str�time_strr r r
�handle_NEWNEWSX s
z NNTPv1Handler.handle_NEWNEWSc C s" |dkr| � d� n
| � d� d S )Nz57-59uK 224 Overview information for 57-58 follows
57 Re: ANN: New Plone book with strong Python (and Zope) themes throughout Doug Hellmann <doug.hellmann-Re5JQEeQqe8AvxtiuMwx3w@public.gmane.org> Sat, 19 Jun 2010 18:04:08 -0400 <4FD05F05-F98B-44DC-8111-C6009C925F0C@gmail.com> <hvalf7$ort$1@dough.gmane.org> 7103 16 Xref: news.gmane.io gmane.comp.python.authors:57
58 Looking for a few good bloggers Doug Hellmann <doug.hellmann-Re5JQEeQqe8AvxtiuMwx3w@public.gmane.org> Thu, 22 Jul 2010 09:14:14 -0400 <A29863FA-F388-40C3-AA25-0FD06B09B5BF@gmail.com> 6683 16
59 Re: Message d'erreur incompréhensible (par moi) Eric Brunel <eric.brunel@pragmadev.nospam.com> Wed, 15 Sep 2010 18:09:15 +0200 <eric.brunel-2B8B56.18091515092010@news.wanadoo.fr> <4c90ec87$0$32425$ba4acef3@reader.news.orange.fr> 1641 27 Xref: saria.nerim.net fr.comp.lang.python:1265
.
z1 224 No articles
.)r� )r r� r r r
� handle_XOVERw s
zNNTPv1Handler.handle_XOVER)rf c C sB |d kr.| j r"| �d� | �� q>| �d� n| �d� || _d S )Nz+340 Input article; end with <CR-LF>.<CR-LF>z440 Posting not permittedz240 Article received OK)r� r� r� � posted_body)r rf r r r
� handle_POST� s
zNNTPv1Handler.handle_POSTc C sJ |d kr6| j r*|dkr*| �d� | �� qF| �d� n| �d� || _d S )Nz+<i.am.an.article.you.will.want@example.com>z%335 Send it; end with <CR-LF>.<CR-LF>z435 Article not wantedz235 Article transferred OK)r� r� r� r� )r �
message_idrf r r r
� handle_IHAVE� s
zNNTPv1Handler.handle_IHAVEz� From: "Demo User" <nobody@example.net>
Subject: I am just a test article
Content-Type: text/plain; charset=UTF-8; format=flowed
Message-ID: <i.am.an.article.you.will.want@example.com>ui This is just a test article.
..Here is a dot-starting line.
-- Signed by André.z
c C sd |d kr| � d� n6|dkr(| � d� n"|dkr<| � d� n| � d� d S | � | j� | � d� d S )Nz"220 3000237 <45223423@example.com>z<45223423@example.com>z 220 0 <45223423@example.com>r� z"220 3000234 <45223423@example.com>z430 No Such Article Found�.)r� �sample_article)r r� r r r
�handle_ARTICLE� s
z NNTPv1Handler.handle_ARTICLEc C sd |d kr| � d� n6|dkr(| � d� n"|dkr<| � d� n| � d� d S | � | j� | � d� d S )Nz"221 3000237 <45223423@example.com>z<45223423@example.com>z 221 0 <45223423@example.com>r� z"221 3000234 <45223423@example.com>z430 No Such Article Foundr� )r� � sample_head)r r� r r r
� handle_HEAD� s
zNNTPv1Handler.handle_HEADc C sd |d kr| � d� n6|dkr(| � d� n"|dkr<| � d� n| � d� d S | � | j� | � d� d S )Nz"222 3000237 <45223423@example.com>z<45223423@example.com>z 222 0 <45223423@example.com>r� z"222 3000234 <45223423@example.com>z430 No Such Article Foundr� )r� � sample_body)r r� r r r
� handle_BODY� s
zNNTPv1Handler.handle_BODYc C sj | j r| �d� nT|dkr>| jr,| �d� qf| �d� d| _n(|dkrX| �d� d| _ ntd�|���d S ) Nz502 Already Logged Inrq z 482 User Credential Already Sentz381 Password RequiredT�passz281 Login SuccessfulzUnknown cred type {})r� r� r� r� r� )r Z cred_typer� r r r
�handle_AUTHINFO� s
z NNTPv1Handler.handle_AUTHINFO)N)NN)N)N)N) r r r
r rW 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
r� � s8
&
6
r� c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) �
NNTPv2Handlerc C s0 d}| j s | �|�d�� n| �|�d�� d S )Na 101 Capability list:
VERSION 2 3
IMPLEMENTATION INN 2.5.1{}
HDR
LIST ACTIVE ACTIVE.TIMES DISTRIB.PATS HEADERS NEWSGROUPS OVERVIEW.FMT
OVER
POST
READER
.z
AUTHINFO USERr5 )r� r� r� )r �fmtr r r
�handle_CAPABILITIES� s
z!NNTPv2Handler.handle_CAPABILITIESc C s t d��d S )Nz3MODE READER sent despite READER has been advertised)r� )r �_r r r
� handle_MODE s zNNTPv2Handler.handle_MODENc C s
| � |�S )N)r� )r r� r r r
� handle_OVER s zNNTPv2Handler.handle_OVER)N)r r r
r r r r r r r
r� � s r� c s e Zd Z� fdd�Z� ZS )�CapsAfterLoginNNTPv2Handlerc s | j s| �d� n
t� �� d S )Nz480 You must log in.)r� r� r� r )r )r� r r
r s z/CapsAfterLoginNNTPv2Handler.handle_CAPABILITIES)r r r
r r� r r )r� r
r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ModeSwitchingNNTPv2Handlerc C s
d| _ d S )NF)� _switched)r r r r
r� s z#ModeSwitchingNNTPv2Handler.__init__c C s0 d}| j r | �|�d�� n| �|�d�� d S )Na 101 Capability list:
VERSION 2 3
IMPLEMENTATION INN 2.5.1
HDR
LIST ACTIVE ACTIVE.TIMES DISTRIB.PATS HEADERS NEWSGROUPS OVERVIEW.FMT
OVER
POST
{}READER
.r5 zMODE-)r r� r� )r r� r r r
r s
z.ModeSwitchingNNTPv2Handler.handle_CAPABILITIESc C s d| _ | �d� d S )NTz200 Posting allowed)r r� )r Zwhatr r r
r 0 s z&ModeSwitchingNNTPv2Handler.handle_MODEN)r r r
r� r r r r r r
r s r c s e Zd Z� fdd�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 d5Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#� Z$S )B�NNTPv1v2TestsMixinc s t � �� d S )N)r� r� )r )r� r r
r� 8 s zNNTPv1v2TestsMixin.setUpc C s | � | jj| jj� d S )N)r r r r� )r r r r
r ; s z NNTPv1v2TestsMixin.test_welcomec C s< | j dkr| �d| jj� | j�dd� | �d| jj� d S )Nrm �AUTHINFO�testuser�testpw)rn r7 r �_capsru r1 )r r r r
�
test_authinfo> s
z NNTPv1v2TestsMixin.test_authinfoc C s: | j �� \}}| �|d� | �|t�dddddd�� d S )Nz111 20100914001155i� � � r � �7 )r r- r r, )r r r- r r r
rF F s z NNTPv1v2TestsMixin.test_datec C s6 | � | jj� | j�� }| �|d� | �| jj� d S )Nz205 Bye!)r� r� �closedr rw r r )r r r r r
� test_quitK s
z NNTPv1v2TestsMixin.test_quitc C s2 | j �� \}}| �|d� | �|ddddg� d S )Nz100 Legal commandsz8 authinfo user Name|pass Password|generic <prog> <args>z datez helpz%Report problems to <root@example.org>)r r r )r r r r r r
r Q s z NNTPv1v2TestsMixin.test_helpc C s~ | j �� \}}| �t|�d� |d }| �|tdddd�� | j �d�\}}| �t|�d� |d }| �|td
d d d�� d S )
N� rY zcomp.lang.python.announceZ
0000001153Z
0000000993�mz *distutils*rm r z!gmane.comp.python.distutils.develZ
0000014104Z
0000000001)r r r r r )r r r �gr r r
r! [ s
z NNTPv1v2TestsMixin.test_listc C s� | j �d�\}}}| �|d� | �|d� | �|d� | j �d�\}}}| �|d� | �|d� | �|d� | �tj��}| j �d� W d Q R X | �|jjd� | �tj��}| j �� W d Q R X | �|jjd� d S ) Ni��- z"223 3000234 <45223423@example.com>z<45223423@example.com>z 223 0 <45223423@example.com>r z<non.existent.id>z430 No Such Article Foundz412 No newsgroup selected)r �statr r$ r% re r' r( )r r rR r� r) r r r
� test_stati s z NNTPv1v2TestsMixin.test_statc C s8 | j �� \}}}| �|d� | �|d� | �|d� d S )Nz*223 3000237 <668929@example.org> retrievedi��- z<668929@example.org>)r �nextr )r r rR r� r r r
� test_nexty s z NNTPv1v2TestsMixin.test_nextc C s8 | j �� \}}}| �|d� | �|d� | �|d� d S )Nz,223 3000234 <45223423@example.com> retrievedi��- z<45223423@example.com>)r rC r )r r rR r� r r r
� test_last s z NNTPv1v2TestsMixin.test_lastc C s4 | j �d�}| �|d� | j �d�}| �|d� d S )Nzcomp.lang.pythonz The Python computer language.zcomp.lang.pythonxr5 )r r6 r )r r3 r r r
r9 � s z#NNTPv1v2TestsMixin.test_descriptionc C sn | j �d�\}}| �|d� | �|ddi� | j �d�\}}| �|ddd�� | j �d�\}}| �|i � d S )Nzcomp.lang.pythonz-215 Descriptions in form "group description".z The Python computer language.zcomp.lang.python*z4Announcements about the Python language. (Moderated))zcomp.lang.pythonzcomp.lang.python.announcezcomp.lang.pythonx)r r: r )r r r r r r
r< � s
z$NNTPv1v2TestsMixin.test_descriptionsc C s� | j �d�\}}}}}| �|�d�|� | �|d� | �|d� | �|d� | �|d� | �tj��}| j �d� W d Q R X |j}| �|j �d�|j � d S )Nzfr.comp.lang.pythonz211 i� i� i� zcomp.lang.python.develz411 No such group)
r r r r r r$ r% re r' r( )r r rA rB rC r r) �excr r r
rD � s z NNTPv1v2TestsMixin.test_groupc C s� t � dddddd�}| j�d|�\}}d�| j�}| �||� | �|d d
g� t � dddddd�}| j�d |�\}}| �|d � | �|g � d S )
Ni� r
�
� � � zcomp.lang.pythonzS230 list of newsarticles (NNTP v{0}) created after Mon Sep 13 08:20:04 2010 followszB<a4929a40-6328-491a-aaaf-cb79ed7309a2@q2g2000vbk.googlegroups.com>zB<f30c0419-f549-4218-848f-d7d0131da931@y3g2000vbm.googlegroups.com>zfr.comp.lang.pythonz)230 An empty list of newsarticles follows)r, r �newnewsr� rn r )r r/ r ZidsZexpectedr r r
� test_newnews� s z NNTPv1v2TestsMixin.test_newnewsc C sZ | � t|�d� | � |d �d�d� | � |d d� | � |d d� | � |d d
� d S ) Nr rU zutf-8u -- Signed by André.r� r` ���s .Here is a dot-starting line.���s This is just a test article.)r r �decode)r r r r r
�_check_article_body� s
z&NNTPv1v2TestsMixin._check_article_bodyc C s4 | � t|�d� | � |d d� | � |d d� d S )Nr r s& From: "Demo User" <nobody@example.net>� s7 Message-ID: <i.am.an.article.you.will.want@example.com>)r r )r r r r r
�_check_article_head� s z&NNTPv1v2TestsMixin._check_article_headc C sH | � t|�d� | �|d d� � | �|dd � � | � |d d� d S )Nr
r r# r` )r r r' r% )r r r r r
�_check_article_data� s z&NNTPv1v2TestsMixin._check_article_datac C s | j �� \}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | �tj��}| j �d� W d Q R X | �|jjd � d S )
Nz"220 3000237 <45223423@example.com>i��- z<45223423@example.com>i��- z"220 3000234 <45223423@example.com>z 220 0 <45223423@example.com>r z<non-existent@example.com>z430 No Such Article Found) r r] r r( r$ r% re r' r( )r r �inforR r� r r) r r r
� test_article� s*
z NNTPv1v2TestsMixin.test_articlec C s� t �� }| jj|d�\}}| �|d� |\}}}| �|d� | �|d� | �|g � |�� }| �|�d�t|�� | �|� d�t|�� d S )N)r� z"220 3000237 <45223423@example.com>i��- z<45223423@example.com>sK From: "Demo User" <nobody@example.net>
Subject: I am just a test article
sU This is just a test article.
.Here is a dot-starting line.
-- Signed by André.
)
r� r� r r] r �getvaluer r �asciir� )r �fr r) rR r� r r� r r r
�test_article_file� s
z$NNTPv1v2TestsMixin.test_article_filec C s | j �� \}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | �tj��}| j �d� W d Q R X | �|jjd � d S )
Nz"221 3000237 <45223423@example.com>i��- z<45223423@example.com>i��- z"221 3000234 <45223423@example.com>z 221 0 <45223423@example.com>r z<non-existent@example.com>z430 No Such Article Found) r rd r r' r$ r% re r' r( )r r r) rR r� r r) r r r
� test_head� s*
z NNTPv1v2TestsMixin.test_headc C s� t �� }| jj|d�\}}| �|d� |\}}}| �|d� | �|d� | �|g � |�� }| �|�d�t|�� | � |�
d�t|�� d S )N)r� z"221 3000237 <45223423@example.com>i��- z<45223423@example.com>sK From: "Demo User" <nobody@example.net>
Subject: I am just a test article
sU This is just a test article.
.Here is a dot-starting line.
-- Signed by André.
) r� r� r rd r r+ r r r, r� r� )r r- r r) rR r� r r� r r r
�test_head_file s
z!NNTPv1v2TestsMixin.test_head_filec C s | j �� \}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | j �d�\}}| �|d� |\}}}| �|d� | �|d� | �|� | �tj��}| j �d� W d Q R X | �|jjd � d S )
Nz"222 3000237 <45223423@example.com>i��- z<45223423@example.com>i��- z"222 3000234 <45223423@example.com>z 222 0 <45223423@example.com>r z<non-existent@example.com>z430 No Such Article Found) r rf r r% r$ r% re r' r( )r r r) rR r� r r) r r r
� test_body* s*
z NNTPv1v2TestsMixin.test_bodyc C s� t �� }| jj|d�\}}| �|d� |\}}}| �|d� | �|d� | �|g � |�� }| �|�d�t|�� | � |�
d�t|�� d S )N)r� z"222 3000237 <45223423@example.com>i��- z<45223423@example.com>sK From: "Demo User" <nobody@example.net>
Subject: I am just a test article
sU This is just a test article.
.Here is a dot-starting line.
-- Signed by André.
) r� r� r rf r r+ r� r r, r r� )r r- r r) rR r� r r� r r r
�test_body_fileE s
z!NNTPv1v2TestsMixin.test_body_filec C s� | � |�d�|� | �t|�d� |d \}}| �|d� | �|ddddd d
d d d
�� |d \}}| �|d d � |d \}}| �|d d� d S )Nz224 r& r �9 zEDoug Hellmann <doug.hellmann-Re5JQEeQqe8AvxtiuMwx3w@public.gmane.org>zGRe: ANN: New Plone book with strong Python (and Zope) themes throughoutz Sat, 19 Jun 2010 18:04:08 -0400z0<4FD05F05-F98B-44DC-8111-C6009C925F0C@gmail.com>z <hvalf7$ort$1@dough.gmane.org>Z7103Z16z*news.gmane.io gmane.comp.python.authors:57)rH rG r- z
message-idrI z:bytesz:lines�xrefrY r4 rm rG u0 Re: Message d'erreur incompréhensible (par moi))r r r r )r r � overviewsrR rV r r r
�check_over_xover_respY s$
z(NNTPv1v2TestsMixin.check_over_xover_respc C s" | j �dd�\}}| �||� d S )Nr3 �; )r rO r6 )r r r5 r r r
rS n s z NNTPv1v2TestsMixin.test_xoverc C s | j �d�\}}| �||� d S )N)r3 r7 )r rV r6 )r r r5 r r r
rX r s z NNTPv1v2TestsMixin.test_overs From: "Demo User" <nobody@example.net>
Subject: I am just a test article
Content-Type: text/plain; charset=UTF-8; format=flowed
Message-ID: <i.am.an.article.you.will.want@example.com>
This is just a test article.
.Here is a dot-starting line.
-- Signed by André.
c C sl | j j}| �t|�d� | �|d d� | �|d d� | �|d d� | �|d d � | �|d
d � d S ) NrT rU s .
r� s -- Signed by André.
r" s
r# s ..Here is a dot-starting line.
r s( From: "Demo User" <nobody@example.net>
)r� r� r r )r r r r r
�_check_posted_body� s z%NNTPv1v2TestsMixin._check_posted_bodyc G sd | j }|||�f }d | j_||� }| �� | j �dd�}|||�f }d | j_||� }| �� |S )Ns
�
)� sample_postr� r� r8 �replace)r �func� file_factoryr� �postZ func_argsr r r r
�_check_post_ihave_sub� s z(NNTPv1v2TestsMixin._check_post_ihave_subc G s� | j |f|�dti�}| �||� | j |f|�dti�}| �||� | j |f|�dtji�}| �||� dd� }| j |f|�d|i�}| �||� dd� }| j |f|�d|i�}| �||� d S )Nr= c S s t | jdd��S )NT)�keepends)�iterr� )r� r r r
� iterlines� s z6NNTPv1v2TestsMixin.check_post_ihave.<locals>.iterlinesc S s t | jdd��S )NF)r@ )rA r� )r� r r r
rB � s )r? r\ r � bytearrayr� r� )r r< Z success_respr� r rB r r r
�check_post_ihave� s z#NNTPv1v2TestsMixin.check_post_ihavec C sR | � | jjd� d| j_| �tj��}| j�| j� W d Q R X | � |j
j d� d S )Nz240 Article received OKFz440 Posting not permitted) rD r r> r� r� r$ r% re r: r r' r( )r r) r r r
� test_post� s
z NNTPv1v2TestsMixin.test_postc C sN | � | jjdd� | �tj��}| j�d| j� W d Q R X | �|jj d� d S )Nz235 Article transferred OKz+<i.am.an.article.you.will.want@example.com>z<another.message.id>z435 Article not wanted)
rD r Zihaver$ r% re r: r r' r( )r r) r r r
�
test_ihave� s
z NNTPv1v2TestsMixin.test_ihavec C s. t � dddddd�}| �tj| jjd|� d S )Ni� rY r
r zcomp.lang.python)r, r$ r% �
NNTPDataErrorr r )r r/ r r r
�test_too_long_lines� s z&NNTPv1v2TestsMixin.test_too_long_lines)%r r r
r� r r rF r r r! r r r r9 r< rD r! r% r' r( r* r. r/ r0 r1 r2 r6 rS rX r: r8 r? rD rE rF rH r� r r )r� r
r 6 sB
r c @ s e Zd ZdZeZdd� ZdS )� NNTPv1TestsrY c C s: | j �� }| �|i � | �| j jd� | �| j jd � d S )NrY )r ro r rn �nntp_implementation)r rk r r r
� test_caps� s
zNNTPv1Tests.test_capsN)r r r
rn r� r� rK r r r r
rI � s rI c @ s e Zd ZdZeZdd� ZdS )� NNTPv2Testsrm c
C sb | j �� }| �|ddgddgdgg dddd d
d gg g g d �� | �| j jd
� | �| j jd� d S )N�2�3ZINNz2.5.1ZUSERr� z ACTIVE.TIMESz DISTRIB.PATSZHEADERSr� z OVERVIEW.FMT)�VERSIONZIMPLEMENTATIONr ZHDRri ZOVERZPOST�READERr& z INN 2.5.1)r ro r rn rJ )r rk r r r
rK � s
zNNTPv2Tests.test_capsN)r r r
rn r� r� rK r r r r
rL � s rL c @ s e Zd ZdZeZdd� ZdS )�CapsAfterLoginNNTPv2Testsrm c C s2 | � | jji � | j�dd� | �d| jj� d S )Nr r
rO )r r r ru r7 )r r r r
�test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r r r
rn r r� rR r r r r
rQ � s rQ c @ s e Zd ZdZeZdd� ZdS )�SendReaderNNTPv2Testsrm c C s | � d| jj� d S )NrP )r7 r r )r r r r
�(test_we_are_in_reader_mode_after_connect s z>SendReaderNNTPv2Tests.test_we_are_in_reader_mode_after_connectN)r r r
rn r r� rT r r r r
rS � s rS c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze �
e d
�dd� �Z dS )� MiscTestsc sV � fdd�}|dd� |dd� |dd� |dd� |dd � |d
d � |d d � d S )
Nc s � � t�| �|� d S )N)r r% �
decode_header)�ar� )r r r
�gives s z+MiscTests.test_decode_header.<locals>.givesr5 za plain headerz with extra spaces z%=?ISO-8859-15?Q?D=E9buter_en_Python?=u Débuter en Pythonz=?utf-8?q?Re=3A_=5Bsqlite=5D_probl=C3=A8me_avec_ORDER_BY_sur_des_cha?= =?utf-8?q?=C3=AEnes_de_caract=C3=A8res_accentu=C3=A9es?=uP Re: [sqlite] problème avec ORDER BY sur des chaînes de caractères accentuéesz,Re: =?UTF-8?B?cHJvYmzDqG1lIGRlIG1hdHJpY2U=?=u Re: problème de matriceu0 Re: Message d'erreur incompréhensible (par moi)r )r rX r )r r
�test_decode_header s
z MiscTests.test_decode_headerc C sD dddddddg}| � t�|�dd d
d d ddg� dddddd
dg}| � t�|�dd d
d d ddg� dddddddg}| � t�|�dd d
d d ddg� dddddddddg }| � t�|�dd d
d d ddddg � dddddd
dddg }| � t�|�dd d
d d ddddg � dddddd
ddg}| � t�|�dd d
d d dddg� d S )NzSubject:zFrom:zDate:z Message-ID:z References:z:bytesz:linesrG rH r- z
message-idrI zBytes:zLines:zsubject:zFROM:zDaTe:z message-ID:zBYTES:z Xref:fullzDistribution:fullr4 Z distributionz Xref:FULLzDistribution:FULL)r r% Z_parse_overview_fmt)r r r r r
�test_parse_overview_fmt s@
z!MiscTests.test_parse_overview_fmtc C s� t jdg }dg}t �||�}|\\}}| �|d� | �|dddddd d
d d �� d
g}t �||�}|\\}}| �|d d � dg}t �||�}|\\}}| �|d d� | �|d d� d S )Nr4 z�3000234 I am just a test article "Demo User" <nobody@example.com> 6 Oct 1998 04:38:40 -0500 <45223423@example.com> <45454@example.net> 1234 17 Xref: news.example.com misc.test:3000363i��- zI am just a test articlez "Demo User" <nobody@example.com>z6 Oct 1998 04:38:40 -0500z<45223423@example.com>z<45454@example.net>Z1234Z17z"news.example.com misc.test:3000363)rG rH r- z
message-idrI z:bytesz:linesr4 z�3000234 I am just a test article "Demo User" <nobody@example.com> 6 Oct 1998 04:38:40 -0500 <45223423@example.com> <45454@example.net> 1234 17 z�3000234 I am just a test article "Demo User" <nobody@example.com> 6 Oct 1998 04:38:40 -0500 <45223423@example.com> 1234 17 Xref: rI � r5 )r% Z_DEFAULT_OVERVIEW_FMTZ_parse_overviewr )r r� r ZoverviewrR Zfieldsr r r
�test_parse_overview? s.
z MiscTests.test_parse_overviewc sh � fdd�}|dd dddddd � |d
d dddddd � |d d dddddd � |d
d dddddd � d S )Nc s � � t�| |�tj|� � d S )N)r r% Z_parse_datetimer, )rW r� �c)r r r
rX n s z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624i� r � r �8 � �19990623�135624�990623Z090623i� r )r rX r )r r
�test_parse_datetimem s
z MiscTests.test_parse_datetimec s� � fdd�}|ddddddd d
� |d dddddd d
� |d
ddddddd� � fdd�}|dddd d� |d ddd d� |d
dddd� d S )Nc sF t � | |||||�}� �t�|�||f� � �t�|d�||f� d S )NF)r, r r% �_unparse_datetime) �y�M�d�hr �sr� r� r/ )r r r
rX { s
z.MiscTests.test_unparse_datetime.<locals>.givesi� r r^ r r_ r` ra rb i� Z20000623i� r= rY rm r& Z20100605�010203c s@ t �| ||�}� �t�|�||f� � �t�|d�||f� d S )NF)r, r- r r% re )rf rg rh r� r� r/ )r r r
rX � s
�000000r )r rX r )r r
�test_unparse_datetimex s z MiscTests.test_unparse_datetimec s� � fdd�}|ddddddd d
� |d dddddd d
� |d
ddddddd� � fdd�}|dddd d� |d ddd d� |d
dddd� d S )Nc s0 t � | |||||�}� �t�|d�||f� d S )NT)r, r r% re ) rf rg rh ri r rj r� r� r/ )r r r
rX � s z5MiscTests.test_unparse_datetime_legacy.<locals>.givesi� r r^ r r_ r` rc rb i� Z000623i� r= rY rm r& Z100605rk c s* t �| ||�}� �t�|d�||f� d S )NT)r, r- r r% re )rf rg rh r� r� r/ )r r r
rX � s rl r )r rX r )r r
� test_unparse_datetime_legacy� s z&MiscTests.test_unparse_datetime_legacyzrequires SSL supportc C s | � ttd�� d S )Nr� )r r� r% )r r r r
�test_ssl_support� s zMiscTests.test_ssl_supportN)
r r r
rY rZ r\ rd rm rn r� �
skipUnless�sslro r r r r
rU s (. rU c @ s e Zd Zdd� ZdS )�PublicAPITestsc C sP | � ttd�� dddddddd g}td k r6|�d
� | �ttj�t|�� d S ) N�__all__r rt �NNTPReplyErrorre r& �NNTPProtocolErrorrG rV r� )r r� r% rq r� r �setrs )r Z
target_apir r r
�test_module_all_attribute� s
z(PublicAPITests.test_module_all_attributeN)r r r
rw r r r r
rr � s rr c @ sD e Zd ZejZddd�Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dS )�MockSocketTestsNc
s� G � fdd�d�}G ���fdd�d�� d�g �t d|��. | �||�� | jd||d� W d Q R X W d Q R X | ��� x�D ]}| �|j� q~W d S ) Nc s e Zd Z� fdd�ZdS )zNMockSocketTests.check_constructor_error_conditions.<locals>.mock_socket_modulec s � � S )Nr )Zaddressr� )�
MockSocketr r
�create_connection� s z`MockSocketTests.check_constructor_error_conditions.<locals>.mock_socket_module.create_connectionN)r r r
rz r )ry r r
�mock_socket_module� s r{ c s& e Zd Z�fdd�Z� �fdd�ZdS )zFMockSocketTests.check_constructor_error_conditions.<locals>.MockSocketc s d� d S )NTr )r )�
socket_closedr r
�close� s zLMockSocketTests.check_constructor_error_conditions.<locals>.MockSocket.closec s �� }t |�\}}� �|� |S )N)r� r� )�socket�moder� r r� )�filesr� r r
�makefile� s
zOMockSocketTests.check_constructor_error_conditions.<locals>.MockSocket.makefileN)r r r
r} r� r )r� r� r| r r
ry � s ry Fznntplib.socketZdummy)rq rr )r ZassertRaisesRegex�
nntp_classr r )r r� Zexpected_error_typeZexpected_error_msgru rr r{ r- r )ry r� r� r| r
�"check_constructor_error_conditions� s $
z2MockSocketTests.check_constructor_error_conditionsc C s& G dd� dt �}| �|tj|j� d S )Nc @ s e Zd ZdZdS )z1MockSocketTests.test_bad_welcome.<locals>.Handlerz Bad WelcomeN)r r r
r r r r r
�Handler� s r� )r� r� r% ru r )r r� r r r
�test_bad_welcome� s z MockSocketTests.test_bad_welcomec C s& G dd� dt �}| �|tj|j� d S )Nc @ s e Zd ZdZdS )zEMockSocketTests.test_service_temporarily_unavailable.<locals>.Handlerz#400 Service temporarily unavailableN)r r r
r r r r r
r� � s r� )r� r� r% re r )r r� r r r
�$test_service_temporarily_unavailable� s z4MockSocketTests.test_service_temporarily_unavailablec C s& G dd� dt �}| �|tj|j� d S )Nc @ s e Zd ZdZdS )zEMockSocketTests.test_service_permanently_unavailable.<locals>.Handlerz#502 Service permanently unavailableN)r r r
r r r r r
r� � s r� )r� r� r% r& r )r r� r r r
�$test_service_permanently_unavailable� s z4MockSocketTests.test_service_permanently_unavailablec s, G � fdd�dt �}d� | �|tj� � d S )Nc s e Zd Z� fdd�ZdS )z6MockSocketTests.test_bad_capabilities.<locals>.Handlerc s | � � � d S )N)r� )r )�capabilities_responser r
r � s zJMockSocketTests.test_bad_capabilities.<locals>.Handler.handle_CAPABILITIESN)r r r
r r )r� r r
r� � s r� z201 bad capability)r� r� r% rt )r r� r )r� r
�test_bad_capabilities� s z%MockSocketTests.test_bad_capabilitiesc s8 d}d}G � fdd�dt �}d� | �|tj� ||� d S )Nzt@e.comZpythonc s e Zd Z� fdd�ZdS )z3MockSocketTests.test_login_aborted.<locals>.Handlerc s | � � � d S )N)r� )r r� )�authinfo_responser r
r� � s zCMockSocketTests.test_login_aborted.<locals>.Handler.handle_AUTHINFON)r r r
r� r )r� r r
r� � s r� z 503 Mechanism not recognized)r� r� r% r& )r ru rr r� r )r� r
�test_login_aborted� s z"MockSocketTests.test_login_aborted)NN) r r r
r% r r� r� r� r� r� r� r� r r r r
rx � s
rx c @ s e Zd Zdd� ZdS )�bypass_contextc K s | S )Nr )�sockr� r r r
� wrap_socket s zbypass_context.wrap_socketN)r r r
r� r r r r
r� s r� c @ s e Zd Zedd� �ZdS )� MockSslTestsc O s t j| dti|��S )NZ ssl_context)r% r� r� )r� �kwr r r
r� s zMockSslTests.nntp_classN)r r r
� staticmethodr� r r r r
r� s r� c @ s0 e Zd Zdd� Zdd� Ze�ed�dd� �ZdS ) �LocalServerTestsc C st t � � }t�|�}|�� tj| j|fd�| _| j�� | � | jj
� t tj |dd��
� | _| � | jjd d d � d S )N)�targetr� F)rs )r~ r Z bind_portZlisten� threadingZThread�
run_serverZ
backgroundrW Z
addCleanupr� r ZHOST� __enter__�nntp�__exit__)r r� Zportr r r
r� s
zLocalServerTests.setUpc C s� |� |� � \}}W d Q R X t�� ��}|�|� |�|�d��}|�d� x�|�� }|dkrh|�d� qL|dkr�|�� |�d� t� � }|�
t � |j |dd�}|�|� |�|�d��}qL|d kr�|�d
� P qLt
d �|���qLW W d Q R X d S ) N�rbs 200 Server ready
s CAPABILITIES
s. 101 Capability list:
VERSION 2
STARTTLS
.
s
STARTTLS
s 382 Begin TLS negotiation now
T)Z server_sides QUIT
s
205 Bye!
zUnexpected command {!r})Zaccept�
contextlib� ExitStack�
enter_contextr� Zsendallr� r} rq Z
SSLContextZload_cert_chain�certfiler� r� r� )r r� Zclientr Zcleanup�readerr� �contextr r r
r� s0
zLocalServerTests.run_serverzrequires SSL supportc C s` | j j}| j j}| j �� | �|| j j� | �|| j j� | �| j jtj� | �t | j j� d S )N)
r� r� r� ZstarttlsZassertNotEqualr. rq Z SSLSocketr$ r� )r r� r� r r r
r� : s
z LocalServerTests.test_starttlsN) r r r
r� r� r� rp rq r� r r r r
r�
s r� �__main__);r� r~ r, r� r� r~ r� Zos.path�osr� r� r� r r% r r Z
unittest.mockr rq � ImportErrorr� �pathr� �dirname�__file__r� r r� r r� r� r� Z SSLEOFErrorZTestCaser� rp r� r� r� r� r� r� r� r� r r r rI rL rQ rS rU rr rx r� r� r� r �mainr r r r
�<module> sx
|
, G
# N<