Your IP : 216.73.216.213
�
g f�� � @ s� 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 Xd 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. � Z( e j e d
� Gd/ d0 � d0 e' � � Z) e* d1 k r�e j+ � n d S)2� N)�support)�NNTP� GroupInfo)�patch� c @ s3 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 d d � Z d d � Z d d d � Z d d! � Z d" d# � Z e j e d$ � 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/python34/lib64/python3.4/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 se | j j � \ } } t | � d k ra | j t t | d � � | j t t | d j � � n d S)Nr )r �list�lenr r r r
�group)r r �groupsr r r � test_list% s z!NetworkedNNTPTestsMixin.test_listc C sk | j j | j � \ } } t | � d k rg | j t t | d � � | j t t | d j � � n 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 sT | j t j � � } | j j d � Wd QX| j j } | j | j d � | � d S)NZXYZZYz500 ) � assertRaises�nntplib�NNTPPermanentErrorr Z _shortcmd� exception�responser r )r �cmr r r r �test_unknown_command1 s z,NetworkedNNTPTestsMixin.test_unknown_commandc C s{ t j j � t j d d � } | j j | � \ } } t | � d k rw | j | d t � | j | d j t
� n d S)NZdays� r ) �datetime�dateZtodayZ timedeltar Z newgroupsr �assertIsInstancer r r
)r �dtr r r r r �test_newgroups7 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_descA s z=NetworkedNNTPTestsMixin.test_description.<locals>._check_desc�Pythonz zk.brrtt.baz� )r � descriptionr- �assertInr r )r r/ r. r )r r �test_description@ 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_descriptionsP 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_groupY 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_dated 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_dictk s z'NetworkedNNTPTestsMixin._check_art_dictc C s� | j j | j � \ } } } } } | j j | d | � \ } } t | � d k re | j d � n | 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_xoverv s $ z"NetworkedNNTPTestsMixin.test_xoverc
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 st | j | t j � | d k r5 | j | j | � n 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]i } y | j j | � \ } } WnC t j k
r� } z | j j d � s� � n 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_capabilitieszrequires SSL supportc C s� | j j } | j j } y | j j � Wn" t j k
rM | j d � Yn\ X| j | | j j � | j | | j j � | j | j j t j
� | j t | j j � d S)Nz!STARTTLS not supported by server.)
r �file�sockZstarttlsr r! rK ZassertNotEqualr) �sslZ SSLSocketr �
ValueError)r rn ro r r r �
test_starttls� s z%NetworkedNNTPTestsMixin.test_starttlsc 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 ss d d � } x` t | � D]R } | j d � s4 q n t | | � } t | � sU q n t | | | | � � q Wd S)Nc s% t j � � � f d d � � } | S)Nc s' t j | j � � � | � Wd QXd 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� rL r~ r r r � wrap_methods� s z$NetworkedNNTPTestsMixin.wrap_methodsc s� � f d d � } | j | j d t d d ��* � | j | � � | j � j � � Wd QX| j | � � | j | j d t d d �� � � j � Wd QX| j | � � d S)Nc sD t � d � s d Sy � j � Wn t t f k
r? d SYn Xd S)Nrn FT)�hasattrr �OSError�EOFErrorr )r r r � is_connected s zANetworkedNNTPTestsMixin.test_with_statement.<locals>.is_connected�timeoutru F)�
NNTP_CLASSr} �TIMEOUTr r � assertFalsery )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 rT rU r[ re rm �unittest�
skipUnlessrp rr rx r{ � 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 sK t j d � t j | j � �& | j | j d t d d �| _ Wd QXd S)NZnetworkr� ru F)r Zrequiresr| r} r� r� r )rz r r r �
setUpClass' s
z NetworkedNNTPTests.setUpClassc C s# | j d k r | j j � n d S)N)r ry )rz r r r �
tearDownClass- s z NetworkedNNTPTests.tearDownClassN) r� r� r� r} 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� r} r- r r� r r� r rr r r r r r� 2 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� O s
z_NNTPServerIO.__init__c C s d S)NTr )r r r r �readableX s z_NNTPServerIO.readablec C s d S)NTr )r r r r �writable[ 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� ^ 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� e 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 �readintom s
z_NNTPServerIO.readintoN)
r� r� r� �__doc__r� r� r� r� r� r� r r r r r� J s r� c C s( t | � } t j | | � } | | f S)N)r� r� �BufferedRWPair)r� �siorn r r r �make_mock_filev 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�kwargsrn r r r r� � s z MockedNNTPTestsMixin.make_server)r� r� r� r� r� r� r� r r )r� r r� ~ 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� x: | j � } | s" d S| j j | � | d k r Pq q Wz& | j \ } } | d | j | � Wd d | _ d | _ d | _ Xn x| j | j � � } | s� d S| j d � s� t d j | � � � n | d d � } | j � ^ } } t
| d | j � d � } | d k r.| j � q� y | | � Wn= t
k
r{} z t d j | � � | � WYd d } ~ Xq� X| j r� | | f | _ g | _ q� 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� �endswithrq �format�splitr� �upper�handle_unknown� Exception)r r r~ �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_XOVER^ 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_POST~ 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 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) 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 sU | j d k r( | j d | j j � n | j j d d � | j d | j j � d S)Nrj �AUTHINFO�testuser�testpw)rk r3 r �_capsrw 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 ry r r )r r r r r � test_quit2 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 8 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 B 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 QX| j | j j d � | j t j � � } | j j � Wd QX| 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_statP 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_lastf 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 l s z#NNTPv1v2TestsMixin.test_descriptionc C s� | j j d � \ } } | j | d � | j | i d d 6� | j j d � \ } } | j | i d d 6d d 6� | j j d � \ } } | 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.python.announcezcomp.lang.pythonx)r r5 r )r r r r r r r7 r 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 QX| 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 sn | 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 QX| 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)Nrn 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 sn | 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 QX| 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)Nrn 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 sn | 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 QX| 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)Nrn 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 | i d d 6d d 6d d
6d d 6d
d 6d d 6d d 6d d 6� | d \ } } | j | d d � | d \ } } | j | d d � d S)Nz224 r! r �9 zEDoug Hellmann <doug.hellmann-Re5JQEeQqe8AvxtiuMwx3w@public.gmane.org>rC zGRe: ANN: New Plone book with strong Python (and Zope) themes throughoutrB z Sat, 19 Jun 2010 18:04:08 -0400r( z0<4FD05F05-F98B-44DC-8111-C6009C925F0C@gmail.com>z
message-idz <hvalf7$ort$1@dough.gmane.org>rD Z7103z:bytesZ16z:linesz+news.gmane.org gmane.comp.python.authors:57�xrefrP 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)Nr2 �; )r rJ r5 )r r r4 r r r rN U s z NNTPv1v2TestsMixin.test_xoverc C s, | j j d � \ } } | j | | � d S)Nr2 r6 )r2 r6 )r rR r5 )r r r4 r r r rT Y 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_bodyj 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� r7 �replace)r �func� file_factoryr� �postZ func_argsr r r r �_check_post_ihave_subu 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)Nr; 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 r: Z success_respr� r r@ r r r �check_post_ihave� s z#NNTPv1v2TestsMixin.check_post_ihavec C sj | j | j j d � d | j _ | j t j � � } | j j | j � Wd QX| j | j
j d � d S)Nz240 Article received OKFz440 Posting not permitted) rB r r< r� r� r r rb r8 r r" r# )r r$ r r r � test_post� s z NNTPv1v2TestsMixin.test_postc
C sd | j | j j d d � | j t j � � } | j j d | j � Wd QX| 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)
rB r Zihaver r rb r8 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/ r0 r1 r5 rN rT r8 r7 r= rB rC rD rF 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� rI r r r r rG � s rG 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 | i d d g d 6d d g d 6d g d 6g d 6d
d d d
d d g d 6g d 6g d 6g d 6� | j | j j d � | j | j j d � d S)N�2�3�VERSIONZINNz2.5.1ZIMPLEMENTATIONZUSERr ZHDRr� z ACTIVE.TIMESz DISTRIB.PATSZHEADERSr� z OVERVIEW.FMTrf ZOVERZPOST�READERr! z INN 2.5.1)r rl r rk rH )r rh r r r rI � s
zNNTPv2Tests.test_capsN)r� r� r� r� rk r� r� rI r r r r rJ � s rJ 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 rM )r r r
rw r3 )r r r r �test_caps_only_after_login� s z4CapsAfterLoginNNTPv2Tests.test_caps_only_after_loginN)r� r� r� r� rk r r� rP r r r r rO � s rO 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)NrN )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� rR r r r r rQ � s rQ 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 rV 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:fullr3 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 | i d d 6d d 6d d 6d
d 6d d
6d d 6d d 6d d 6� d g } t j | | � } | \ \ } } | j | d d � d g } t j | | � } | \ \ } } | j | d
d � | j | d d � d S)Nr3 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 articlerB z "Demo User" <nobody@example.com>rC z6 Oct 1998 04:38:40 -0500r( z<45223423@example.com>z
message-idz<45454@example.net>rD Z1234z:bytesZ17z:linesz"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' )rU r� �c)r r r rV U s z,MiscTests.test_parse_datetime.<locals>.givesZ19990623135624i� r � r �8 � �19990623�135624�990623Z090623i� r )r rV r )r r �test_parse_datetimeT 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 rV b s
z.MiscTests.test_unparse_datetime.<locals>.givesi� r r\ r r] 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 rc )rd re rf r� r� r* )r r r rV l s
�000000r )r rV r )r r �test_unparse_datetime_ 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 rc ) rd re rf rg r rh r� r� r* )r r r rV y s z5MiscTests.test_unparse_datetime_legacy.<locals>.givesi� r r\ r r] r^ ra r` i� Z000623i� r8 rP rj r! Z100605ri c s; t j | | | � } � j t j | d � | | f � d S)NT)r' r( r r rc )rd re rf r� r� r* )r r r rV � s rj r )r rV r )r r � test_unparse_datetime_legacyv 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� rW rX rZ rb rk rl r� r� rp rm r r r r rS � s ( . rS 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 ss | j t t d � � d d d d d d d d g } t d k rP | j d
� n | j t t j � t | � � d S) N�__all__r rv �NNTPReplyErrorrb r! �NNTPProtocolErrorrE rT r� )r r� r rp r� r �setro )r Z
target_apir r r �test_module_all_attribute� s z(PublicAPITests.test_module_all_attributeN)r� r� r� r� rs r r r r rn � s rn 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 | � �7 | j | | � � | j d d | d | �Wd QXWd QX| 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� rv r )ru r r �mock_socket_module� s rw 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� rn )�filesr� r r �makefile� s
zOMockSocketTests.check_constructor_error_conditions.<locals>.MockSocket.makefileN)r� r� r� ry r} r )r| r� rx r r ru � s ru Fznntplib.socketZdummyrs rt )r ZassertRaisesRegex�
nntp_classr r )r r� Zexpected_error_typeZexpected_error_msgrw rt rw r, r )ru r| r� rx 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 rq 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 unavilableN)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 unavilableN)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 rp )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 rw rt 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 rt � s rt 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 )ro r� 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� �__main__),r� rz r' r� r� r� �
contextlibZtestr r r r Z
unittest.mockr rp � ImportErrorr� r r� ZTestCaser� r� r� r� r� r� r� r� r� r� r r r rG rJ rO rQ rS rn rt r� r� r� �mainr r r r �<module> sZ
�
, � F
� �
� N