U j@sdZddlmZmZmZddlmZddlmZddl m Z m Z ddl m Z mZmZddlmZddlmZdd lmZed Zed Zererdd lmZdd lmZGdddeZdS)z! Tests for L{twisted.conch.tap}. )AnyTupleUnion)StreamServerEndpointService)error)FilePasswordDBICredentialsChecker)ISSHPrivateKeyIUsernamePasswordUsernamePassword)Deferred) requireModule)TestCase cryptographyztwisted.conch.unix)tap)OpenSSHFactoryc@seZdZdZesdZesdZdZddddZddd d Z ddd d Z ddd dZ ddddZ e dddZeddddZddddZdS)MakeServiceTestsz' Tests for L{tap.makeService}. zcan't run without cryptographyz can't run on non-posix computers)siamusersthisispasswordN)returnc CsB||_t|jd}|d|jW5QRXt|_dS)z/ Create a file with two users. zwb+:N) mktempfilenameopenwritejoinusernamePasswordrOptionsoptions)selffr?/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/conch/test/test_tap.pysetUp(s zMakeServiceTests.setUpcCs@t}t|}||t||jjd||jt dS)z L{tap.makeService} returns a L{StreamServerEndpointService} instance running on TCP port 22, and the linked protocol factory is an instance of L{OpenSSHFactory}. N) rr makeServiceassertIsInstancer assertEqualZendpointZ_portfactoryr)rconfigservicerrr test_basic1s   zMakeServiceTests.test_basiccCsNd}|t|jdd|t|jdd||t|jdd|fdS)z Make sure that if the C{--auth} command-line option is not passed, the default checkers are (for backwards compatibility): SSH and UNIX ZcredInterfacesz)SSH should be one of the default checkersz*UNIX should be one of the default checkers credCheckersz&There should be %d checkers by defaultN)ZassertInr rr r%len)rZ numCheckersrrr test_defaultAuths=s  z"MakeServiceTests.test_defaultAuthscCs0|jdd|jg|t|jdddS)z The C{--auth} command-line option will add a checker to the list of checkers, and it should be the only auth checker --authfile:r+Nr parseOptionsrr%r,rrrr test_authAddedTszMakeServiceTests.test_authAddedcCs4|jdd|jddg|t|jdddS)z Multiple C{--auth} command-line options will add all checkers specified to the list ofcheckers, and there should only be the specified auth checkers (no default checkers). r.r/zmemory:testuser:testpasswordr+r*Nr1r3rrr test_multipleAuthAdded\sz'MakeServiceTests.test_multipleAuthAddedcCsT|jdd|jg|jdd}||tt|jdd}|||t j S)z The checker created by the C{--auth} command-line option returns a L{Deferred} that fails with L{UnauthorizedLogin} when presented with credentials that are unknown to that checker. r.r/r+rsfake) rr2rr$rr rZ assertFailurerequestAvatarIdrZUnauthorizedLogin)rcheckerinvalidrrr test_authFailurels z!MakeServiceTests.test_authFailurecsdjddjgjdd}tj|}tttdfddfdd }| |S) z The checker created by the C{--auth} command-line option returns a L{Deferred} that returns the avatar id when presented with credentials that are known to that checker. r.r/r+r6rN)usernamercs|jdS)N)r%r;)r;correctrrr checkSuccesssz7MakeServiceTests.test_authSuccess..checkSuccess) rr2rr rr7rbytesrZ addCallback)rr8dr>rr<r test_authSuccess{s   "z!MakeServiceTests.test_authSuccesscCs8t}t|}|jj}|t|jt t hdS)z The L{OpenSSHFactory} built by L{tap.makeService} has a portal with L{ISSHPrivateKey} and L{IUsernamePassword} interfaces registered as checkers. N) rrr#r&portalr%setZcheckerskeysr r )rr'r(rBrrr test_checkerss  zMakeServiceTests.test_checkers)__name__ __module__ __qualname____doc__rskipunixrr!r)r-r4r5rr:r rArErrrr rs  rN)rItypingrrrZtwisted.application.internetrZ twisted.credrZtwisted.cred.checkersrrZtwisted.cred.credentialsr r r Ztwisted.internet.deferr Ztwisted.python.reflectr Ztwisted.trial.unittestrrrKZ twisted.conchrZ$twisted.conch.openssh_compat.factoryrrrrrr s