U  W[W@sdZz ddlZWnek r(dZYnXz ddlZWnek rNdZYnXzddlmZWnek rxdZYnXerererddlmZddlmZddl m Z ddl m Z ddl mZdd l mZmZdd lmZGd d d eZdS) z! Tests for L{twisted.conch.tap}. N)unix)tap)OpenSSHFactory)StreamServerEndpointService)error)ISSHPrivateKey)IUsernamePasswordUsernamePassword)TestCasec@sleZdZdZesdZesdZes$dZdZddZ dd Z d d Z d d Z ddZ ddZddZddZdS)MakeServiceTestsz' Tests for L{tap.makeService}. zcan't run without cryptographyzCannot run without PyASN1z can't run on non-posix computers)siamusersthisispasswordc CsB||_t|jd}|d|jW5QRXt|_dS)z/ Create a file with two users. zwb+:N) ZmktempfilenameopenwritejoinusernamePasswordrOptionsoptions)selffr=/usr/lib/python3/dist-packages/twisted/conch/test/test_tap.pysetUp4s 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 makeServiceZassertIsInstancer assertEqualZendpointZ_portfactoryr)rconfigservicerrr test_basic>s   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)ZassertInrrrrlen)rZ numCheckersrrrtest_defaultAuthsKsz"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 parseOptionsr rr"rrrrtest_authAddedZszMakeServiceTests.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 Nr'r)rrrtest_multipleAuthAddedcs z'MakeServiceTests.test_multipleAuthAddedcCsH|jdd|jg|jdd}t|jdd}|||tjS)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!rZfake) rr(r r rZ assertFailurerequestAvatarIdrZUnauthorizedLogin)rcheckerZinvalidrrrtest_authFailurensz!MakeServiceTests.test_authFailurecsPjddjgjdd}tj|}fdd}||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!r,cs|jdS)N)rusername)r0Zcorrectrrr checkSuccesssz7MakeServiceTests.test_authSuccess..checkSuccess)rr(r r rr-Z addCallback)rr.dr2rr1rtest_authSuccess|s   z!MakeServiceTests.test_authSuccesscCs<t}t|}|jj}|t|jtt t gdS)z The L{OpenSSHFactory} built by L{tap.makeService} has a portal with L{ISSHPrivateKey} and L{IUsernamePassword} interfaces registered as checkers. N) rrrrportalrsetZcheckerskeysrr)rrrr5rrr test_checkerss   zMakeServiceTests.test_checkersN)__name__ __module__ __qualname____doc__ cryptographyskippyasn1rrrrr#r*r+r/r4r8rrrrr $s     r )r<r= ImportErrorr?Z twisted.conchrrZ$twisted.conch.openssh_compat.factoryrZtwisted.application.internetrZ twisted.credrZtwisted.cred.credentialsrrr Ztwisted.trial.unittestr r rrrrs*