File: //usr/local/lib/python3.7/test/__pycache__/test_nntplib.cpython-37.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 ZdZdZdS )�SSLErrorz6Non-existent exception class when we lack SSL support.zThis will never be raised.N)�__name__�
__module__� __qualname__�__doc__�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 � descriptionr3 �assertIn�DESCr# r )r r5 r4 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/ �dictr3 r r7 )r r Zdescsr4 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 r3 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< rF �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 r3 �xoverr �skipTestrF r@ rO ) r r rB rC rD �namer �art_numrM 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 r3 �overrO rF r@ )
r r rB rC rD rR �startr rS rM r
r
r � test_over� s
z!NetworkedNNTPTestsMixin.test_overc C sP | j �| j�\}}}}}| j �d|�\}}x |D ]}| �tt|d �� q0W d S )NrH � )r r r3 Zxhdrr r r )r r rB rC rD rR 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 )NrV )� .s .
s .
)r/ r& Z ArticleInfor Znumberr �bytesr2 )r r �articlerS 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 )
NrZ 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
)rb r )rc r
r rd � s ra )r r r3 �headr& �NNTPTemporaryErrorr) r rQ r r_ �bodyr^ r r )
r r rB rC rD rR rS re �erg r^ Zfiltered_head_linesZfiltered_linesr
)rc 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 �tupler8 )�capsZ caps_list)r r
r � _check_caps� s z>NetworkedNNTPTestsMixin.test_capabilities.<locals>._check_caps� )rF r � nntp_version�getcapabilitiesZ capabilities)r rm r rl 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)ry r� rR 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)�timeoutrt z(?i)KEY.TOO.SMALLzGot z connecting to )
�
NNTP_CLASSr| �TIMEOUTr r � assertFalserx 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+ r1 r: r= rE rG rO rT r� ZskipIfrY r[ r_ ri rq rw rz � 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� rt z(?i)KEY.TOO.SMALLz got z connecting to z got EOF error on connecting to )r Zrequiresr{ r| r� r� r r r� r� r r� r� �
EOF_ERRORS)ry r� r
r
r �
setUpClass2 s
z NetworkedNNTPTests.setUpClassc C s | j d k r| j �� d S )N)r rx )ry r
r
r �
tearDownClassC s
z NetworkedNNTPTests.tearDownClassN) r r r
r| r3 r# r9 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| r3 r# r9 r� r& r� r" �
test_starttlsr
r
r
r r� H s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
_NNTPServerIOz�A raw IO object allowing NNTP commands to be received and processed
by a handler. The handler can push responses which can then be read
from the IO object.c C s> t j�| � t �� | _t �� | _|| _| j�| jj| j � d S )N)
�io� RawIOBase�__init__�BytesIO�c2s�s2c�handlerrX �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 �|� dS )z.Push (buffer) some data to send to the client.r rn N)r� �tell�seek�write)r �data�posr
r
r r� w s
z_NNTPServerIO.push_datac C s4 | j �� }| j �|� | j �|� | j�� t|�S )z The client sends us some data)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 )z#The client wants to read a responseN)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 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 ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd8d d �Zd d � Zd d!� Zd9d"d#�Zd$d%� Zd&d'� Zdd(�d)d*�Zdd(�d+d,�Zd-Zd.Zed/ e Zd:d0d1�Z d;d2d3�Z d<d4d5�Z d6d7� Z dS )=�
NNTPv1HandlerzA handler for RFC 977z200 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 rX � 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 .
rg Fz
z line doesn't end with \r\n: {!r}���Zhandle_zcommand failed: {!r})r� r� rg �appendZ
body_callbackr� �endswith�
ValueError�format�splitr� �upper�handle_unknown� Exception)r r r} �tokens�cmdrh r
r
r r� � s@
"
z NNTPv1Handler.process_pendingc C s
d| _ dS )z7Flag that the client is expected to post a request bodyTN)r� )r r
r
r � expect_body� s zNNTPv1Handler.expect_bodyc C s | � |� dS )zPush some binary dataN)r� )r r� r
r
r r� � s zNNTPv1Handler.push_datac C s4 t �|�}d�|�� �d }|�d�}| �|� dS )zPush a string literalz
zutf-8N)�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� nf|dkrP| � d� nR|dkr�|d k sdt�|d krx| � 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� �AssertionError)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)rg c C sL |d kr.| j r"| �d� | �� qH| �d� n| j s8t�| �d� || _d S )Nz+340 Input article; end with <CR-LF>.<CR-LF>z440 Posting not permittedz240 Article received OK)r� r� r� r� � posted_body)r rg r
r
r � handle_POST� s
zNNTPv1Handler.handle_POSTc C sT |d kr6| j r*|dkr*| �d� | �� qP| �d� n| j s@t�| �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� )r �
message_idrg 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 Inrr 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 r rX 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� � s:
&
6
r� c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS )
�
NNTPv2Handlerz"A handler for RFC 3977 (NNTP "v2")c 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 USERr6 )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 r � s r c s e Zd ZdZ� fdd�Z� ZS )�CapsAfterLoginNNTPv2Handlerz3A handler that allows CAPABILITIES only after loginc 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 r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ModeSwitchingNNTPv2Handlerz$A server that starts in transit modec 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
.r6 zMODE-)r r� r� )r r r
r
r r s
z.ModeSwitchingNNTPv2Handler.handle_CAPABILITIESc C s& | j s|dkst�d| _ | �d� d S )N�readerTz200 Posting allowed)r r� r� )r Zwhatr
r
r r 0 s z&ModeSwitchingNNTPv2Handler.handle_MODEN)r 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 )Nrn �AUTHINFO�testuser�testpw)ro r8 r �_capsrv r2 )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 rG F s z NNTPv1v2TestsMixin.test_datec C s6 | � | jj� | j�� }| �|d� | �| jj� d S )Nz205 Bye!)r� r� �closedr rx 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� rZ zcomp.lang.python.announceZ
0000001153Z
0000000993�mz *distutils*rn 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& rf r( r) )r r rS 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 rS 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 rD r )r r rS 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.pythonxr6 )r r7 r )r r4 r
r
r r: � 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& rf r( r) )r r rB rC rD r r* �excr
r
r rE � 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� ro r )r r0 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" rV zutf-8u -- Signed by André.r� ra ���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& ra )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& rf r( r) )r r �inforS 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, rS 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 re r r* r% r& rf r( r) )r r r, rS 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 re r r. r r r/ r� r� )r r0 r r, rS 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 rg r r( r% r& rf r( r) )r r r, rS 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 rg r r. r� r r/ r r� )r r0 r r, rS 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)rI rH r. z
message-idrJ z:bytesz:lines�xrefrZ r7 rn rH u0 Re: Message d'erreur incompréhensible (par moi))r r r r )r r � overviewsrS rW r
r
r �check_over_xover_respY s$
z(NNTPv1v2TestsMixin.check_over_xover_respc C s" | j �dd�\}}| �||� d S )Nr6 �; )r rP r9 )r r r8 r
r
r rT n s z NNTPv1v2TestsMixin.test_xoverc C s | j �d�\}}| �||� d S )N)r6 r: )r rW r9 )r r r8 r
r
r rY 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 ) NrU rV 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� r; �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)rC )rD r� )r� r
r
r rE � s )rB r] r � bytearrayr� r� )r r? Z success_respr� r rE 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) rG r rA r� r� r% r& rf 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)
rG r Zihaver% r& rf 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� rZ r r zcomp.lang.python)r- r% r& �
NNTPDataErrorr r# )r r0 r
r
r �test_too_long_lines� s z&NNTPv1v2TestsMixin.test_too_long_lines)%r r r
r� r r rG r r r" r r r r: r= rE r$ r( r* r+ r- r1 r2 r3 r4 r5 r9 rT rY r= r; rB rG rH rI rK r� r
r
)r� r r
6 sB
r
c @ s e Zd ZdZdZeZdd� ZdS )� NNTPv1Testsz*Tests an NNTP v1 server (no capabilities).rZ c C s: | j �� }| �|i � | �| j jd� | �| j jd � d S )NrZ )r rp r ro �nntp_implementation)r rl r
r
r � test_caps� s
zNNTPv1Tests.test_capsN)r r r
r ro r� r� rN r
r
r
r rL � s rL c @ s e Zd ZdZdZeZdd� ZdS )� NNTPv2Testsz,Tests an NNTP v2 server (with capabilities).rn 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 ZHDRrj ZOVERZPOST�READERr) z INN 2.5.1)r rp r ro rM )r rl r
r
r rN � s
zNNTPv2Tests.test_capsN)r r r
r ro r r� rN r
r
r
r rO � s rO c @ s e Zd ZdZdZeZdd� ZdS )�CapsAfterLoginNNTPv2TestszCTests a probably NNTP v2 server with capabilities only after login.rn c C s2 | � | jji � | j�dd� | �d| jj� d S )Nr r
rR )r r r rv r8 )r r
r
r �test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r r r
r ro r r� rU r
r
r
r rT � s rT c @ s e Zd ZdZdZeZdd� ZdS )�SendReaderNNTPv2TestszoSame tests as for v2 but we tell NTTP to send MODE READER to a server
that isn't in READER mode by default.rn c C s | � d| jj� d S )NrS )r8 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
r ro r r� rW r
r
r
r rV � s rV 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>.givesr6 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 r[ 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:linesrH rI r. z
message-idrJ zBytes:zLines:zsubject:zFROM:zDaTe:z message-ID:zBYTES:z Xref:fullzDistribution:fullr7 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 )Nr7 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)rH rI r. z
message-idrJ z:bytesz:linesr7 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: rJ � r6 )r& Z_DEFAULT_OVERVIEW_FMTZ_parse_overviewr )r r r ZoverviewrS 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- )rZ r� �c)r r
r r[ n s z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624i� r � r �8 � �19990623�135624�990623Z090623i� r
)r r[ 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� r0 )r r
r r[ { s
z.MiscTests.test_unparse_datetime.<locals>.givesi� r ra r rb rc rd re i� Z20000623i� r> rZ rn r) Z20100605�010203c s@ t �| ||�}� �t�|�||f� � �t�|d�||f� d S )NF)r- r. r r& rh )ri rj rk r� r� r0 )r r
r r[ � s
�000000r
)r r[ 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& rh ) ri rj rk rl r rm r� r� r0 )r r
r r[ � s z5MiscTests.test_unparse_datetime_legacy.<locals>.givesi� r ra r rb rc rf re i� Z000623i� r> rZ rn r) Z100605rn c s* t �| ||�}� �t�|d�||f� d S )NT)r- r. r r& rh )ri rj rk r� r� r0 )r r
r r[ � s ro r
)r r[ 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
r\ r] r_ rg rp rq r� �
skipUnless�sslrr r
r
r
r rX s (. rX c @ s e Zd ZdZdd� ZdS )�PublicAPITestsz>Ensures that the correct values are exposed in the public API.c C sP | � ttd�� dddddddd g}td k r6|�d
� | �ttj�t|�� d S ) N�__all__r ru �NNTPReplyErrorrf r' �NNTPProtocolErrorrJ rY r� )r r� r& rt r� r �setrv )r Z
target_apir
r
r �test_module_all_attribute� s
z(PublicAPITests.test_module_all_attributeN)r r r
r rz r
r
r
r ru � s ru c @ sH e Zd ZdZejZddd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dS )�MockSocketTestszaTests involving a mock socket object
Used where the _NNTPServerIO file object is not enough.Nc
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
r} r
)r| 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 r| � s r| Fznntplib.socketZdummy)rr rs )r ZassertRaisesRegex�
nntp_classr r )r r� Zexpected_error_typeZexpected_error_msgrv rs r~ r0 r
)r| 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& rx 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& rf 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& rw )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 rv rs 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
r r{ � s
r{ c @ s e Zd ZdZdd� ZdS )�bypass_contextz'Bypass encryption and actual SSL modulec K s | S )Nr
)�sockr� r
r
r � wrap_socket s zbypass_context.wrap_socketN)r 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)rt )r� r Z bind_portZlisten� threadingZThread�
run_serverZ
backgroundrX 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� rt Z
SSLContextZload_cert_chain�certfiler� r� r� )r r� Zclientr Zcleanupr r� �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/ rt Z SSLSocketr% r� )r r� r� r
r
r r� : s
z LocalServerTests.test_starttlsN) r r r
r� r� r� rs rt 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 rt � ImportErrorr� �pathr� �dirname�__file__r� r r� r r� r� r� Z SSLEOFErrorZTestCaser� rs r� r� r� r� r� r� r� r r r r
rL rO rT rV rX ru r{ r� r� r� r �mainr
r
r
r �<module> sx
|
, G
# N<