Your IP : 216.73.216.85
��Yf� � @ sw d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
d d l m Z m
Z
d d l Z d d l m Z y d d l Z Wn e k
r� d Z Yn Xy d d l Z Wn e k
r� d Z Yn Xd Z e j j e j j e � d � Z Gd d � d � Z e j � Gd d
� d
e e j � Z e j e d � Gd d
� d
e � � Z Gd d � d e j � Z d d � Z! Gd d � d � Z" Gd d � d e" � Z# Gd d � d � Z$ Gd d � d e$ � Z% Gd d � d e% � Z& Gd d � d e% � Z' Gd d � d � Z( Gd d! � d! e( e" e j � Z) Gd" d# � d# e( e" e j � Z* Gd$ d% � d% e" e j � Z+ Gd& d' � d' e# e j � Z, Gd( d) � d) e j � Z- Gd* d+ � d+ e j � Z. Gd, d- � d- e j � Z/ Gd. d/ � d/ � Z0 e j e d � Gd0 d1 � d1 e/ � � Z1 e j e d2 � Gd3 d4 � d4 e j � � Z2 e3 d5 k rse j4 � d S)6� N)�support)�NNTP� GroupInfo)�patch� z keycert3.pemc @ s' e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z e j d d � d d � � Z d d � Z d d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z e d* d+ � � Z d, d- � Z d S).�NetworkedNNTPTestsMixinc C s) | j j � } | j t t | � � d S)N)�serverZ
getwelcome� assertEqual�str�type)�self�welcome� r �6/opt/alt/python35/lib64/python3.5/test/test_nntplib.py� test_welcome! s z$NetworkedNNTPTestsMixin.test_welcomec C sY | j j � \ } } | j | j d � | � x$ | D] } | j t t | � � q5 Wd S)Nz100 )r �help�
assertTrue�
startswithr r
r )r �resp�lines�liner r r � test_help% s
z!NetworkedNNTPTestsMixin.test_helpc C sb | j j � \ } } t | � d k r^ | j t t | d � � | j t t | d j � � d S)Nr )r �list�lenr r r r
�group)r r �groupsr r r � test_list+ s z!NetworkedNNTPTestsMixin.test_listc C sh | j j | j � \ } } t | � d k rd | j t t | d � � | j t t | d j � � d S)Nr ) r r � GROUP_PATr r r r r
r )r r r r r r �test_list_active1 s z(NetworkedNNTPTestsMixin.test_list_activec
C sU | j t j � � } | j j d � Wd QRX| j j } | j | j d � | � d S)NZXYZZYz500 ) � assertRaises�nntplib�NNTPPermanentErrorr Z _shortcmd� exception�responser r )r �cmr r r r �test_unknown_command7 s z,NetworkedNNTPTestsMixin.test_unknown_commandc C sx t j j � t j d d � } | j j | � \ } } t | � d k rt | j | d t � | j | d j t
� d S)NZdays� r ) �datetime�dateZtodayZ timedeltar Z newgroupsr �assertIsInstancer r r
)r �dtr r r r r �test_newgroups= s
z&NetworkedNNTPTestsMixin.test_newgroupsc s� � f d d � } � j j � j � } | | � � j d | � � j j � j � } | | � � j j d � } � j | d � d S)Nc s' � j | t � � j � j | � d S)N)r) r
� assertNotIn�
GROUP_NAME)�desc)r r r � _check_descG s z=NetworkedNNTPTestsMixin.test_description.<locals>._check_desc�Pythonz zk.brrtt.baz� )r � descriptionr- �assertInr r )r r/ r. r )r r �test_descriptionF s
z(NetworkedNNTPTestsMixin.test_descriptionc C s� | j j | j � \ } } | j | j d � p< | j d � | � | j | t � | | j } | j | | j j | j � � d S)Nz215 z282 )
r � descriptionsr r r r) �dictr- r r2 )r r Zdescsr. r r r �test_descriptionsV s "
z)NetworkedNNTPTestsMixin.test_descriptionsc C s� | j j | j � } | j d t | � � | \ } } } } } | j | | j � | j | t � | j | t � | j | t � | j | | � | j | j d � | � d S)N� z211 )
r r r- r r r) �int�assertLessEqualr r )r �resultr �count�first�lastr r r r �
test_group_ s z"NetworkedNNTPTestsMixin.test_groupc C sR | j j � \ } } | j | t j � | j | j d � | j | j d � d S)Ni� i� )r r( r) r' �assertGreaterEqualZyearr: )r r r( r r r � test_datej s z!NetworkedNNTPTestsMixin.test_datec C sr | j | t � | j | j � d d d d d d d h � x0 | j � D]" } | j | t t d � f � qH Wd S)N�subject�fromr( z
message-id�
referencesz:bytesz:lines)r) r6 r@ �keys�valuesr
r )r �art_dict�vr r r �_check_art_dictq s z'NetworkedNNTPTestsMixin._check_art_dictc C s� | j j | j � \ } } } } } | j j | d | � \ } } t | � d k rb | j d � | d \ } } | j | | d � | j | | � | j | � d S)Nr8 r zno articles retrieved) r r r- �xoverr �skipTestr@ r: rI ) r r r< r= r> �namer �art_numrG 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 r- �overrI r@ r: )
r r r< r= r> rL �startr rM rG r r r � test_over� s $
z!NetworkedNNTPTestsMixin.test_overc C sn | j j | j � \ } } } } } | j j d | � \ } } x( | D] } | j t t | d � � qF Wd S)NrB rP )r r r- Zxhdrr r
r )r r r< r= r> rL r r r r r � test_xhdr� s $
z!NetworkedNNTPTestsMixin.test_xhdrNc C sq | j | t j � | d k r2 | j | j | � x! | j D] } | j | t � q< W| j | j d d � d S)NrP � .� .
� .
rQ )rV rW rX )r) r Z ArticleInfor Znumberr �bytesr, )r r �articlerM r r r r �check_article_resp� s z*NetworkedNNTPTestsMixin.check_article_respc
s� | j j | j � \ } } } } } x� | | | d f D]f } y | j j | � \ } } Wn@ t j k
r� } z | j j d � s� � w8 WYd d } ~ Xn XPq8 W| j d � | j | j d � | � | j
| | | � | j j | � \ } } | j | j d � | � | j
| | | � | j j | � \ } }
| j | j d � | � | j
| |
| � d d � � � f d d
� | j
D� } � f d d
� |
j
D� } | j | | d g | j
� d S)
NrP 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 | � q Sr r )�.0r )� blacklistr r �
<listcomp>� s zBNetworkedNNTPTestsMixin.test_article_head_body.<locals>.<listcomp>c s" g | ] } � | � s | � q Sr r )r] r )r^ r r r_ � s � )r r r- �headr �NNTPTemporaryErrorr# r rK r r[ �bodyrZ r r )
r r r< r= r> rL rM ra �erc rZ Zfiltered_head_linesZfiltered_linesr )r^ r �test_article_head_body� s, $
z.NetworkedNNTPTestsMixin.test_article_head_bodyc s^ � f d d � } � j � j j d � | � j j � � � j j � \ } } | | � d S)Nc s4 | d } � j | t t f � � j d | � d S)N�LISTz OVERVIEW.FMT)r) r �tupler3 )�capsZ caps_list)r r r � _check_caps� s
z>NetworkedNNTPTestsMixin.test_capabilities.<locals>._check_caps� )r@ r � nntp_version�getcapabilitiesZ capabilities)r ri r rh r )r r �test_capabilities� s
z)NetworkedNNTPTestsMixin.test_capabilitiesc C s; d } d } | j t j | j j d | d | d d �d S)NZ notarealuserZnotarealpassword�user�password�usenetrcF)r r � NNTPErrorr �login)r ZbaduserZbadpwr r r � test_zlogin� s z#NetworkedNNTPTestsMixin.test_zloginc C s. t | � } z | j j � Wd d | _ Xd S)N)r r �quit)r �clsr r r � test_zzquit� s z#NetworkedNNTPTestsMixin.test_zzquitc C sm d d � } xZ t | � D]L } | j d � s1 q t | | � } t | � sO q t | | | | � � q Wd S)Nc s% t j � � � f d d � � } | S)Nc s( t j | j � � � | � Wd QRXd S)N)r �transient_internet� NNTP_HOST)r )�methr r �wrapped� s zHNetworkedNNTPTestsMixin.wrap_methods.<locals>.wrap_meth.<locals>.wrapped)� functools�wraps)ry rz r )ry r � wrap_meth� s !z7NetworkedNNTPTestsMixin.wrap_methods.<locals>.wrap_methZtest_)�dirr �getattr�callable�setattr)ru r} rL ry r r r � wrap_methods� s z$NetworkedNNTPTestsMixin.wrap_methodsc s� � f d d � } | j | j d t d d ��* � | j | � � | j � j � � Wd QRX| j | � � | j | j d t d d �� � � j � Wd QRX| j | � � d S)Nc sD t � d � s d Sy � j � Wn t t f k
r? d SYn Xd S)N�fileFT)�hasattrr �OSError�EOFErrorr )r r r � is_connected� s zANetworkedNNTPTestsMixin.test_with_statement.<locals>.is_connected�timeoutrp F)�
NNTP_CLASSrx �TIMEOUTr r � assertFalsert )r r� r )r r �test_with_statement� s !!z+NetworkedNNTPTestsMixin.test_with_statement)�__name__�
__module__� __qualname__r r r r r% r+ r4 r7 r? rA rI rN �unittestZskipIfrT rU r[ re rm rs rv � classmethodr� r� r r r r r s*
r c @ sL e Z d Z d Z d Z d Z e Z e d d � � Z e d d � � Z
d S) �NetworkedNNTPTestsznews.trigofacile.comzfr.comp.lang.pythonzfr.comp.lang.*c C s� t j d � t j | j � �[ y% | j | j d t d d �| _ Wn. t k
ru t j d | | j f � � Yn XWd QRXd S)NZnetworkr� rp Fz$%s got EOF error on connecting to %r)
r Zrequiresrw rx r� r� r r� r� ZSkipTest)ru r r r �
setUpClass s
z NetworkedNNTPTests.setUpClassc C s | j d k r | j j � d S)N)r rt )ru r r r �
tearDownClass( s z NetworkedNNTPTests.tearDownClassN) r� r� r� rx r- r r r� r� r� r� r r r r r� s r� zrequires SSL supportc @ s@ e Z d Z d Z d Z d Z e e d d � Z d Z d Z
d S)�NetworkedNNTP_SSLTestsz
nntp.aioe.orgzcomp.lang.pythonz comp.lang.*�NNTP_SSLN) r� r� r� rx r- r r r r� r �
test_starttlsr r r r r� - s r� c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)�
_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 sW t j j | � t j � | _ t j � | _ | | _ | j j | j j | j � d S)N)
�io� RawIOBase�__init__�BytesIO�c2s�s2c�handlerrS �readline� push_data)r r� r r r r� J s
z_NNTPServerIO.__init__c C s d S)NTr )r r r r �readableS s z_NNTPServerIO.readablec C s d S)NTr )r r r r �writableV s z_NNTPServerIO.writablec C sF | j j � } | j j d d � | j j | � | j j | � d S)z.Push (buffer) some data to send to the client.r rj N)r� �tell�seek�write)r �data�posr r r r� Y s z_NNTPServerIO.push_datac C sF | j j � } | j j | � | 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 sE | j j � | j j t | � � } t | � } | | d | � <| S)z#The client wants to read a responseN)r� r� r� �readr )r Zbufr� �nr r r �readintoh s
z_NNTPServerIO.readintoN)
r� r� r� �__doc__r� r� r� r� r� r� r r r r r� E s r� c C s( t | � } t j | | � } | | f S)N)r� r� �BufferedRWPair)r� �sior� r r r �make_mock_fileq s r� c sF e Z d Z d Z � f d d � Z � f d d � Z d d � Z � S)�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 sI | j � | _ t | j � \ | _ } t j | d | | � | _ | j S)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� r r )r� r r� y s r� c s" e Z d Z � f d d � Z � S)� MockedNNTPWithReaderModeMixinc s! t � j � | j d d � d S)NZ
readermodeT)r� r� r� )r )r� r r r� � s
z#MockedNNTPWithReaderModeMixin.setUp)r� r� r� r� r r )r� r r� � s r� c @ st e Z d Z d Z d Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d d � Z d d � Z d d! � Z d d d" d# � Z d$ d% � Z d& d' � Z d( d d) d* � Z d( d d+ d, � Z d- Z d. Z e d/ e Z d d0 d1 � Z d d2 d3 � Z d d4 d5 � Z d6 d7 � Z d S)8�
NNTPv1HandlerzA handler for RFC 977z200 NNTP mock serverc C sD 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 rS � s zNNTPv1Handler.startc C s t | d d � S)Nzutf-8�surrogateescape)r
)r r� r r r �_decode� s zNNTPv1Handler._decodec C s� | j r� x7 | j � } | s" d S| j j | � | d k r Pq Wz& | j \ } } | | d | j � Wd d | _ d | _ d | _ Xx
| j | j � � } | s� d S| j d � s� t d j | � � � | d d � } | j � ^ } } t
| d | j � d � } | d k r%| j � q� y | | � Wn= t
k
rr} z t d j | � � | � WYd d } ~ Xq� X| j r� | | f | _ g | _ q� Wd S)
Ns .
rc Fz
z line doesn't end with \r\n: {!r}rj Zhandle_zcommand failed: {!r}���)r� r� rc �appendZ
body_callbackr� �endswith�
ValueError�format�splitr �upper�handle_unknown� Exception)r r ry �tokens�cmdrd r r r r� � s@
+ z NNTPv1Handler.process_pendingc C s
d | _ d S)z7Flag that the client is expected to post a request bodyTN)r� )r r r r � expect_body� s zNNTPv1Handler.expect_bodyc C s | j | � d S)zPush some binary dataN)r� )r r� r r r r� � s zNNTPv1Handler.push_datac C sH t j | � } d j | j � � d } | j d � } | j | � d S)zPush a string literalz
zutf-8N)�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 s6 | d k r | j d � n | j d j | � � 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 se | d k r | j d � nE | d k r8 | j d � n) | d k rT | 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 k r | j d � n� | d k rT | d k rD | j d � q� | j d � n� | d k rp | j d � ns | d k r� | d k s� t � | d k r� | j d
� q� | d k r� | 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� �AssertionError)r �actionZparamr r r � handle_LIST s.
zNNTPv1Handler.handle_LISTc C s� | d k r4 | d k r4 | d k r4 | j d � n} | d k rh | d k rh | d k rh | j d � nI | d k r� | d k r� | d k r� | 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_NEWNEWS: s
z NNTPv1Handler.handle_NEWNEWSc C s- | d k r | 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_XOVERY s
zNNTPv1Handler.handle_XOVERrc c C sh | d k r? | j r/ | j d � | j � qd | j d � n% | j sN t � | j 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 rc r r r � handle_POSTy s
zNNTPv1Handler.handle_POSTc C st | d k rK | j r; | d k r; | j d � | j � qp | j d � n% | j sZ t � | j 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_idrc 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 s� | d k r | j d � nI | d k r8 | j d � n- | d k rT | 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 s� | d k r | j d � nI | d k r8 | j d � n- | d k rT | 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 s� | d k r | j d � nI | d k r8 | j d � n- | d k rT | 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 s� | j r | j d � nx | d k rW | j r> | j d � q� | j d � d | _ n: | d k r| | j d � d | _ n t d j | � � � d S) Nz502 Already Logged Inrn 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) r� r� r� r� r
rS 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 Z d Z d Z d d � Z d d � Z d d d � Z d S) �
NNTPv2Handlerz"A handler for RFC 3977 (NNTP "v2")c C sB d } | j s( | j | j d � � n | j | j 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 USERr1 )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)r� r� r� r� r� r� r� r r r r r� � s r� c s( e Z d Z d Z � f d d � Z � S)�CapsAfterLoginNNTPv2Handlerz3A handler that allows CAPABILITIES only after loginc s* | j s | j d � 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 Z d Z d Z d d � Z d d � Z d d � Z d S) �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 sB d } | j r( | j | j d � � n | j | j 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
.r1 zMODE-)r r� r� )r r� r r r r� s
z.ModeSwitchingNNTPv2Handler.handle_CAPABILITIESc C s6 | j r | d k s t � d | _ | j d � d S)N�readerTz200 Posting allowed)r r� r� )r Zwhatr r r r� s z&ModeSwitchingNNTPv2Handler.handle_MODEN)r� r� r� r� r� r� r� r r r r r � s r c s� e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z! d> d? � Z" d@ dA � Z# � S)B�NNTPv1v2TestsMixinc s t � j � d S)N)r� r� )r )r� r r r� s zNNTPv1v2TestsMixin.setUpc C s | j | j j | j j � d S)N)r r r
r� )r r r r r s z NNTPv1v2TestsMixin.test_welcomec C sR | j d k r% | j d | j j � | j j d d � | j d | j j � d S)Nrj �AUTHINFO�testuser�testpw)rk r3 r �_capsrr r, )r r r r �
test_authinfo s z NNTPv1v2TestsMixin.test_authinfoc C sQ | j j � \ } } | j | d � | j | t j d d d d d d � � d S)Nz111 20100914001155i� � � r � �7 )r r( r r' )r r r( r r r rA ( s z NNTPv1v2TestsMixin.test_datec C sI | j | j j � | j j � } | j | d � | j | j j � d S)Nz205 Bye!)r� r� �closedr rt r r )r r r r r � test_quit- s z NNTPv1v2TestsMixin.test_quitc C sE | j j � \ } } | j | d � | j | d d d d g � 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 3 s z NNTPv1v2TestsMixin.test_helpc C s� | j j � \ } } | j t | � d � | d } | j | t d d d d � � | j j d � \ } } | j t | � d � | d } | j | t d
d d d � � d S)
N� rP zcomp.lang.python.announceZ
0000001153Z
0000000993�mz *distutils*rj 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 j d � \ } } } | j | d � | j | d � | j | d � | j j d � \ } } } | j | d � | j | d � | j | d � | j t j � � } | j j d � Wd QRX| j | j j d � | j t j � � } | j j � Wd QRX| j | j j d � 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 rb r" r# )r r rM r� r$ r r r � test_statK s z NNTPv1v2TestsMixin.test_statc C sL | 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 rM r� r r r � test_next[ s z NNTPv1v2TestsMixin.test_nextc C sL | j j � \ } } } | j | d � | j | d � | j | d � d S)Nz,223 3000234 <45223423@example.com> retrievedi��- z<45223423@example.com>)r r> r )r r rM r� r r r � test_lasta s z NNTPv1v2TestsMixin.test_lastc C sH | j j d � } | j | d � | j j d � } | j | d � d S)Nzcomp.lang.pythonz The Python computer language.zcomp.lang.pythonxr1 )r r2 r )r r. r r r r4 g s z#NNTPv1v2TestsMixin.test_descriptionc C s� | j j d � \ } } | j | d � | j | d d i � | j j d � \ } } | j | d d d d i � | j j d � \ } } | j | i � d S)Nzcomp.lang.pythonz-215 Descriptions in form "group description".z The Python computer language.zcomp.lang.python*zcomp.lang.python.announcez4Announcements about the Python language. (Moderated)zcomp.lang.pythonx)r r5 r )r r r r r r r7 m s
z$NNTPv1v2TestsMixin.test_descriptionsc
C s� | j j d � \ } } } } } | j | j d � | � | j | d � | j | d � | j | d � | j | d � | j t j � � } | j j d � Wd QRX| j } | j | 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 rb r" r# )r r r< r= r> r r$ �excr r r r? { s ! z NNTPv1v2TestsMixin.test_groupc C s� t j d d d d d d � } | j j d | � \ } } d j | j � } | j | | � | j | d d
g � t j d d d d d d � } | j j d | � \ } } | 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� rk r )r r* r ZidsZexpectedr r r � test_newnews� s
z NNTPv1v2TestsMixin.test_newnewsc C ss | j t | � d � | j | d
j d � d � | j | d d � | j | d d � | j | d
d � d S)Nr rP zutf-8u -- Signed by André.rj r` � s .Here is a dot-starting line.s This is just a test article.rQ r� ������)r r �decode)r r r r r �_check_article_body� s
z&NNTPv1v2TestsMixin._check_article_bodyc C sB | 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 s\ | j t | � d � | j | d d � � | j | d d � � | j | d d � d S)Nr r r` r )r r r# r" )r r r r r �_check_article_data� s z&NNTPv1v2TestsMixin._check_article_datac
C so | j j � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j t j � � } | j j d � Wd QRX| j | j j d � 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 rZ r r$ r r rb r" r# )r r �inforM r� r r$ r r r � test_article� s*
z NNTPv1v2TestsMixin.test_articlec C s� t j � } | j j d | � \ } } | j | d � | \ } } } | j | d � | j | d � | j | g � | j � } | j | j d � t | � � | j | j d � t | � � d S)Nr� 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 rZ r �getvaluer r �asciir� )r �fr r% rM r� r r� r r r �test_article_file� s
z$NNTPv1v2TestsMixin.test_article_filec
C so | j j � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j t j � � } | j j d � Wd QRX| j | j j d � 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 ra r r# r r rb r" r# )r r r% rM r� r r$ r r r � test_head� s*
z NNTPv1v2TestsMixin.test_headc C s� t j � } | j j d | � \ } } | j | d � | \ } } } | j | d � | j | d � | j | g � | j � } | j | j d � t | � � | j | j
d � t | � � d S)Nr� 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 ra r r' r r r( r� r� )r r) r r% rM r� r r� r r r �test_head_file� s
z!NNTPv1v2TestsMixin.test_head_filec
C so | j j � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j j d � \ } } | j | d � | \ } } } | j | d � | j | d � | j | � | j t j � � } | j j d � Wd QRX| j | j j d � 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 rc r r" r r rb r" r# )r r r% rM r� r r$ r r r � test_body s*
z NNTPv1v2TestsMixin.test_bodyc C s� t j � } | j j d | � \ } } | j | d � | \ } } } | j | d � | j | d � | j | g � | j � } | j | j d � t | � � | j | j
d � t | � � d S)Nr� 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 rc r r' r� r r( r r� )r r) r r% rM r� r r� r r r �test_body_file' s
z!NNTPv1v2TestsMixin.test_body_filec C s� | j | j d � | � | j t | � d � | d \ } } | j | d � | j | d d d d d d
d d d
d d d d d d d i � | d \ } } | j | d d � | d \ } } | j | d d � d S)Nz224 r r �9 rC zEDoug Hellmann <doug.hellmann-Re5JQEeQqe8AvxtiuMwx3w@public.gmane.org>rB zGRe: ANN: New Plone book with strong Python (and Zope) themes throughoutr( z Sat, 19 Jun 2010 18:04:08 -0400z
message-idz0<4FD05F05-F98B-44DC-8111-C6009C925F0C@gmail.com>rD z <hvalf7$ort$1@dough.gmane.org>z:bytesZ7103z:linesZ16�xrefz+news.gmane.org gmane.comp.python.authors:57rP rj u0 Re: Message d'erreur incompréhensible (par moi))r r r r )r r � overviewsrM rR r r r �check_over_xover_resp; s$
z(NNTPv1v2TestsMixin.check_over_xover_respc C s/ | j j d d � \ } } | j | | � d S)Nr/ �; )r rJ r2 )r r r1 r r r rN P s z NNTPv1v2TestsMixin.test_xoverc C s, | j j d � \ } } | j | | � d S)Nr/ r3 )r/ r3 )r rR r2 )r r r1 r r r rT T 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 s� | j j } | j t | � d � | j | d d � | j | d
d � | j | d d � | j | d d � | j | d
d � d S)NrO rP s .
rj s -- Signed by André.
r s
r s ..Here is a dot-starting line.
r s( From: "Demo User" <nobody@example.net>
rQ r� r r )r� r� r r )r r r r r �_check_posted_bodye s z%NNTPv1v2TestsMixin._check_posted_bodyc G s� | j } | | | � f } d | j _ | | � } | j � | j j d d � } | | | � f } d | j _ | | � } | j � | S)Ns
s
)� sample_postr� r� r4 �replace)r �func� file_factoryr� �postZ func_argsr r r r �_check_post_ihave_subp s
z(NNTPv1v2TestsMixin._check_post_ihave_subc G s� | j | | d t �} | j | | � | j | | d t �} | j | | � | j | | d t j �} | j | | � d d � } | j | | d | �} | j | | � d d � } | j | | d | �} | j | | � d S)Nr8 c S s t | j d d � � S)N�keependsT)�iterr� )r� r r r � iterlines� s z6NNTPv1v2TestsMixin.check_post_ihave.<locals>.iterlinesc S s t | j d d � � S)Nr; F)r< r� )r� r r r r= � s )r: rY r � bytearrayr� r� )r r7 Z success_respr� r r= r r r �check_post_ihave� s z#NNTPv1v2TestsMixin.check_post_ihavec
C sk | j | j j d � d | j _ | j t j � � } | j j | j � Wd QRX| j | j
j d � d S)Nz240 Article received OKFz440 Posting not permitted) r? r r9 r� r� r r rb r5 r r" r# )r r$ r r r � test_post� s z NNTPv1v2TestsMixin.test_postc
C se | j | j j d d � | j t j � � } | j j d | j � Wd QRX| j | j j d � d S)Nz235 Article transferred OKz+<i.am.an.article.you.will.want@example.com>z<another.message.id>z435 Article not wanted)
r? r Zihaver r rb r5 r r" r# )r r$ r r r �
test_ihave� s z NNTPv1v2TestsMixin.test_ihavec C sA t j d d d d d d � } | j t j | j j d | � d S)Ni� rP 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 rA r r r r r r r4 r7 r? r r" r# r$ r& r* r+ r, r- r. r2 rN rT r5 r4 r: r? r@ rA rC r r )r� r r sB
r c @ s. e Z d Z d Z d Z e Z d d � Z d S)� NNTPv1Testsz*Tests an NNTP v1 server (no capabilities).rP c C sO | j j � } | j | i � | j | j j d � | j | j j d � d S)NrP )r rl r rk �nntp_implementation)r rh r r r � test_caps� s zNNTPv1Tests.test_capsN)r� r� r� r� rk r� r� rF r r r r rD � s rD c @ s. e Z d Z d Z d Z e Z d d � Z d S)� NNTPv2Testsz,Tests an NNTP v2 server (with capabilities).rj c C s� | j j � } | j | d d d g d d d g d d g d g d
d d d
d d d g d g d g d g i � | j | j j d � | j | j j d � d S)N�VERSION�2�3ZIMPLEMENTATIONZINNz2.5.1r ZUSERZHDRrf r� z ACTIVE.TIMESz DISTRIB.PATSZHEADERSr� z OVERVIEW.FMTZOVERZPOST�READERr z INN 2.5.1)r rl r rk rE )r rh r r r rF � s
zNNTPv2Tests.test_capsN)r� r� r� r� rk r� r� rF r r r r rG � s rG c @ s. e Z d Z d Z d Z e Z d d � Z d S)�CapsAfterLoginNNTPv2TestszCTests a probably NNTP v2 server with capabilities only after login.rj c C sC | j | j j i � | j j d d � | j d | j j � d S)Nr r rH )r r r rr r3 )r r r r �test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r� r� r� r� rk r� r� rM r r r r rL � s rL c @ s. e Z d Z d Z d Z e Z d d � Z d S)�SendReaderNNTPv2TestszoSame tests as for v2 but we tell NTTP to send MODE READER to a server
that isn't in READER mode by default.rj c C s | j d | j j � d S)NrK )r3 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� rk r r� rO r r r r rN � s rN c @ sv e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e j
e d
� d d � � Z d S)� MiscTestsc sq � f d d � } | d d � | d d � | d d � | d d � | d d � | d
d � | d d � d S)
Nc s � j t j | � | � d S)N)r r �
decode_header)�ar� )r r r �gives� s z+MiscTests.test_decode_header.<locals>.givesr1 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 rS r )r r �test_decode_header� s
z MiscTests.test_decode_headerc C s� d d d d d d d g } | j t j | � d d d
d d d d g � d d d d d d
d g } | j t j | � d d d
d d d d g � d d d d d d d g } | j t j | � d d d
d d d d g � d d d d d d d d d g } | j t j | � d d d
d d d d d d g � d d d d d d
d d d g } | j t j | � d d d
d d d d d d g � d d d d d d
d d g } | j t j | � d d d
d d d d d g � d S)NzSubject:zFrom:zDate:z Message-ID:z References:z:bytesz:linesrB rC r( z
message-idrD zBytes:zLines:zsubject:zFROM:zDaTe:z message-ID:zBYTES:z Xref:fullzDistribution:fullr0 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 j d g } d g } t j | | � } | \ \ } } | j | d � | j | d d d d d d d
d d d
d d d d d d i � d g } t j | | � } | \ \ } } | j | d d � d g } t j | | � } | \ \ } } | j | d d � | j | d d � d S)Nr0 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��- rB zI am just a test articlerC z "Demo User" <nobody@example.com>r( z6 Oct 1998 04:38:40 -0500z
message-idz<45223423@example.com>rD z<45454@example.net>z:bytesZ1234z:linesZ17z"news.example.com misc.test:3000363z�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: � r1 )r Z_DEFAULT_OVERVIEW_FMTZ_parse_overviewr )r r� r ZoverviewrM Zfieldsr r r �test_parse_overview! s.
z MiscTests.test_parse_overviewc s� � f d d � } | d d d d d d d d � | d
d d d d d d d � | d d d d d d d d � | d
d d d d d d d � d S)Nc s) � j t j | | � t j | � � d S)N)r r Z_parse_datetimer' )rR r� �c)r r r rS P s z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624i� r � r �8 � �19990623�135624�990623Z090623i� r )r rS r )r r �test_parse_datetimeO s
z MiscTests.test_parse_datetimec s� � f d d � } | d d d d d d d d
� | d d d d d d d d
� | d
d d d d d d d � � f d d � } | d d d d d � | d d d d d � | d
d d d d � d S)Nc sc t j | | | | | | � } � j t j | � | | f � � j t j | d � | | f � d S)NF)r' r r �_unparse_datetime) �y�M�d�hr �sr� r� r* )r r r rS ] s
z.MiscTests.test_unparse_datetime.<locals>.givesi� r rY r rZ r[ r\ r] i� Z20000623i� r8 rP rj r Z20100605�010203c sZ t j | | | � } � j t j | � | | f � � j t j | d � | | f � d S)NF)r' r( r r r` )ra rb rc r� r� r* )r r r rS g s
�000000r )r rS r )r r �test_unparse_datetimeZ s z MiscTests.test_unparse_datetimec s� � f d d � } | d d d d d d d d
� | d d d d d d d d
� | d
d d d d d d d � � f d d � } | d d d d d � | d d d d d � | d
d d d d � d S)Nc sD t j | | | | | | � } � j t j | d � | | f � d S)NT)r' r r r` ) ra rb rc rd r re r� r� r* )r r r rS t s z5MiscTests.test_unparse_datetime_legacy.<locals>.givesi� r rY r rZ r[ r^ r] i� Z000623i� r8 rP rj r Z100605rf c s; t j | | | � } � j t j | d � | | f � d S)NT)r' r( r r r` )ra rb rc r� r� r* )r r r rS | s rg r )r rS r )r r � test_unparse_datetime_legacyq s z&MiscTests.test_unparse_datetime_legacyzrequires SSL supportc C s | j t t d � � d S)Nr� )r r� r )r r r r �test_ssl_support� s zMiscTests.test_ssl_supportN)
r� r� r� rT rU rW r_ rh ri r� �
skipUnless�sslrj r r r r rP � s ( . rP c @ s" e Z d Z d Z d d � Z d S)�PublicAPITestsz>Ensures that the correct values are exposed in the public API.c C sp | j t t d � � d d d d d d d d g } t d k rM | j d
� | j t t j � t | � � d S) N�__all__r rq �NNTPReplyErrorrb r! �NNTPProtocolErrorrB rQ r� )r r� r rl r� r �setrn )r Z
target_apir r r �test_module_all_attribute� s
z(PublicAPITests.test_module_all_attributeN)r� r� r� r� rr r r r r rm � s rm c @ sm e Z d Z d Z e j Z d d d d � Z d d � Z d d � Z d d
� Z
d d � Z d
d � Z d S)�MockSocketTestszaTests involving a mock socket object
Used where the _NNTPServerIO file object is not enough.Nc s� G� f d d � d � } G� � � f d d � d � � d � g � t d | � �8 | j | | � � | j d d | d | �Wd QRXWd QRX| j � � x � D] } | j | j � q� Wd S)
Nc s" e Z d Z � f d d � Z d S)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� ru r )rt r r �mock_socket_module� s rv c s7 e Z d Z � f d d � Z � � f d d � Z d S)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� rx r| r )r{ r� rw r r rt � s rt Fznntplib.socketZdummyrn ro )r ZassertRaisesRegex�
nntp_classr r
)r r� Zexpected_error_typeZexpected_error_msgrr ro rv r) r )rt r{ r� rw r �"check_constructor_error_conditions� s '
z2MockSocketTests.check_constructor_error_conditionsc C s3 Gd d � d t � } | j | t j | j � d S)Nc @ s e Z d Z d Z d S)z1MockSocketTests.test_bad_welcome.<locals>.Handlerz Bad WelcomeN)r� r� r� r
r r r r �Handler� s r )r� r~ r rp r
)r r r r r �test_bad_welcome� s z MockSocketTests.test_bad_welcomec C s3 Gd d � d t � } | j | t j | j � d S)Nc @ s e Z d Z d Z d S)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 rb r
)r r r r r �$test_service_temporarily_unavailable� s z4MockSocketTests.test_service_temporarily_unavailablec C s3 Gd d � d t � } | j | t j | j � d S)Nc @ s e Z d Z d Z d S)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� f d d � d t � } d � | j | t j � � d S)Nc s" e Z d Z � f d d � Z d S)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 ro )r r r )r� r �test_bad_capabilities� s z%MockSocketTests.test_bad_capabilitiesc sN d } d } G� f d d � d t � } d � | j | t j � | | � d S)Nzt@e.com�pythonc s" e Z d Z � f d d � Z d S)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 rr ro r r )r� r �test_login_aborted� s z"MockSocketTests.test_login_aborted)
r� r� r� r� r r r} r~ r� r� r� r� r� r r r r rs � s rs c @ s" e Z d Z d Z d d � Z d S)�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 Z d Z e d d � � Z d S)� MockSslTestsc O s t j | d t | � 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 @ sF e Z d Z d d � Z d d � Z e j e d � d d � � Z d S) �LocalServerTestsc C s� t j � } t j | � } | j � t j d | j d | f � | _ | j j � | j | j j
� t t j | d d �j
� | _ | j | j j d d d � d S)N�targetr� rp F)ry r Z bind_portZlisten� threadingZThread�
run_serverZ
backgroundrS Z
addCleanupr� r ZHOST� __enter__�nntp�__exit__)r r� Zportr r r r� � s
!zLocalServerTests.setUpc C s3 | � | j � \ } } Wd QRXt j � � } | j | � | j | j d � � } | j d � x� | j � } | d k r� | j d � qd | d k r� | j � | j d � t j | d d d t
�} | j | � | j | j d � � } qd | d
k r| j d � Pqd t d j | � � � qd WWd QRXd S)
N�rbs 200 Server ready
s CAPABILITIES
s. 101 Capability list:
VERSION 2
STARTTLS
.
s
STARTTLS
s 382 Begin TLS negotiation now
Z server_sideT�certfiles QUIT
s
205 Bye!
zUnexpected command {!r})
Zaccept�
contextlib� ExitStack�
enter_contextr| Zsendallr� rx rl r� r� r� r� )r r� Zclientr� Zcleanupr r� r r r r� � s,
zLocalServerTests.run_serverzrequires SSL supportc C s� | j j } | j j } | j j � | j | | j j � | j | | j j � | j | j j t j � | j t | j j � d S)N)
r� r� r� ZstarttlsZassertNotEqualr) rl Z SSLSocketr r� )r r� r� r r r r� s
z LocalServerTests.test_starttlsN) r� r� r� r� r� r� rk rl r� r r r r r� � s r� �__main__)5r� ry r' r� r� r{ r� Zos.path�osZtestr r r r Z
unittest.mockr rl � ImportErrorr� r� �pathr� �dirname�__file__r� r r� ZTestCaser� rk r� r� r� r� r� r� r� r� r� r r rD rG rL rN rP rm rs r� r� r� r� �mainr r r r �<module> sh
!�
, � F
� �
� N :