o >h@sdZddlmZmZmZddlmZddlmZddl m Z m Z ddl m Z mZmZddlmZddlmZdd lmZed Zed ZerTerTdd lmZdd lmZGdddeZdS)z! Tests for L{twisted.conch.tap}. )AnyTupleUnion)StreamServerEndpointService)error)FilePasswordDBICredentialsChecker)ISSHPrivateKeyIUsernamePasswordUsernamePassword)Deferred) requireModule)TestCase cryptographyztwisted.conch.unix)tap)OpenSSHFactoryc@seZdZdZes dZesdZdZdddZdd d Z dd d Z dd dZ dddZ de fddZdedfddZdddZdS)MakeServiceTestsz' Tests for L{tap.makeService}. zcan't run without cryptographyz can't run on non-posix computers)siamusersthisispasswordreturnNcCsV||_t|jd}|d|jWdn1swYt|_dS)z/ Create a file with two users. zwb+:N) mktempfilenameopenwritejoinusernamePasswordrOptionsoptions)selffr{/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/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 assertEqualendpoint_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 credInterfacesz)SSH should be one of the default checkersz*UNIX should be one of the default checkers credCheckersz&There should be %d checkers by defaultN)assertInr rr r%len)r 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%r0rrrr 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). r3r4zmemory:testuser:testpasswordr.r,Nr6r8rrr 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. r3r4r.rsfake) rr7rr$rr r assertFailurerequestAvatarIdrUnauthorizedLogin)rcheckerinvalidrrr test_authFailurels  z!MakeServiceTests.test_authFailurecsfjddjgjdd}tj|}dtttdfddffd d }| |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. r3r4r.r;usernamerrNcs|jdS)N)r%rB)rBcorrectrrr checkSuccesssz7MakeServiceTests.test_authSuccess..checkSuccess) rr7rr rr=rbytesr addCallback)rr?drErrCr 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%setcheckerskeysr r )rr)r*rJrrr test_checkerss  zMakeServiceTests.test_checkers)rN)__name__ __module__ __qualname____doc__rskipunixrr!r+r2r9r:rrAr rIrNrrrr rs   rN)rRtypingrrrtwisted.application.internetr twisted.credrtwisted.cred.checkersrrtwisted.cred.credentialsr r r twisted.internet.deferr twisted.python.reflectr twisted.trial.unittestrrrT twisted.conchr$twisted.conch.openssh_compat.factoryrrrrrr s