|
Server : LiteSpeed System : Linux server51.dnsbootclub.com 4.18.0-553.62.1.lve.el8.x86_64 #1 SMP Mon Jul 21 17:50:35 UTC 2025 x86_64 User : nandedex ( 1060) PHP Version : 8.1.33 Disable Function : NONE Directory : /opt/cppython/lib/python3.8/test/__pycache__/ |
U
>��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 z 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
�0/opt/cppython/lib/python3.8/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 s> | j �� \}}| �|�d�|� |D ]}| �tt|�� q$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 r ) r r � GROUP_PATr! r r r r r" r# r
r
r �test_list_active8 s z(NetworkedNNTPTestsMixin.test_list_activec C sB | � tj��}| j�d� W 5 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 r )r2 r � assertNotIn�
GROUP_NAME)�desc�r r
r � _check_descN s z=NetworkedNNTPTestsMixin.test_description.<locals>._check_descz zk.brrtt.baz� )r � descriptionr6 �assertIn�DESCr& r )r r9 r7 r
r8 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 r2 �dictr6 r r; )r r Zdescsr7 r
r
r �test_descriptions] s �
z)NetworkedNNTPTestsMixin.test_descriptionsc C s� | j �| j�}| �dt|�� |\}}}}}| �|| j� | �|t� | �|t� | �|t� | �||� | �|� d�|� d S )N� �211 )
r r" r6 r r! r2 �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 r1 r2 r0 �assertGreaterEqualZyearrE �r r r1 r
r
r � test_dateq s z!NetworkedNNTPTestsMixin.test_datec
C sP | � |t� | �|�� dddddddh� |�� D ]}| � |ttd �f� q2d S )N�subject�fromr1 �
message-id�
references�:bytes�:lines)r2 r@ rK �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 )NrB r zno articles retrieved) r r" r6 �xoverr! �skipTestrK rE rX ) r r rG rH rI �namer �art_numrV 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" r6 �overrX rK rE )
r r rG rH rI r[ �startr r\ rV r
r
r � test_over� s
z!NetworkedNNTPTestsMixin.test_overc C sL | j �| j�\}}}}}| j �d|�\}}|D ]}| �tt|d �� q.d S )NrN � )r r" r6 Zxhdrr r r )r r rG rH rI r[ r r r
r
r � test_xhdr� s z!NetworkedNNTPTestsMixin.test_xhdrNc C sR | � |tj� |d k r$| �|j|� |jD ]}| � |t� q*| �|jd d� d S )Nr_ )� .s .
� .
)r2 r) Z ArticleInfor Znumberr �bytesr5 )r r �articler\ r r
r
r �check_article_resp� s
z*NetworkedNNTPTestsMixin.check_article_respzFIXME: see bpo-32128c
sV | j �| j�\}}}}}|||d fD ]X}z| j �|�\}}W n: tjk
rx } z|j�d�s`� W Y �q&W 5 d }~X Y nX q�q&| �d� | � |�d�|� | �
|||� | j � |�\}} | � |�d�|� | �
|| |� | j � |�\}}
| � |�d�|� | �
||
|� dd� � � fd d
�|j
D �} � fd d
�|
j
D �} | �| | d g | j
� d S )
Nrc 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 �Z blacklistr
r �
<listcomp>� s �zBNetworkedNNTPTestsMixin.test_article_head_body.<locals>.<listcomp>c s g | ]}� |�s|�qS r
r
rl rn r
r ro � s �rk )r r" r6 �headr) �NNTPTemporaryErrorr, r rZ r ri �bodyrh r r )
r r rG rH rI r[ r\ rp �err rh Zfiltered_head_linesZfiltered_linesr
rn 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�LIST� OVERVIEW.FMT)r2 r �tupler< )�capsZ caps_listr8 r
r � _check_caps� s z>NetworkedNNTPTestsMixin.test_capabilities.<locals>._check_caps� )rK r � nntp_version�getcapabilitiesZ capabilities)r ry r rx r
r8 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 5 d |_X d S r )r r �quit)r �clsr
r
r � test_zzquit� s z#NetworkedNNTPTestsMixin.test_zzquitc C sJ dd� }t | �D ]4}|�d�s qt| |�}t|�s4qt| |||�� qd S )Nc s t �� �� fdd��}|S )Nc s$ t �| j�� � | � W 5 Q R X d S r )r �transient_internet� NNTP_HOSTr8 ��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)r� r� r[ r� r
r
r � wrap_methods� s
z$NetworkedNNTPTestsMixin.wrap_methodsc
s� � fdd�}zz| j | jtdd�� � | �|� � | �� �� � W 5 Q R X | �|� � | j | jtdd��� � �� W 5 Q R X | �|� � W nJ tk
r� } z,t� d|j
�r�t � d|� d| j����� W 5 d }~X Y nX d S )Nc s: t � d�sdS z � �� W n ttfk
r4 Y dS X dS )N�fileFT)�hasattrr �OSError�EOFErrorr
�r r
r � is_connected s
zANetworkedNNTPTestsMixin.test_with_statement.<locals>.is_connectedF��timeoutr� �(?i)KEY.TOO.SMALLzGot � connecting to )
�
NNTP_CLASSr� �TIMEOUTr r � assertFalser� 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. r4 r> rA rJ rM rX r] r� ZskipIfrb rd ri rt r} r� r� � classmethodr� r� r
r
r
r r & s0
r c @ s8 e Zd ZdZdZdZdZeZe dd� �Z
e dd� �Z d S )
�NetworkedNNTPTestsznews.trigofacile.com�fr.comp.lang.pythonzfr.comp.lang.*�Pythonc C s� t �d� t �| j��� z| j| jtdd�| _W nv tk
r~ } z0t� d|j
�rlt � | � d|� d| j����� W 5 d }~X Y n* t
k
r� t � | � d| j����Y nX W 5 Q R X d S )NZnetworkFr� r� z got r� z got EOF error on connecting to )r Zrequiresr� r� r� r� r r r� r� r r� r� �
EOF_ERRORS)r� r� r
r
r �
setUpClass2 s
� z NetworkedNNTPTests.setUpClassc C s | j d k r| j �� d S r )r r� )r� r
r
r �
tearDownClassC s
z NetworkedNNTPTests.tearDownClassN) r r r
r� r6 r& r= r r� r� r� r� r
r
r
r r� ) s
r� �requires SSL supportc @ s0 e Zd ZdZdZdZdZeedd�Z dZ
dZ dS )�NetworkedNNTP_SSLTestsz
nntp.aioe.orgz aioe.testzaioe.*�test�NNTP_SSLN) r r r
r� r6 r& r= 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 r )
�io� RawIOBase�__init__�BytesIO�c2s�s2c�handlerra �readline� push_data)r r� r
r
r r� h s
z_NNTPServerIO.__init__c C s dS �NTr
r8 r
r
r �readableq s z_NNTPServerIO.readablec C s dS r� r
r8 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 rz 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 r )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 r ��super�setUp� make_serverr8 �� __class__r
r r� � s
zMockedNNTPTestsMixin.setUpc s t � �� | `d S r )r� �tearDownr r8 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
readermoder� r8 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 ra � s zNNTPv1Handler.startc C s t |dd�S )N�utf-8�surrogateescape)r �r r� r
r
r �_decode� s zNNTPv1Handler._decodec
C s, | j rb| �� }|sd S | j�|� |dkrq.qz | j\}}||d| ji� W 5 d | _d | _d| _ X | �| �� �}|sxd S |�d�s�td�|���|d d� }|� � ^}}t
| d|� � d �}|d kr�| � � qbz ||� W n4 t
k
�r } ztd�|��|�W 5 d }~X Y qbX | j rb||f| _g | _qbd S ) Nrf Frr �
z line doesn't end with \r\n: {!r}���Zhandle_zcommand failed: {!r})r� r� rr �appendZ
body_callbackr� �endswith�
ValueError�format�splitr� �upper�handle_unknown� Exception)r r r� �tokens�cmdrs 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� r8 r
r
r � expect_body� s zNNTPv1Handler.expect_bodyc C s | � |� dS )zPush some binary dataN)r� r� r
r
r r� � s zNNTPv1Handler.push_datac C s4 t �|�}d�|�� �d }|�d�}| �|� dS )zPush a string literalr� r� N)�textwrap�dedent�join�
splitlines�encoder� )r Zlitr
r
r �push_lit� s
zNNTPv1Handler.push_litc C s | � d� d S )Nz 500 What?�r� r8 r
r
r r� � s z NNTPv1Handler.handle_unknownc C s | � | j� d S r )r� r r8 r
r
r r� � s z NNTPv1Handler.handle_welcomec C s | � d� d S �Nz205 Bye!r� r8 r
r
r � handle_QUIT� s zNNTPv1Handler.handle_QUITc C s | � d� d S )N�111 20100914001155r� r8 r
r
r � handle_DATE s zNNTPv1Handler.handle_DATEc C s( |dkr| � d� n| � d�|�� d S )Nr� z$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� r8 r
r
r � handle_HELP s zNNTPv1Handler.handle_HELPNc C sJ |d kr| � d� n2|dkr(| � d� n |dkr<| � d� n
| � d� d S )N�412 No newsgroup selected�3000234�"223 3000234 <45223423@example.com>�<45223423@example.com>� 223 0 <45223423@example.com>�430 No Such Article Foundr� �r Z message_specr
r
r � handle_STAT s zNNTPv1Handler.handle_STATc C s | � d� d S )N�*223 3000237 <668929@example.org> retrievedr� r8 r
r
r � handle_NEXT s zNNTPv1Handler.handle_NEXTc C s | � d� d S )N�,223 3000234 <45223423@example.com> retrievedr� r8 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
.�ACTIVE� *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"
.rv a 215 Order of fields in overview database.
Subject:
From:
Date:
Message-ID:
References:
Bytes:
Lines:
Xref:full
.�
NEWSGROUPS�comp.lang.pythonz� 215 Descriptions in form "group description".
comp.lang.python The Python computer language.
.�comp.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 keywordr� )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 )
Nr Z20100913Z082004a 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 � handle_XOVERw s
�zNNTPv1Handler.handle_XOVER)rr c C sB |d kr.| j r"| �d� | �� q>| �d� n| �d� || _d S )Nz+340 Input article; end with <CR-LF>.<CR-LF>�440 Posting not permitted�240 Article received OK�r� r� r� � posted_body)r rr r
r
r � handle_POST� s
zNNTPv1Handler.handle_POSTc C sJ |d kr6| j r*|dkr*| �d� | �� qF| �d� n| �d� || _d S )N�+<i.am.an.article.you.will.want@example.com>z%335 Send it; end with <CR-LF>.<CR-LF>�435 Article not wanted�235 Article transferred OKr )r �
message_idrr 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 )N�"220 3000237 <45223423@example.com>r � 220 0 <45223423@example.com>r �"220 3000234 <45223423@example.com>r �.)r� �sample_articler 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 )N�"221 3000237 <45223423@example.com>r � 221 0 <45223423@example.com>r �"221 3000234 <45223423@example.com>r r% )r� � sample_headr 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 )N�"222 3000237 <45223423@example.com>r � 222 0 <45223423@example.com>r �"222 3000234 <45223423@example.com>r r% )r� � sample_bodyr 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 Inr~ 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 ra r� r� r� r� r� r� r� r� r� r� r r r
r r r r r r! r+ r0 r& r' r, r1 r3 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 USERr: )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 r )r r r
r
r � handle_OVER s zNNTPv2Handler.handle_OVER)N)r r r
r r7 r9 r: r
r
r
r r4 � s r4 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� r7 r8 r� r
r r7 s z/CapsAfterLoginNNTPv2Handler.handle_CAPABILITIES)r r r
r r7 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)� _switchedr8 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
.r: zMODE-)r> r� r� r5 r
r
r r7 s
z.ModeSwitchingNNTPv2Handler.handle_CAPABILITIESc C s d| _ | �d� d S )NTz200 Posting allowed)r> r� )r Zwhatr
r
r r9 0 s z&ModeSwitchingNNTPv2Handler.handle_MODEN)r r r
r r� r7 r9 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 r )r� r� r8 r� r
r r� 8 s zNNTPv1v2TestsMixin.setUpc C s | � | jj| jj� d S r )r r r r� r8 r
r
r r ; s z NNTPv1v2TestsMixin.test_welcomec C s< | j dkr| �d| jj� | j�dd� | �d| jj� d S )Nrz �AUTHINFO�testuser�testpw)r{ r< r �_capsr� r5 r8 r
r
r �
test_authinfo> s
z NNTPv1v2TestsMixin.test_authinfoc C s: | j �� \}}| �|d� | �|t�dddddd�� d S )Nr� �� � � r � �7 )r r1 r r0 rL r
r
r rM F s z NNTPv1v2TestsMixin.test_datec C s6 | � | jj� | j�� }| �|d� | �| jj� d S r� )r� r� �closedr r� 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� rc �comp.lang.python.announceZ
0000001153Z
0000000993�mr rz 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 5 Q R X | �|jjd� | �tj��}| j �� W 5 Q R X | �|jjd� d S ) N��- r r r r z<non.existent.id>r r )r �statr r( r) rq r+ r, )r r r\ r r- r
r
r � test_stati s z NNTPv1v2TestsMixin.test_statc C s8 | j �� \}}}| �|d� | �|d� | �|d� d S )Nr ��- z<668929@example.org>)r �nextr �r r r\ r r
r
r � test_nexty s z NNTPv1v2TestsMixin.test_nextc C s8 | j �� \}}}| �|d� | �|d� | �|d� d S )Nr rP r )r rI r rU r
r
r � test_last s z NNTPv1v2TestsMixin.test_lastc C s4 | j �d�}| �|d� | j �d�}| �|d� d S )Nr � The Python computer language.�comp.lang.pythonxr: )r r; r )r r7 r
r
r r> � s z#NNTPv1v2TestsMixin.test_descriptionc C sn | j �d�\}}| �|d� | �|ddi� | j �d�\}}| �|ddd�� | j �d�\}}| �|i � d S )Nr z-215 Descriptions in form "group description".rX r z4Announcements about the Python language. (Moderated))r rM rY )r r? r r# r
r
r rA � s ��z$NNTPv1v2TestsMixin.test_descriptionsc C s� | j �d�\}}}}}| �|�d�|� | �|d� | �|d� | �|d� | �|d� | �tj��}| j �d� W 5 Q R X |j}| �|j �d�|j � d S )Nr� rC i� i� i� zcomp.lang.python.develz411 No such group)
r r" r r r r( r) rq r+ r, )r r rG rH rI r" r- �excr
r
r rJ � s �z NNTPv1v2TestsMixin.test_groupc C s� t � dddddd�}| j�d|�\}}d�| j�}| �||� | �|d d
g� t � dddddd�}| j�d |�\}}| �|d � | �|g � d S )
NrE rF �
� � � r zS230 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>r� z)230 An empty list of newsarticles follows)r0 r �newnewsr� r{ r )r r3 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^ r_ r� u -- Signed by André.r� rk ���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! rd r
r
r �_check_article_head� s z&NNTPv1v2TestsMixin._check_article_headc C sH | � t|�d� | �|d d� � | �|dd � � | � |d d� d S )NrF r^ rb rk )r r! rg re rd 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 5 Q R X | �|jjd � d S )
Nr" rS r rP r$ r# r �<non-existent@example.com>r ) r rh r rh r( r) rq r+ r, �r r �infor\ 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� r" rS r �K From: "Demo User" <nobody@example.net>
Subject: I am just a test article
�U This is just a test article.
.Here is a dot-starting line.
-- Signed by André.
)
r� r� r rh r �getvaluer r �asciir� �r �fr rk r\ 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 5 Q R X | �|jjd � d S )
Nr( rS r rP r* r) r ri r ) r rp r rg r( r) rq r+ r, rj 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 )Nrm r( rS r rn ro ) r� r� r rp r rp r r rq r� r� rr 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 5 Q R X | �|jjd � d S )
Nr- rS r rP r/ r. r ri r ) r rr r re r( r) rq r+ r, rj 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 )Nrm r- rS r rn ro ) r� r� r rr r rp r� r rq r r� rr 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 rf 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)rO rN r1 rP rQ rR rS �xrefrc rz rz rN �0 Re: Message d'erreur incompréhensible (par moi))r r r r! )r r � overviewsr\ r` r
r
r �check_over_xover_respY s( �
�z(NNTPv1v2TestsMixin.check_over_xover_respc C s" | j �dd�\}}| �||� d S )Nry �; )r rY r} �r r r| r
r
r r] n s z NNTPv1v2TestsMixin.test_xoverc C s | j �d�\}}| �||� d S )N)ry r~ )r r` r} r r
r
r rb 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 ) Nr^ r_ rf r� s -- Signed by André.
ra �
rb s ..Here is a dot-starting line.
r s( From: "Demo User" <nobody@example.net>
)r� r r r! rd r
r
r �_check_posted_body� s z%NNTPv1v2TestsMixin._check_posted_bodyc G sd | j }|||�f }d | j_||� }| �� | j �dd�}|||�f }d | j_||� }| �� |S )Nr� �
)� 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 )NFr� r� r� r
r
r r� � s )r� rg r � bytearrayr� r� )r r� Z success_respr� r r� r
r
r �check_post_ihave� s z#NNTPv1v2TestsMixin.check_post_ihavec C sR | � | jjd� d| j_| �tj��}| j�| j� W 5 Q R X | � |j
j d� d S )Nr Fr ) r� r r� r� r� r( r) rq r� r r+ r, �r r- r
r
r � test_post� s
�z NNTPv1v2TestsMixin.test_postc C sN | � | jjdd� | �tj��}| j�d| j� W 5 Q R X | �|jj d� d S )Nr r z<another.message.id>r )
r� r Zihaver( r) rq r� r r+ r, r� r
r
r �
test_ihave� s �
�z NNTPv1v2TestsMixin.test_ihavec C s. t � dddddd�}| �tj| jjd|� d S )NrE rc rF r r )r0 r( r) �
NNTPDataErrorr r_ )r r3 r
r
r �test_too_long_lines� s �z&NNTPv1v2TestsMixin.test_too_long_lines)%r r r
r� r rD rM rK r r% rR rV rW r> rA rJ r` re rg rh rl rt ru rv rw rx r} r] rb r� r� r� r� r� r� r� r� r
r
r� r r? 6 sD
� r? c @ s e Zd ZdZdZeZdd� ZdS )� NNTPv1Testsz*Tests an NNTP v1 server (no capabilities).rc c C s: | j �� }| �|i � | �| j jd� | �| j jd � d S )Nrc �r r| r r{ Znntp_implementation�r rx r
r
r � test_caps� s
zNNTPv1Tests.test_capsN)r r r
r r{ r� r� r� r
r
r
r r� � s r� c @ s e Zd ZdZdZeZdd� ZdS )� NNTPv2Testsz,Tests an NNTP v2 server (with capabilities).rz 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 rv )�VERSIONZIMPLEMENTATIONr@ ZHDRru ZOVERZPOST�READERrf z INN 2.5.1r� r� r
r
r r� � s"
�� zNNTPv2Tests.test_capsN)r r r
r r{ r4 r� r� r
r
r
r r� � s r� c @ s e Zd ZdZdZeZdd� ZdS )�CapsAfterLoginNNTPv2TestszCTests a probably NNTP v2 server with capabilities only after login.rz c C s2 | � | jji � | j�dd� | �d| jj� d S )NrA rB r� )r r rC r� r< r8 r
r
r �test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r r r
r r{ r; r� r� r
r
r
r r� � s r� 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.rz c C s | � d| jj� d S )Nr� )r< r rC r8 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 r{ r< r� r� r
r
r
r r� � s r� 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 r )r r) �
decode_header)�ar� r8 r
r �gives s z+MiscTests.test_decode_header.<locals>.givesr: 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 matricer{ r
�r r� r
r8 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:rR rS rN rO r1 rP rQ zBytes:zLines:zsubject:zFROM:zDaTe:z message-ID:zBYTES:z Xref:fullzDistribution:fullrz Z distributionz Xref:FULLzDistribution:FULL)r r) Z_parse_overview_fmtrd 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 )Nrz 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:3000363rP zI am just a test articlez "Demo User" <nobody@example.com>z6 Oct 1998 04:38:40 -0500r z<45454@example.net>Z1234Z17z"news.example.com misc.test:3000363)rN rO r1 rP rQ rR rS rz 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: rQ � r: )r) Z_DEFAULT_OVERVIEW_FMTZ_parse_overviewr )r r6 r Zoverviewr\ �fieldsr
r
r �test_parse_overview? s6 �
�
�
�
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 r )r r) Z_parse_datetimer0 )r� r� �cr8 r
r r� n s �z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624�� rL � r[ �8 � �19990623�135624�990623Z090623i� r
r� r
r8 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 r= �r0 r r) �_unparse_datetime� �y�M�d�hrN �sr r r3 r8 r
r r� { s ��z.MiscTests.test_unparse_datetime.<locals>.givesr� rL r� r[ r� r� r� r� �� Z20000623rE rB rc rz rf Z20100605�010203c s@ t �| ||�}� �t�|�||f� � �t�|d�||f� d S r= �r0 r1 r r) r� �r� r� r� r r r3 r8 r
r r� � s ���000000r
r� r
r8 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 r� r� r� r8 r
r r� � s �z5MiscTests.test_unparse_datetime_legacy.<locals>.givesr� rL r� r[ r� r� r� r� r� Z000623rE rB rc rz rf Z100605r� c s* t �| ||�}� �t�|d�||f� d S r� r� r� r8 r
r r� � s �r� r
r� r
r8 r � test_unparse_datetime_legacy� s z&MiscTests.test_unparse_datetime_legacyr� c C s | � ttd�� d S )Nr� )r r� r) r8 r
r
r �test_ssl_support� s zMiscTests.test_ssl_supportN)
r r r
r� r� r� r� r� r� r� �
skipUnless�sslr� r
r
r
r r� s (.
r� 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 r� �NNTPReplyErrorrq r* �NNTPProtocolErrorr� r� r� )r r� r) r� r� r �setr� )r Z
target_apir
r
r �test_module_all_attribute� s �
z(PublicAPITests.test_module_all_attributeN)r r r
r r� r
r
r
r r� � s r� 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 5 Q R X W 5 Q R X | ��� �D ]}| �|j� q|d S ) Nc s e Zd Z� fdd�ZdS )zNMockSocketTests.check_constructor_error_conditions.<locals>.mock_socket_modulec s � � S r r
)�addressr� ��
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 r� r
r8 )�
socket_closedr
r �close� s zLMockSocketTests.check_constructor_error_conditions.<locals>.MockSocket.closec s �� }t |�\}}� �|� |S r )r� r� )�socket�moder� r8 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.socket�dummy)r~ r )r ZassertRaisesRegex�
nntp_classr rJ )r r� Zexpected_error_typeZexpected_error_msgr� r r� rs 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) r� 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 unavailableNr� r
r
r
r r� � s r� )r� r� r) rq 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 unavailableNr� r
r
r
r r� � s 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 r r� r8 �Zcapabilities_responser
r r7 � s zJMockSocketTests.test_bad_capabilities.<locals>.Handler.handle_CAPABILITIESN)r r r
r7 r
r� r
r r� � s r� z201 bad capability)r� r� r) 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 r r� )r r� �Zauthinfo_responser
r r3 � s zCMockSocketTests.test_login_aborted.<locals>.Handler.handle_AUTHINFON)r r r
r3 r
r� r
r r� � s r� z 503 Mechanism not recognized)r� r� r) r* )r r� r 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 r r
)�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)r� )r� r Z bind_port�listen� threading�Thread�
run_serverZ
backgroundra Z
addCleanupr� r ZHOST� __enter__�nntp�__exit__)r r� �portr
r
r r� s
�
zLocalServerTests.setUpc C s� |� |� � \}}W 5 Q R X t�� ��}|�|� |�|�d��}|�d� |�� }|dkrf|�d� qJ|dkr�|�� |�d� t� � }|�
t � |j |dd�}|�|� |�|�d��}qJ|d kr�|�d
� q�qJt
d �|���qJW 5 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})�accept�
contextlib� ExitStack�
enter_contextr� �sendallr� r� r� Z
SSLContextZload_cert_chain�certfiler� r� r� )r r� �clientr8 �cleanup�readerr� �contextr
r
r r� s4
�
�
zLocalServerTests.run_serverr� c C s` | j j}| j j}| j �� | �|| j j� | �|| j j� | �| j jtj� | �t | j j� d S r )
r� r� r� ZstarttlsZassertNotEqualr2 r� Z SSLSocketr( r� )r r� r� r
r
r r� : s
z LocalServerTests.test_starttlsN) r r r
r� r� r� r� r� r� r
r
r
r r�
s
r� �__main__);r� r� r0 r� r� r� r� Zos.path�osr� r� r� r r) r r Z
unittest.mockr r� � ImportErrorr� �pathr� �dirname�__file__r� r r� r r� r� r� Z SSLEOFErrorZTestCaser� r� r� r� r� r� r� r� r� r4 r; r< r? r� r� r� r� r� r� r� r� r� r� r �mainr
r
r
r �<module> sz
|
, G
� # N
<