U P|j@szdZddlmZmZddlmZddlmZmZddl m Z ddl m Z Gddde Z Gd d d ZGd d d e Zd S)z Tests for L{twisted.mail.tap}. )defer endpoints) protocols)Options makeService) UsageError)TestCasec@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS) OptionsTestszK Tests for the command line option parser used for I{twistd mail}. c Cs0||_t|jd}|dW5QRXdS)Nwzsomeuser: differentuser )mktemp aliasFilenameopenwrite)selfZ aliasFilerB/tmp/pip-unpacked-wheel-audpp6lv/twisted/mail/test/test_options.pysetUps zOptionsTests.setUpcCs|ttjd|jgdS)zi Test that adding an aliases(5) file before adding a domain raises a UsageError. --aliasesN) assertRaisesrr parseOptionsr rrrrtestAliasesWithoutDomains z%OptionsTests.testAliasesWithoutDomaincCstddd|jgdS)z Test that adding an aliases(5) file to an IAliasableDomain at least doesn't raise an unhandled exception. z--maildirdbmdomainzexample.com=example.comrN)rrr rrrr testAliases"szOptionsTests.testAliasescCsFt}|d|dg|t||d|||dtjdS)z Use L{Options} to parse a single service configuration parameter and verify that an endpoint of the correct type is added to the list for that service. z--ztcp:1234rNrr assertEquallenassertIsInstancerZTCP4ServerEndpoint)rserviceoptionsrrr _endpointTest0szOptionsTests._endpointTestcCs|ddS)z When I{--smtp} is given a TCP endpoint description as an argument, a TCPServerEndpoint is added to the list of SMTP endpoints. smtpNr rrrrtest_endpointSMTP;szOptionsTests.test_endpointSMTPcCs|ddS)z When I{--pop3} is given a TCP endpoint description as an argument, a TCPServerEndpoint is added to the list of POP3 endpoints. pop3Nr"rrrrtest_endpointPOP3BszOptionsTests.test_endpointPOP3cCsht}|g|t|dd||ddtj|t|dd||ddtjdS)zO POP3 and SMTP each listen on a TCP4ServerEndpoint by default. r$rrr!Nrrrrrrtest_protoDefaultsIs  zOptionsTests.test_protoDefaultscCsxt}|dg||ddg||ddgt}|dg||ddg||ddgdS)zg The I{--no-pop3} and I{--no-smtp} options disable POP3 and SMTP respectively. --no-pop3Nr$r! --no-smtp)rrrZ _getEndpointsZassertNotEqualr&rrrtest_protoDisableVs  zOptionsTests.test_protoDisablecCst}|t|jddgdS)zI If all protocols are disabled, L{UsageError} is raised. r(r)N)rrrrr&rrrtest_allProtosDisabledErrores z(OptionsTests.test_allProtosDisabledErrorcCs,t}|t|jdg}|dt|dS)zu If I{--esmtp} is given without I{--hostname}, L{Options.parseOptions} raises L{UsageError}. z--esmtpz--esmtp requires --hostnameN)rrrrrstr)rrexcrrrtest_esmtpWithoutHostnamensz&OptionsTests.test_esmtpWithoutHostnamecCsbt}|ddg|t|dd|dd}|j}|jjj}|D]}||||qHdS)zC Tests that the --auth option registers a checker. z--authzmemory:admin:admin:bob:passwordZ credCheckersrrN)rrrrZcredentialInterfacesrZ smtpPortalZcheckers)rrcheckerZ interfacesZregistered_checkersZifacerrr test_authws  zOptionsTests.test_authN)__name__ __module__ __qualname____doc__rrrr r#r%r'r*r+r.r0rrrrr s     r c@seZdZdZdZddZdS) SpyEndpointzG SpyEndpoint remembers what factory it is told to listen with. NcCs||_tdS)N) listeningWithrZsucceed)rfactoryrrrlistenszSpyEndpoint.listen)r1r2r3r4r6r8rrrrr5sr5c@s(eZdZdZddZddZddZdS) MakeServiceTestsz3 Tests for L{twisted.mail.tap.makeService} cCsbt}t}t}||g||<t|}||||j||j|||j|dS)z Configure a service with two endpoints for the protocol associated with C{key} and verify that when the service is started a factory of type C{factoryClass} is used to listen on each of them. N) r5rrZprivilegedStartServiceZ startServiceZ addCleanupZ stopServicerr6)rkeyZ factoryClassZ cleartextsecureconfigrrrr_endpointServerTests  z$MakeServiceTests._endpointServerTestcCs|dtjdS)z If one or more endpoints is included in the configuration passed to L{makeService} for the C{"pop3"} key, a service for starting a POP3 server is constructed for each of them and attached to the returned service. r$N)r=rZ POP3Factoryrrrr test_pop3szMakeServiceTests.test_pop3cCs|dtjdS)z If one or more endpoints is included in the configuration passed to L{makeService} for the C{"smtp"} key, a service for starting an SMTP server is constructed for each of them and attached to the returned service. r!N)r=rZ SMTPFactoryrrrr test_smtpszMakeServiceTests.test_smtpN)r1r2r3r4r=r>r?rrrrr9s r9N)r4Ztwisted.internetrrZ twisted.mailrZtwisted.mail.taprrZtwisted.python.usagerZtwisted.trial.unittestrr r5r9rrrrs   v