Your IP : 216.73.216.213
3
��2a� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
mZ d dl Z d dlmZ y d dlZW n ek
r� dZY nX y d dlZW n ek
r� dZY nX dZejjejje�d�Zedk r�ejZnG dd� de�ZG d d
� d
�Z e j � G d d � d e ej �Z ej 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 �Z-G d&d'� d'e%ej �Z.G d(d)� d)e&ej �Z/G d*d+� d+ej �Z0G d,d-� d-ej �Z1G d.d/� d/ej �Z2G d0d1� d1�Z3ej ed
�G d2d3� d3e2��Z4ej ed4�G d5d6� d6ej ��Z5e6d7k�r�ej7� dS )8� N)�support)�NNTP� GroupInfo)�patch� z keycert3.pemc @ s e Zd ZdZdS )�SSLErrorzThis will never be raised.N)�__name__�
__module__� __qualname__�reason� r r �1/opt/alt/python36/lib64/python3.6/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� Zejdd�dd � �Zd d � Zd/d d!�Zejdd"�d#d$� �Zd%d&� Zd'd(� Zd)d*� Zed+d,� �Zd-d.� Zd S )0�NetworkedNNTPTestsMixinc C s | j j� }| jtt|�� d S )N)�serverZ
getwelcome� assertEqual�str�type)�self�welcomer r r
� test_welcome* s
z$NetworkedNNTPTestsMixin.test_welcomec C sB | j j� \}}| j|jd�|� x |D ]}| jtt|�� 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 j� \}}t|�dkrD| jtt|d �� | jtt|d j�� d S )Nr )r �list�lenr r r r �group)r r �groupsr r r
� test_list4 s z!NetworkedNNTPTestsMixin.test_listc C sL | j j| j�\}}t|�dkrH| jtt|d �� | jtt|d j�� d S )Nr ) r r � GROUP_PATr r r r r r )r r r r r r
�test_list_active: s z(NetworkedNNTPTestsMixin.test_list_activec
C sB | j tj��}| jjd� W d Q R X |jj}| j|jd�|� 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 jj� t jdd� }| jj|�\}}t|�dkrT| j|d t� | j|d j t
� d S )N� )Zdaysr ) �datetime�dateZtodayZ timedeltar Z newgroupsr �assertIsInstancer r r )r �dtr r r r r
�test_newgroupsF s
z&NetworkedNNTPTestsMixin.test_newgroupsc sb � fdd�}� j j� j�}||� � j� j|� � j j� j�}||� � j jd�}� j|d� d S )Nc s � j | t� � j� j| � d S )N)r. r � assertNotIn�
GROUP_NAME)�desc)r r r
� _check_descP s z=NetworkedNNTPTestsMixin.test_description.<locals>._check_descz zk.brrtt.baz� )r � descriptionr2 �assertIn�DESCr" r )r r4 r3 r )r r
�test_descriptionO s z(NetworkedNNTPTestsMixin.test_descriptionc C s^ | j j| j�\}}| j|jd�p(|jd�|� | j|t� || j }| j|| 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| j�}| jdt|�� |\}}}}}| j|| j� | j|t� | j|t� | j|t� | j||� | j|j d�|� d S )N� z211 )
r r r2 r r r. �int�assertLessEqualr r )r �resultr �count�first�lastr r r r
�
test_grouph s z"NetworkedNNTPTestsMixin.test_groupc C s< | j j� \}}| j|tj� | j|jd� | j|jd� d S )Ni� i� )r r- r. r, �assertGreaterEqualZyearr? )r r r- r r r
� test_dates s z!NetworkedNNTPTestsMixin.test_datec C sT | j |t� | j|j� dddddddh� x$|j� D ]}| j |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_dictz s
z'NetworkedNNTPTestsMixin._check_art_dictc C sz | j j| j�\}}}}}| j j|d |�\}}t|�dkrD| jd� |d \}}| j||d � | j||� | j|� 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| j�\}}}}}|d }| j j|d f�\}}|d \}} | j| � | j j||f�\}}|d \}} | j||� | j||� | j| � 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�\}}}}}| j jd|�\}}x |D ]}| jtt|d �� q0W d S )NrG rU )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 | j |tj� |d k r$| j|j|� x|jD ]}| j |t� q,W | j|jd d� d S )NrU � .� .
� .
rV )r[ r\ r] )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| j�\}}}}}xl|||d fD ]P}y| j j|�\}}W n4 tjk
rt } z|jjd�sb� w(W Y d d }~X nX P q(W | jd� | j |jd�|� | j
|||� | j j |�\}} | j |jd�|� | j
|| |� | j j |�\}}
| j |jd�|� | j
||
|� dd� � � fd d
�|j
D �} � fd d
�|
j
D �} | j| | d g | j
� d S )
NrU z423 z(could not find a suitable article numberz221 z222 z220 c S s
| j d�S )Ns X-Antivirus)r )r r r r
�<lambda>� s 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 � )r r r2 �headr% �NNTPTemporaryErrorr( r rP r r` �bodyr_ r r )
r r rA rB rC rQ rR rf �erh r_ Zfiltered_head_linesZfiltered_linesr )rc r
�test_article_head_body� s,
z.NetworkedNNTPTestsMixin.test_article_head_bodyc sD � fdd�}� j � jjd� |� jj� � � jj� \}}||� d S )Nc s( | d }� j |ttf� � jd|� 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 rn r rm 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| jj� 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}|jd�s"qt| |�}t|�s6qt| |||�� qW d S )Nc s t j� �� fdd��}|S )Nc s$ t j| 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)rz r� rQ r~ r r r
� wrap_methods� s
z$NetworkedNNTPTestsMixin.wrap_methodsc s� � fdd�}yz| j | jtdd�� � | j|� � | j� j� � W d Q R X | j|� � | j | jtdd��� � j� W d Q R X | j|� � W nJ tk
r� } z.tj d|j
�r�t j d|� d| j����� W Y d d }~X nX d S )Nc s8 t � d�sdS y � j� 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)�timeoutru z(?i)KEY.TOO.SMALLzGot z connecting to )
�
NNTP_CLASSr} �TIMEOUTr r � assertFalsery 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� ZskipIfrY rZ r` rj rr rx r{ � 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 jd� t j| j��� y| j| jtdd�| _W nv tk
r~ } z2tj d|j
�rlt j | � d|� d| j����� W Y d d }~X n* t
k
r� t j | � d| j����Y nX W d Q R X d S )NZnetworkF)r� ru 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� r� )rz r� r r r
�
setUpClass/ s
z NetworkedNNTPTests.setUpClassc C s | j d k r| j j� d S )N)r ry )rz r r r
�
tearDownClass@ 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� E s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
_NNTPServerIOc C s> t jj| � t j� | _t j� | _|| _| jj| jj| j � d S )N)
�io� RawIOBase�__init__�BytesIO�c2s�s2c�handlerrX �readline� push_data)r r� r r r
r� e s
z_NNTPServerIO.__init__c C s dS )NTr )r r r r
�readablen s z_NNTPServerIO.readablec C s dS )NTr )r r r r
�writableq s z_NNTPServerIO.writablec C s4 | j j� }| j jdd� | j j|� | j j|� d S )Nr ro )r� �tell�seek�write)r �data�posr r r
r� t s
z_NNTPServerIO.push_datac C s4 | j j� }| j j|� | j j|� | jj� t|�S )N)r� r� r� r� r� �process_pendingr )r �br� r r r
r� { s
z_NNTPServerIO.writec C s2 | j j� | jjt|��}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� ` s r� c C s t | �}tj||�}||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 � j� | j� d S )N)�super�setUp� make_server)r )� __class__r r
r� � s
zMockedNNTPTestsMixin.setUpc s t � j� | `d S )N)r� �tearDownr )r )r� r r
r� � s
z MockedNNTPTestsMixin.tearDownc O s6 | j � | _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 � j� | 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| _| j� 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*| j� }|sd S | jj|� |dkrP qW z | j\}}||d| ji� W d d | _d | _d| _ X x�| j| j� �}|s~d S |jd�s�tdj|���|d d � }|j � ^}}t
| d|j � d �}|d kr�| j � qhy ||� W n4 t
k
�r } ztdj|��|�W Y d d }~X qhX | j rh||f| _g | _qhW d S )
Ns .
rh Fz
z line doesn't end with \r\n: {!r}ro Zhandle_zcommand failed: {!r}���)r� r� rh �appendZ
body_callbackr� �endswith�
ValueError�format�splitr� �upper�handle_unknown� Exception)r r r~ �tokens�cmdri 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 | j |� d S )N)r� )r r� r r r
r� � s zNNTPv1Handler.push_datac C s4 t j|�}dj|j� �d }|jd�}| j|� d S )Nz
zutf-8)�textwrap�dedent�join�
splitlines�encoder� )r Zlitr r r
�push_lit� s
zNNTPv1Handler.push_litc C s | j d� d S )Nz 500 What?)r� )r r r r
r� � s z NNTPv1Handler.handle_unknownc C s | j | j� d S )N)r� r )r r r r
r� � s z NNTPv1Handler.handle_welcomec C s | j d� d S )Nz205 Bye!)r� )r r r r
� handle_QUIT� s zNNTPv1Handler.handle_QUITc C s | j d� d S )Nz111 20100914001155)r� )r r r r
� handle_DATE� s zNNTPv1Handler.handle_DATEc C s( |dkr| j d� n| j dj|�� 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 | j 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| j d� n2|dkr(| j d� n |dkr<| j d� n
| j 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 | j d� d S )Nz*223 3000237 <668929@example.org> retrieved)r� )r r r r
� handle_NEXT s zNNTPv1Handler.handle_NEXTc C s | j 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| j d� n�|dkr<|dkr0| j d� q�| j d� nZ|dkrP| j d� nF|dkr�|d krl| j d
� q�|d kr�| j d � q�| j d
� n
| j 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$| j d� nZ|dkrH|dkrH|dkrH| j d� n6|dkrt|dkrt|d krt| j d
d d � n
| j 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>
.�20100101�100101Z090000z
too long linei� z
.zK 230 An empty list of newsarticles follows
.)r� r� )r� )r r �date_str�time_strr r r
�handle_NEWNEWSU s
z NNTPv1Handler.handle_NEWNEWSc C s" |dkr| j d� n
| j d� d S )Nz57-59uL 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.org 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_XOVERt s
zNNTPv1Handler.handle_XOVER)rh c C sB |d kr.| j r"| jd� | j� q>| jd� n| jd� || _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 rh r r r
� handle_POST� s
zNNTPv1Handler.handle_POSTc C sJ |d kr6| j r*|dkr*| jd� | j� qF| jd� n| jd� || _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_idrh 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| j d� n6|dkr(| j d� n"|dkr<| j d� n| j d� d S | j | j� | 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| j d� n6|dkr(| j d� n"|dkr<| j d� n| j d� d S | j | j� | 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| j d� n6|dkr(| j d� n"|dkr<| j d� n| j d� d S | j | j� | 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| jd� nT|dkr>| jr,| jd� qf| jd� d| _n(|dkrX| jd� d| _ ntdj|���d S ) Nz502 Already Logged Inrs 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 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� � s8
&
6
r� c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) �
NNTPv2Handlerc C s0 d}| j s | j|jd�� n| j|jd�� 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
| j |�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| jd� n
t� j� 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 | j|jd�� n| j|jd�� 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| _ | jd� d S )NTz200 Posting allowed)r r� )r Zwhatr r r
r - 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 � j� d S )N)r� r� )r )r� r r
r� 5 s zNNTPv1v2TestsMixin.setUpc C s | j | jj| jj� d S )N)r r r r� )r r r r
r 8 s z NNTPv1v2TestsMixin.test_welcomec C s< | j dkr| jd| jj� | jjdd� | jd| jj� d S )Nro �AUTHINFO�testuser�testpw)rp r7 r �_capsrw r1 )r r r r
�
test_authinfo; s
z NNTPv1v2TestsMixin.test_authinfoc C s: | j j� \}}| j|d� | j|tjdddddd�� d S )Nz111 20100914001155i� � � r � �7 )r r- r r, )r r r- r r r
rF C s z NNTPv1v2TestsMixin.test_datec C s6 | j | jj� | jj� }| j|d� | j| jj� d S )Nz205 Bye!)r� r� �closedr ry r r )r r r r r
� test_quitH s
z NNTPv1v2TestsMixin.test_quitc C s2 | j j� \}}| j|d� | j|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 N s z NNTPv1v2TestsMixin.test_helpc C s~ | j j� \}}| jt|�d� |d }| j|tdddd�� | j jd�\}}| jt|�d� |d }| j|td
d d d�� d S )
N� rU zcomp.lang.python.announceZ
0000001153Z
0000000993�mz *distutils*ro r z!gmane.comp.python.distutils.develZ
0000014104Z
0000000001)r r r r r )r r r �gr r r
r! X s
z NNTPv1v2TestsMixin.test_listc C s� | j jd�\}}}| j|d� | j|d� | j|d� | j jd�\}}}| j|d� | j|d� | j|d� | jtj��}| j jd� W d Q R X | j|jjd� | jtj��}| j j� W d Q R X | j|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% rg r' r( )r r rR r� r) r r r
� test_statf s z NNTPv1v2TestsMixin.test_statc C s8 | j j� \}}}| j|d� | j|d� | j|d� d S )Nz*223 3000237 <668929@example.org> retrievedi��- z<668929@example.org>)r �nextr )r r rR r� r r r
� test_nextv s z NNTPv1v2TestsMixin.test_nextc C s8 | j j� \}}}| j|d� | j|d� | j|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 jd�}| j|d� | j jd�}| j|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 jd�\}}| j|d� | j|ddi� | j jd�\}}| j|ddd�� | j jd�\}}| j|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 jd�\}}}}}| j|jd�|� | j|d� | j|d� | j|d� | j|d� | jtj��}| j jd� W d Q R X |j}| j|j jd�|j � d S )Nzfr.comp.lang.pythonz211 i� i� i� zcomp.lang.python.develz411 No such group)
r r r r r r$ r% rg r' r( )r r rA rB rC r r) �excr r r
rD � s z NNTPv1v2TestsMixin.test_groupc C s� t j dddddd�}| jjd|�\}}dj| j�}| j||� | j|d d
g� t j dddddd�}| jjd |�\}}| j|d � | j|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� rp r )r r/ r ZidsZexpectedr r r
� test_newnews� s z NNTPv1v2TestsMixin.test_newnewsc C sZ | j t|�d� | j |d
jd�d� | j |d d� | j |d d� | j |d
d � d S )Nr" rU zutf-8u -- Signed by André.ro re � s .Here is a dot-starting line.s This is just a test article.rV r� ������)r r �decode)r r r r r
�_check_article_body� s
z&NNTPv1v2TestsMixin._check_article_bodyc C s4 | j t|�d� | j |d d� | j |d d� d S )Nr" r s& From: "Demo User" <nobody@example.net>r% 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 | j t|�d� | j|d d� � | j|dd � � | j |d d� d S )Nr r" re r' )r r r* r) )r r r r r
�_check_article_data� s z&NNTPv1v2TestsMixin._check_article_datac
C s | j j� \}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | jtj��}| j jd� W d Q R X | j|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% rg r' r( )r r �inforR r� r r) r r r
� test_article� s*
z NNTPv1v2TestsMixin.test_articlec C s� t j� }| jj|d�\}}| j|d� |\}}}| j|d� | j|d� | j|g � |j� }| j|jd�t|�� | j|j 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 j� \}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | jtj��}| j jd� W d Q R X | j|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 rf r r* r$ r% rg r' r( )r r r, rR r� r r) r r r
� test_head� s*
z NNTPv1v2TestsMixin.test_headc C s� t j� }| jj|d�\}}| j|d� |\}}}| j|d� | j|d� | j|g � |j� }| j|jd�t|�� | j |j
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 rf r r. r r r/ r� r� )r r0 r r, rR r� r r� r r r
�test_head_file s
z!NNTPv1v2TestsMixin.test_head_filec
C s | j j� \}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | j jd�\}}| j|d� |\}}}| j|d� | j|d� | j|� | jtj��}| j jd� W d Q R X | j|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 rh r r) r$ r% rg r' r( )r r r, rR r� r r) r r r
� test_body' s*
z NNTPv1v2TestsMixin.test_bodyc C s� t j� }| jj|d�\}}| j|d� |\}}}| j|d� | j|d� | j|g � |j� }| j|jd�t|�� | j |j
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 rh r r. r� r r/ r r� )r r0 r r, rR r� r r� r r r
�test_body_fileB s
z!NNTPv1v2TestsMixin.test_body_filec C s� | j |jd�|� | jt|�d� |d \}}| j|d� | j|ddddd d
d d d
�� |d \}}| j|d d � |d \}}| j|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.org gmane.comp.python.authors:57)rH rG r- z
message-idrI z:bytesz:lines�xrefrU r7 ro rG u0 Re: Message d'erreur incompréhensible (par moi))r r r r )r r � overviewsrR rW r r r
�check_over_xover_respV s$
z(NNTPv1v2TestsMixin.check_over_xover_respc C s" | j jdd�\}}| j||� d S )Nr6 �; )r rO r9 )r r r8 r r r
rS k s z NNTPv1v2TestsMixin.test_xoverc C s | j jd�\}}| j||� d S )Nr6 r: )r6 r: )r rW r9 )r r r8 r r r
rY o 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}| jt|�d� | j|d d� | j|d
d� | j|d d� | j|d d � | j|d
d � d S )NrT rU s .
ro s -- Signed by André.
r% s
r" s ..Here is a dot-starting line.
r s( From: "Demo User" <nobody@example.net>
rV r� r& r' )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� | j jdd�}|||�f }d | j_||� }| 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||� | j |f|�dti�}| j||� | j |f|�dtji�}| j||� dd� }| j |f|�d|i�}| j||� dd� }| j |f|�d|i�}| j||� 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 | j | jjd� d| j_| jtj��}| jj| j� W d Q R X | j |j
j d� d S )Nz240 Article received OKFz440 Posting not permitted) rG r rA r� r� r$ r% rg r= r r' r( )r r) r r r
� test_post� s
z NNTPv1v2TestsMixin.test_postc
C sN | j | jjdd� | jtj��}| jjd| j� W d Q R X | j|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% rg r= r r' r( )r r) r r r
�
test_ihave� s
z NNTPv1v2TestsMixin.test_ihavec C s. t j dddddd�}| jtj| jjd|� d S )Ni� rU 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- r1 r2 r3 r4 r5 r9 rS rY r= r; rB rG rH rI rK r� r r )r� r
r
3 sB
r
c @ s e Zd ZdZeZdd� ZdS )� NNTPv1TestsrU c C s: | j j� }| j|i � | j| j jd� | j| j jd � d S )NrU )r rq r rp �nntp_implementation)r rm r r r
� test_caps� s
zNNTPv1Tests.test_capsN)r r r
rp r� r� rN r r r r
rL � s rL c @ s e Zd ZdZeZdd� ZdS )� NNTPv2Testsro c C sb | j j� }| j|ddgddgdgg dddd d
d gg g g d �� | j| j jd
� | j| j jd� d S )N�2�3ZINNz2.5.1ZUSERr� z ACTIVE.TIMESz DISTRIB.PATSZHEADERSr� z OVERVIEW.FMT)�VERSIONZIMPLEMENTATIONr ZHDRrk ZOVERZPOST�READERr% z INN 2.5.1)r rq r rp rM )r rm r r r
rN � s
zNNTPv2Tests.test_capsN)r r r
rp r r� rN r r r r
rO � s rO c @ s e Zd ZdZeZdd� ZdS )�CapsAfterLoginNNTPv2Testsro c C s2 | j | jji � | jjdd� | jd| jj� d S )Nr r
rR )r r r rw r7 )r r r r
�test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r r r
rp r r� rU r r r r
rT � s rT c @ s e Zd ZdZeZdd� ZdS )�SendReaderNNTPv2Testsro c C s | j d| jj� d S )NrS )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
rp r r� rW r r r r
rV � s rV c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
e d
�dd� �Z dS )� MiscTestsc sV � fdd�}|dd� |dd� |dd� |dd� |dd � |d
d � |d d � d S )
Nc s � j tj| �|� 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 r[ r )r r
�test_decode_header s
z MiscTests.test_decode_headerc C sD dddddddg}| j tj|�dd d
d d ddg� dddddd
dg}| j tj|�dd d
d d ddg� dddddddg}| j tj|�dd d
d d ddg� dddddddddg }| j tj|�dd d
d d ddddg � dddddd
dddg }| j tj|�dd d
d d ddddg � dddddd
ddg}| j tj|�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: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 j||�}|\\}}| j|d� | j|dddddd d
d d �� d
g}t j||�}|\\}}| j|d d � dg}t j||�}|\\}}| j|d d� | j|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)rG rH r- z
message-idrI 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: 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 � j tj| |�tj|� � d S )N)r r% Z_parse_datetimer, )rZ r� �c)r r r
r[ k s z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624i� r � r �8 � �19990623�135624�990623Z090623i� r )r r[ r )r r
�test_parse_datetimej 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 j | |||||�}� jtj|�||f� � jtj|d�||f� d S )NF)r, r r% �_unparse_datetime) �y�M�d�hr �sr� r� r/ )r r r
r[ x s
z.MiscTests.test_unparse_datetime.<locals>.givesi� r ra r rb rc rd re i� Z20000623i� r= rU ro r% Z20100605�010203c s@ t j| ||�}� jtj|�||f� � jtj|d�||f� d S )NF)r, r- r r% rh )ri rj rk r� r� r/ )r r r
r[ � s
�000000r )r r[ r )r r
�test_unparse_datetimeu 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 j | |||||�}� jtj|d�||f� d S )NT)r, r r% rh ) ri rj rk rl r rm r� r� r/ )r r r
r[ � s z5MiscTests.test_unparse_datetime_legacy.<locals>.givesi� r ra r rb rc rf re i� Z000623i� r= rU ro r% Z100605rn c s* t j| ||�}� jtj|d�||f� d S )NT)r, r- r r% rh )ri rj rk r� r� r/ )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 | j 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 Zdd� ZdS )�PublicAPITestsc C sP | j ttd�� dddddddd g}td k r6|jd
� | jttj�t|�� d S ) N�__all__r rv �NNTPReplyErrorrg 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
rz r r r r
ru � s ru 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|��. | j||�� | jd||d� W d Q R X W d Q R X | j�� x�D ]}| j|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 |�\}}� j|� |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)rs rt )r ZassertRaisesRegex�
nntp_classr r )r r� Zexpected_error_typeZexpected_error_msgrw rt r~ r0 r )r| r� r� r r
�"check_constructor_error_conditions� s $
z2MockSocketTests.check_constructor_error_conditionsc C s& G dd� dt �}| j|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 �}| j|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% rg r )r r� r r r
�$test_service_temporarily_unavailable� s z4MockSocketTests.test_service_temporarily_unavailablec C s& G dd� dt �}| j|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� | j|tj� � d S )Nc s e Zd Z� fdd�ZdS )z6MockSocketTests.test_bad_capabilities.<locals>.Handlerc s | j � � 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� | j|tj� ||� d S )Nzt@e.com�pythonc s e Zd Z� fdd�ZdS )z3MockSocketTests.test_login_aborted.<locals>.Handlerc s | j � � 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 rw rt 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{ � s
r{ 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� zrequires multithreadingc @ s0 e Zd Zdd� Zdd� Zejed�dd� �ZdS ) �LocalServerTestsc C st t j � }tj|�}|j� tj| j|fd�| _| jj� | j | jj
� t tj |dd�j
� | _| j | jjd d d � d S )N)�targetr� F)ru )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� |� |j � \}}W d Q R X tj� ��}|j|� |j|jd��}|jd� x�|j� }|dkrh|jd� qL|dkr�|j� |jd� tj � }|j
t � |j |dd�}|j|� |j|jd��}qL|d kr�|jd
� P qLt
d j|���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 Zcleanup�readerr� �contextr r r
r� s0
zLocalServerTests.run_serverzrequires SSL supportc C s` | j j}| j j}| j j� | j|| j j� | j|| j j� | j| j jtj� | jt | j j� d S )N)
r� r� r� ZstarttlsZassertNotEqualr. rt Z SSLSocketr$ r� )r r� r� r r r
r� 7 s
z LocalServerTests.test_starttlsN) r r r
r� r� r� rs rt r� r r r r
r� s r� �__main__)8r� r� r, r� r� r� r� Zos.path�osr� r� r r% r r Z
unittest.mockr rt � ImportErrorr� r� �pathr� �dirname�__file__r� r r� r r� ZTestCaser� 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> sz
| , G
# N
<