o b@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}. cCsF||_t|jd}|dWddS1swYdS)Nwzsomeuser: differentuser )mktemp aliasFilenameopenwrite)self aliasFiler@/usr/lib/python3/dist-packages/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 rrrrtestAliasesWithoutDomainsz%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 assertEquallenassertIsInstancerTCP4ServerEndpoint)rserviceoptionsrrr _endpointTest0szOptionsTests._endpointTestcC|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;zOptionsTests.test_endpointSMTPcCr#)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_endpointPOP3Br'zOptionsTests.test_endpointPOP3cCsht}|g|t|dd||ddtj|t|dd||ddtjdS)zO POP3 and SMTP each listen on a TCP4ServerEndpoint by default. r(rrr$Nrrr!rrrtest_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)rrr _getEndpointsassertNotEqualr*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)rr!excrrrtest_esmtpWithoutHostnamensz&OptionsTests.test_esmtpWithoutHostnamecCsbt}|ddg|t|dd|dd}|j}|jjj}|D] }||||q$dS)zC Tests that the --auth option registers a checker. z--authzmemory:admin:admin:bob:password credCheckersrrN)rrrrcredentialInterfacesr smtpPortalcheckers)rr!checker interfacesregistered_checkersifacerrr test_authws  zOptionsTests.test_authN)__name__ __module__ __qualname____doc__rrrr"r&r)r+r0r1r4r=rrrrr s    r c@seZdZdZdZddZdS) SpyEndpointzG SpyEndpoint remembers what factory it is told to listen with. NcCs||_tdS)N) listeningWithrsucceed)rfactoryrrrlistens zSpyEndpoint.listen)r>r?r@rArCrFrrrrrBs rBc@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) rBrrprivilegedStartService startService addCleanup stopServicerrC)rkey factoryClass cleartextsecureconfigr rrr_endpointServerTests  z$MakeServiceTests._endpointServerTestcC|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)rQr POP3Factoryrrrr test_pop3zMakeServiceTests.test_pop3cCrR)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)rQr SMTPFactoryrrrr test_smtprUzMakeServiceTests.test_smtpN)r>r?r@rArQrTrWrrrrrGs  rGN)rAtwisted.internetrr twisted.mailrtwisted.mail.taprrtwisted.python.usagertwisted.trial.unittestrr rBrGrrrrs   v